feat: add arg dns server

zsqk · Feb 27, 2023 · bcf050d · bcf050d
1 parent 6626906
commit bcf050d
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 3 deletions.
diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -13,5 +13,8 @@
   "editor.formatOnSave": true,
   "[sql]": {
     "editor.formatOnSave": false
+  },
+  "[json]": {
+    "editor.defaultFormatter": "denoland.vscode-deno"
   }
 }
diff --git a/README.md b/README.md
@@ -14,6 +14,9 @@ OPTIONS:
 -d, --domain
         Set domain
 
+    --dns
+        Set DNS server.
+
 -t, --token
         Set he.net auth token
 

diff --git a/deno.json b/deno.json
@@ -2,6 +2,7 @@
   "compilerOptions": {
     "strict": true
   },
+  "lock": false,
   "lint": {
     "files": {
       "include": [

diff --git a/src/cli.ts b/src/cli.ts
@@ -6,6 +6,7 @@ const args = parse<
     {
       d: string;
       domain: string;
+      dns: string;
       t: string;
       token: string;
       delaytime: number;
@@ -45,6 +46,7 @@ const timeout = args.timeout ?? 5000;
 const delay = Number(
   args.delaytime ?? Deno.env.get('USEDDNS_DELAY_TIME') ?? 5000,
 );
+const dnsServer = args.dns;
 
 if (!domain || !token) {
   throw new Error(
@@ -81,7 +83,7 @@ let lastFailInfo = '';
 const displayInfoLoopTime = 60000;
 
 setInterval(async () => {
-  const [err, res] = await checkIP(token, domain, { timeout });
+  const [err, res] = await checkIP(token, domain, { timeout, dnsServer });
   const d = new Date();
   if (err) {
     failTime += 1;

diff --git a/src/useddns.ts b/src/useddns.ts
@@ -30,11 +30,11 @@ export function getDNSIP(
 export async function checkIP(
   token: string,
   domain: string,
-  { timeout }: { timeout?: number } = {},
+  { timeout, dnsServer }: { timeout?: number; dnsServer?: string } = {},
 ): Promise<[err: Error] | [err: null, res: string]> {
   try {
     const currentIP = await getCurrentIP({ timeout });
-    const DNSIP = await getDNSIP(domain, { timeout });
+    const DNSIP = await getDNSIP(domain, { timeout, dns: dnsServer });
     if (!currentIP || !DNSIP) {
       throw new Error(
         'Did not get IP: ' + JSON.stringify({ currentIP, DNSIP }),