@rongyan/wxpost-server 0.1.0

package/src/token.js

@@ -0,0 +1,93 @@
+'use strict';
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const https = require('https');
+
+const STABLE_TOKEN_URL = 'https://api.weixin.qq.com/cgi-bin/stable_token';
+const CACHE_PATH = path.join(os.homedir(), '.@rongyan', 'tokens.json');
+const CACHE_TMP = CACHE_PATH + '.tmp';
+
+// 提前 5 分钟视为过期，避免边界问题
+const EXPIRE_BUFFER_MS = 5 * 60 * 1000;
+const REQUEST_TIMEOUT_MS = 10 * 1000;
+
+function readCache() {
+  try {
+    const raw = fs.readFileSync(CACHE_PATH, 'utf-8');
+    return JSON.parse(raw);
+  } catch {
+    return {};
+  }
+}
+
+function writeCache(cache) {
+  // 原子写：先写临时文件再 rename，防止 crash 导致缓存损坏；权限 0600 保护 token
+  fs.writeFileSync(CACHE_TMP, JSON.stringify(cache, null, 2) + '\n', { mode: 0o600, encoding: 'utf-8' });
+  fs.renameSync(CACHE_TMP, CACHE_PATH);
+}
+
+function post(url, body) {
+  return new Promise((resolve, reject) => {
+    const data = JSON.stringify(body);
+    const req = https.request(url, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        'Content-Length': Buffer.byteLength(data),
+      },
+    }, (res) => {
+      let raw = '';
+      res.on('data', (chunk) => { raw += chunk; });
+      res.on('end', () => {
+        try {
+          resolve(JSON.parse(raw));
+        } catch (e) {
+          reject(new Error(`响应解析失败: ${raw}`));
+        }
+      });
+    });
+    req.setTimeout(REQUEST_TIMEOUT_MS, () => req.destroy(new Error('获取 access_token 超时')));
+    req.on('error', reject);
+    req.write(data);
+    req.end();
+  });
+}
+
+/**
+ * 获取稳定版 access_token，自动读缓存、到期后刷新。
+ * @param {string} appId
+ * @param {string} appSecret
+ * @param {boolean} [forceRefresh=false]
+ */
+async function getStableToken(appId, appSecret, forceRefresh = false) {
+  const cache = readCache();
+  const cached = cache[appId];
+  const now = Date.now();
+
+  if (!forceRefresh && cached && cached.expiresAt - EXPIRE_BUFFER_MS > now) {
+    return cached.accessToken;
+  }
+
+  const result = await post(STABLE_TOKEN_URL, {
+    grant_type: 'client_credential',
+    appid: appId,
+    secret: appSecret,
+    force_refresh: forceRefresh,
+  });
+
+  if (result.errcode) {
+    throw new Error(`获取 access_token 失败 [${result.errcode}]: ${result.errmsg}`);
+  }
+
+  cache[appId] = {
+    accessToken: result.access_token,
+    expiresAt: now + result.expires_in * 1000,
+  };
+  writeCache(cache);
+
+  return result.access_token;
+}
+
+module.exports = { getStableToken };

package/src/upload-image.js

@@ -0,0 +1,75 @@
+'use strict';
+
+const https = require('https');
+
+const UPLOAD_URL = 'https://api.weixin.qq.com/cgi-bin/media/uploadimg';
+const MAX_SIZE = 1 * 1024 * 1024; // 1MB，微信要求严格小于
+const REQUEST_TIMEOUT_MS = 10 * 1000;
+
+function detectFormat(buffer) {
+  if (buffer.length < 4) return null;
+  if (buffer[0] === 0xFF && buffer[1] === 0xD8 && buffer[2] === 0xFF) return 'jpeg';
+  if (buffer[0] === 0x89 && buffer[1] === 0x50 && buffer[2] === 0x4E && buffer[3] === 0x47) return 'png';
+  return null;
+}
+
+function clientError(msg) {
+  const err = new Error(msg);
+  err.statusCode = 400;
+  return err;
+}
+
+function uploadImageToWx({ accessToken, buffer }) {
+  const format = detectFormat(buffer);
+  if (!format) {
+    return Promise.reject(clientError('不支持的图片格式，内容图仅支持 JPG/PNG'));
+  }
+  if (buffer.length >= MAX_SIZE) {
+    return Promise.reject(clientError(`图片大小 ${(buffer.length / 1024).toFixed(1)}KB 须小于 1MB`));
+  }
+
+  const mimetype = format === 'png' ? 'image/png' : 'image/jpeg';
+  const ext = format === 'png' ? '.png' : '.jpg';
+  const boundary = `----WxUploadBoundary${Date.now()}`;
+
+  const head = Buffer.from(
+    `--${boundary}\r\n` +
+    `Content-Disposition: form-data; name="media"; filename="upload${ext}"\r\n` +
+    `Content-Type: ${mimetype}\r\n\r\n`
+  );
+  const tail = Buffer.from(`\r\n--${boundary}--\r\n`);
+  const body = Buffer.concat([head, buffer, tail]);
+
+  return new Promise((resolve, reject) => {
+    const req = https.request(
+      `${UPLOAD_URL}?access_token=${accessToken}`,
+      {
+        method: 'POST',
+        headers: {
+          'Content-Type': `multipart/form-data; boundary=${boundary}`,
+          'Content-Length': body.length,
+        },
+      },
+      (res) => {
+        let raw = '';
+        res.on('data', (chunk) => { raw += chunk; });
+        res.on('end', () => {
+          let result;
+          try {
+            result = JSON.parse(raw);
+          } catch {
+            return reject(new Error(`微信接口响应解析失败: ${raw}`));
+          }
+          if (result.errcode) return reject(new Error(`微信上传图片失败 [${result.errcode}]: ${result.errmsg}`));
+          resolve(result.url);
+        });
+      }
+    );
+    req.setTimeout(REQUEST_TIMEOUT_MS, () => req.destroy(new Error('上传内容图超时')));
+    req.on('error', reject);
+    req.write(body);
+    req.end();
+  });
+}
+
+module.exports = { uploadImageToWx, MAX_SIZE };