update

agent/task.ts

@@ -0,0 +1,92 @@
+import { app } from './app.ts';
+import { createStorage } from "unstorage";
+import fsDriver from "unstorage/drivers/fs";
+
+const storage = createStorage({
+  driver: fsDriver({
+    base: process.cwd() + '/storage/ddns-agent'
+  }),
+});
+
+type CloudflareConfig = {
+  // Cloudflare 访问地址
+  website: string;
+  domain: string;
+  zone_id: string;
+  // 更新IPv4记录的ID
+  record_id4: string;
+  // 更新IPv6记录的ID
+  record_id6: string;
+  api_token: string;
+  ipv6: string;
+  ipv4: string;
+  flag: number; // 0: 不更新, 1: 仅IPv4, 2: 仅IPv6, 3: IPv4和IPv6
+  time: string; // 上次更新时间戳
+}
+app.route({
+  path: 'ip',
+  key: 'task',
+  description: `执行IP更新任务
+1. 读取配置文件cloudflare.json
+2. 根据flag决定更新IPv4和/或IPv6地址
+3. 获取当前公网IP地址
+4. 如果IP地址有变化则调用 router: cf/update 更新DNS记录
+5. 保存最新的IP地址和更新时间戳到配置文件
+`,
+}).define(async (ctx) => {
+  const config = await storage.getItem<CloudflareConfig>('cloudflare.json');
+  if (!config) {
+    ctx.throw?.('未找到配置');
+  }
+  const now = Date.now();
+  const date = new Date(now);
+  const updateIp = async (isV4 = true) => {
+    const res = await app.call({ path: 'ip', key: isV4 ? 'v4' : 'v6' });
+    if (res.code !== 200) {
+      ctx.throw(res.message);
+    }
+    const newIp = res.body.ip as string;
+    const oldIp = isV4 ? config.ipv4 : config.ipv6;
+    if (newIp !== oldIp) {
+      // IP地址有变化，更新DNS记录
+      await app.call({ path: 'cf', key: 'update' }, {
+        zone_id: config.zone_id,
+        record_id: isV4 ? config.record_id4 : config.record_id6,
+        domain: config.domain,
+        new_ip: newIp,
+        api_token: config.api_token,
+        type: isV4 ? 'A' : 'AAAA',
+      });
+      // 更新配置文件中的IP地址
+      if (isV4) {
+        config.ipv4 = newIp;
+      } else {
+        config.ipv6 = newIp;
+      }
+      config.time = date.toLocaleString();
+      await storage.setItem('cloudflare.json', config);
+      console.log(date.toLocaleString() + ` 更新 ${isV4 ? 'IPv4' : 'IPv6'} 地址为: ${newIp}`);
+    } else {
+      console.log(date.toLocaleString() + ` ${isV4 ? 'IPv4' : 'IPv6'} 地址未改变: ${newIp}`);
+    }
+  }
+  if (config.flag === 1) {
+    await updateIp(true);
+  }
+  if (config.flag === 2) {
+    await updateIp(false);
+  }
+  if (config.flag === 3) {
+    await updateIp(true);
+    await updateIp(false);
+  }
+  ctx.body = { message: '任务完成' };
+}).addTo(app);
+
+export const main = () => {
+  app.call({
+    path: 'ip',
+    key: 'task',
+  })
+}
+