ts

src/api.ts

@@ -0,0 +1,182 @@
+import fetch from 'node-fetch';
+import fs from 'node:fs';
+import util from 'node:util';
+import path from 'node:path';
+import ProgressBar from 'progress';
+import packageJson from '../package.json';
+import tcpp from 'tcp-ping';
+import filesizeParser from 'filesize-parser';
+import { pricingPageUrl } from './utils';
+import { Session } from 'types';
+
+const tcpPing = util.promisify(tcpp.ping);
+
+let session: Session | undefined;
+let savedSession: Session | undefined;
+
+const defaultEndpoint = 'https://update.reactnative.cn/api';
+let host = process.env.PUSHY_REGISTRY || defaultEndpoint;
+
+const userAgent = `react-native-update-cli/${packageJson.version}`;
+
+export const getSession = function () {
+  return session;
+};
+
+export const replaceSession = function (newSession: { token: string }) {
+  session = newSession;
+};
+
+export const loadSession = async function () {
+  if (fs.existsSync('.update')) {
+    try {
+      replaceSession(JSON.parse(fs.readFileSync('.update', 'utf8')));
+      savedSession = session;
+    } catch (e) {
+      console.error(
+        'Failed to parse file `.update`. Try to remove it manually.',
+      );
+      throw e;
+    }
+  }
+};
+
+export const saveSession = function () {
+  // Only save on change.
+  if (session !== savedSession) {
+    const current = session;
+    const data = JSON.stringify(current, null, 4);
+    fs.writeFileSync('.update', data, 'utf8');
+    savedSession = current;
+  }
+};
+
+export const closeSession = function () {
+  if (fs.existsSync('.update')) {
+    fs.unlinkSync('.update');
+    savedSession = undefined;
+  }
+  session = undefined;
+  host = process.env.PUSHY_REGISTRY || defaultEndpoint;
+};
+
+async function query(url: string, options: fetch.RequestInit) {
+  const resp = await fetch(url, options);
+  const text = await resp.text();
+  let json;
+  try {
+    json = JSON.parse(text);
+  } catch (e) {
+    if (resp.statusText.includes('Unauthorized')) {
+      throw new Error('登录信息已过期，请使用 pushy login 命令重新登录');
+    } else {
+      throw new Error(`Server error: ${resp.statusText}`);
+    }
+  }
+
+  if (resp.status !== 200) {
+    throw Object.assign(
+      new Error(json.message || json.error || resp.statusText),
+      {
+        status: resp.status,
+      },
+    );
+  }
+  return json;
+}
+
+function queryWithoutBody(method: string) {
+  return function (api: string) {
+    return query(host + api, {
+      method,
+      headers: {
+        'User-Agent': userAgent,
+        'X-AccessToken': session ? session.token : '',
+      },
+    });
+  };
+}
+
+function queryWithBody(method: string) {
+  return function (api: string, body: Record<string, any>) {
+    return query(host + api, {
+      method,
+      headers: {
+        'User-Agent': userAgent,
+        'Content-Type': 'application/json',
+        'X-AccessToken': session ? session.token : '',
+      },
+      body: JSON.stringify(body),
+    });
+  };
+}
+
+export const get = queryWithoutBody('GET');
+export const post = queryWithBody('POST');
+export const put = queryWithBody('PUT');
+export const doDelete = queryWithBody('DELETE');
+
+export async function uploadFile(fn: string, key: string) {
+  const { url, backupUrl, formData, maxSize } = await post('/upload', {
+    ext: path.extname(fn),
+  });
+  let realUrl = url;
+  if (backupUrl) {
+    if (global.USE_ACC_OSS) {
+      realUrl = backupUrl;
+    } else {
+      const pingResult = await tcpPing({
+        address: url.replace('https://', ''),
+        attempts: 4,
+        timeout: 1000,
+      });
+      // console.log({pingResult});
+      if (isNaN(pingResult.avg) || pingResult.avg > 150) {
+        realUrl = backupUrl;
+      }
+    }
+    // console.log({realUrl});
+  }
+
+  const fileSize = fs.statSync(fn).size;
+  if (maxSize && fileSize > filesizeParser(maxSize)) {
+    throw new Error(
+      `此文件大小 ${(fileSize / 1048576).toFixed(
+        1,
+      )}m , 超出当前额度 ${maxSize} 。您可以考虑升级付费业务以提升此额度。详情请访问: ${pricingPageUrl}`,
+    );
+  }
+
+  const bar = new ProgressBar('  上传中 [:bar] :percent :etas', {
+    complete: '=',
+    incomplete: ' ',
+    total: fileSize,
+  });
+
+  const form = new FormData();
+
+  Object.entries(formData).forEach(([k, v]) => {
+    form.set(k, v);
+  });
+  const fileStream = fs.createReadStream(fn);
+  fileStream.on('data', function (data) {
+    bar.tick(data.length);
+  });
+
+  if (key) {
+    form.set('key', key);
+  }
+  form.set('file', fileStream);
+
+  const res = await fetch(url, {
+    method: 'POST',
+    body: form,
+  });
+
+  if (res.status > 299) {
+    throw new Error(`${res.status}: ${await res.text()}`);
+  }
+
+  // const body = await response.json();
+  return { hash: key };
+}