jazzymc/homarr
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

feat: add dns disable timer (#2029)

Browse Source
This commit is contained in:
Yossi Hillali
2024-05-23 19:18:06 +03:00
committed by GitHub
parent cb2b28ca4b
commit afdca50bd2
8 changed files with 273 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/server/api/routers/dns-hole/router.ts
View File

@@ -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);
}