Feature: Add dns disable timer

public/locales/en/common.json

@@ -17,6 +17,7 @@
   "disabled": "Disabled",
   "enableAll": "Enable all",
   "disableAll": "Disable all",
+  "setTimer": "Set timer",
   "version": "Version",
   "changePosition": "Change position",
   "remove": "Remove",

public/locales/en/modules/dns-hole-controls.json

@@ -14,5 +14,12 @@
         "text": "There was a problem connecting to your DNS Hole(s). Please verify your configuration/integration(s)."
       }
     }
+  },
+  "durationModal": {
+    "title": "Set disable duration time",
+    "hours": "Hours",
+    "minutes": "Minutes",
+    "unlimited": "leave empty for unlimited",
+    "set": "Set"
   }
 }

src/server/api/routers/dns-hole/router.ts

@@ -14,6 +14,7 @@ export const dnsHoleRouter = createTRPCRouter({
     .input(
       z.object({
         action: z.enum(['enable', 'disable']),
+        duration: z.number(),
         configName: z.string(),
         appsToChange: z.optional(z.array(z.string())),
       })
@@ -32,12 +33,12 @@ export const dnsHoleRouter = createTRPCRouter({
       await Promise.all(
         applicableApps.map(async (app) => {
           if (app.integration?.type === 'pihole') {
-            await processPiHole(app, input.action === 'enable');
+            await processPiHole(app, input.action === 'enable', input.duration);
 
             return;
           }
 
-          await processAdGuard(app, input.action === 'enable');
+          await processAdGuard(app, input.action === 'enable', input.duration);
         })
       );
     }),
@@ -89,7 +90,7 @@ export const dnsHoleRouter = createTRPCRouter({
     }),
 });
 
-const processAdGuard = async (app: ConfigAppType, enable: boolean) => {
+const processAdGuard = async (app: ConfigAppType, enable: boolean, duration: number = 0) => {
   const adGuard = new AdGuard(
     app.url,
     findAppProperty(app, 'username'),
@@ -106,13 +107,13 @@ const processAdGuard = async (app: ConfigAppType, enable: boolean) => {
   }
 
   try {
-    await adGuard.disable();
+    await adGuard.disable(duration);
   } catch (error) {
     Consola.error((error as Error).message);
   }
 };
 
-const processPiHole = async (app: ConfigAppType, enable: boolean) => {
+const processPiHole = async (app: ConfigAppType, enable: boolean, duration: number = 0) => {
   const pihole = new PiHoleClient(app.url, findAppProperty(app, 'apiKey'));
 
   if (enable) {
@@ -125,7 +126,7 @@ const processPiHole = async (app: ConfigAppType, enable: boolean) => {
   }
 
   try {
-    await pihole.disable();
+    await pihole.disable(duration);
   } catch (error) {
     Consola.error((error as Error).message);
   }

src/tools/server/sdk/adGuard/adGuard.ts

@@ -59,8 +59,8 @@ export class AdGuard {
       .reduce((sum, filter) => filter.rules_count + sum, 0);
   }
 
-  async disable() {
-    await this.changeProtectionStatus(false);
+  async disable(duration: number) {
+    await this.changeProtectionStatus(false, duration);
   }
   async enable() {
     await this.changeProtectionStatus(true);
@@ -69,7 +69,7 @@ export class AdGuard {
   /**
    * Make a post request to the AdGuard API to change the protection status based on the value of newStatus
    * @param {boolean} newStatus - The new status of the protection
-   * @param {number} duration - Duration of a pause, in milliseconds. Enabled should be false.
+   * @param {number} duration - Duration of a pause, in seconds. Enabled should be false.
    * @returns {string} - The response from the AdGuard API
    */
   private async changeProtectionStatus(newStatus: boolean, duration = 0) {
@@ -78,7 +78,7 @@ export class AdGuard {
         `${this.baseHostName}/control/protection`,
         {
           enabled: newStatus,
-          duration,
+          duration: duration * 1000,
         },
         {
           headers: {

src/tools/server/sdk/pihole/piHole.ts

@@ -37,16 +37,19 @@ export class PiHoleClient {
     return response.status === 'enabled';
   }
 
-  async disable() {
-    const response = await this.sendStatusChangeRequest('disable');
+  async disable(duration: number) {
+    const response = await this.sendStatusChangeRequest('disable', duration);
     return response.status === 'disabled';
   }
 
   private async sendStatusChangeRequest(
-    action: 'enable' | 'disable'
+    action: 'enable' | 'disable',
+    duration = 0
   ): Promise<PiHoleApiStatusChangeResponse> {
     const response = await fetch(
-      `${this.baseHostName}/admin/api.php?${action}&auth=${this.apiToken}`
+      duration !== 0
+        ? `${this.baseHostName}/admin/api.php?${action}=${duration}&auth=${this.apiToken}`
+        : `${this.baseHostName}/admin/api.php?${action}&auth=${this.apiToken}`
     );
 
     if (response.status !== 200) {