mirror of
https://github.com/ajnart/homarr.git
synced 2025-11-10 15:35:55 +01:00
🏗️ Migrate usenet history to tRPC
This commit is contained in:
@@ -1,20 +1,17 @@
|
|||||||
import { useMutation, useQuery } from '@tanstack/react-query';
|
import { useMutation, useQuery } from '@tanstack/react-query';
|
||||||
import axios from 'axios';
|
import axios from 'axios';
|
||||||
import { Results } from 'sabnzbd-api';
|
import { Results } from 'sabnzbd-api';
|
||||||
|
import { useConfigContext } from '~/config/provider';
|
||||||
|
import { api } from '~/utils/api';
|
||||||
|
import { UsenetInfoRequestParams, UsenetInfoResponse } from '../../../pages/api/modules/usenet';
|
||||||
|
import type { UsenetHistoryRequestParams } from '../../../pages/api/modules/usenet/history';
|
||||||
|
import { UsenetPauseRequestParams } from '../../../pages/api/modules/usenet/pause';
|
||||||
import type {
|
import type {
|
||||||
UsenetQueueRequestParams,
|
UsenetQueueRequestParams,
|
||||||
UsenetQueueResponse,
|
UsenetQueueResponse,
|
||||||
} from '../../../pages/api/modules/usenet/queue';
|
} from '../../../pages/api/modules/usenet/queue';
|
||||||
import type {
|
|
||||||
UsenetHistoryRequestParams,
|
|
||||||
UsenetHistoryResponse,
|
|
||||||
} from '../../../pages/api/modules/usenet/history';
|
|
||||||
import { UsenetInfoRequestParams, UsenetInfoResponse } from '../../../pages/api/modules/usenet';
|
|
||||||
import { UsenetPauseRequestParams } from '../../../pages/api/modules/usenet/pause';
|
|
||||||
import { queryClient } from '../../../tools/server/configurations/tanstack/queryClient.tool';
|
|
||||||
import { UsenetResumeRequestParams } from '../../../pages/api/modules/usenet/resume';
|
import { UsenetResumeRequestParams } from '../../../pages/api/modules/usenet/resume';
|
||||||
import { api } from '~/utils/api';
|
import { queryClient } from '../../../tools/server/configurations/tanstack/queryClient.tool';
|
||||||
import { useConfigContext } from '~/config/provider';
|
|
||||||
|
|
||||||
const POLLING_INTERVAL = 2000;
|
const POLLING_INTERVAL = 2000;
|
||||||
|
|
||||||
@@ -51,21 +48,20 @@ export const useGetUsenetDownloads = (params: UsenetQueueRequestParams) =>
|
|||||||
}
|
}
|
||||||
);
|
);
|
||||||
|
|
||||||
export const useGetUsenetHistory = (params: UsenetHistoryRequestParams) =>
|
export const useGetUsenetHistory = (params: UsenetHistoryRequestParams) => {
|
||||||
useQuery(
|
const { name: configName } = useConfigContext();
|
||||||
['usenetHistory', ...Object.values(params)],
|
return api.usenet.history.useQuery(
|
||||||
async () =>
|
{
|
||||||
(
|
configName: configName!,
|
||||||
await axios.get<UsenetHistoryResponse>('/api/modules/usenet/history', {
|
...params,
|
||||||
params,
|
},
|
||||||
})
|
|
||||||
).data,
|
|
||||||
{
|
{
|
||||||
refetchInterval: POLLING_INTERVAL,
|
refetchInterval: POLLING_INTERVAL,
|
||||||
keepPreviousData: true,
|
keepPreviousData: true,
|
||||||
retry: 2,
|
retry: 2,
|
||||||
}
|
}
|
||||||
);
|
);
|
||||||
|
};
|
||||||
|
|
||||||
export const usePauseUsenetQueue = (params: UsenetPauseRequestParams) =>
|
export const usePauseUsenetQueue = (params: UsenetPauseRequestParams) =>
|
||||||
useMutation(
|
useMutation(
|
||||||
|
|||||||
@@ -2,10 +2,11 @@ import dayjs from 'dayjs';
|
|||||||
import { Client } from 'sabnzbd-api';
|
import { Client } from 'sabnzbd-api';
|
||||||
import { z } from 'zod';
|
import { z } from 'zod';
|
||||||
import { TRPCError } from '@trpc/server';
|
import { TRPCError } from '@trpc/server';
|
||||||
import { NzbgetStatus } from '~/server/api/routers/usenet/nzbget/types';
|
import { NzbgetHistoryItem, NzbgetStatus } from '~/server/api/routers/usenet/nzbget/types';
|
||||||
import { getConfig } from '~/tools/config/getConfig';
|
import { getConfig } from '~/tools/config/getConfig';
|
||||||
import { createTRPCRouter, publicProcedure } from '../../trpc';
|
import { createTRPCRouter, publicProcedure } from '../../trpc';
|
||||||
import { NzbgetClient } from './nzbget/nzbget-client';
|
import { NzbgetClient } from './nzbget/nzbget-client';
|
||||||
|
import { UsenetHistoryItem } from '~/widgets/useNet/types';
|
||||||
|
|
||||||
export const usenetRouter = createTRPCRouter({
|
export const usenetRouter = createTRPCRouter({
|
||||||
info: publicProcedure
|
info: publicProcedure
|
||||||
@@ -91,6 +92,84 @@ export const usenetRouter = createTRPCRouter({
|
|||||||
eta: eta.asSeconds(),
|
eta: eta.asSeconds(),
|
||||||
};
|
};
|
||||||
}),
|
}),
|
||||||
|
history: publicProcedure
|
||||||
|
.input(
|
||||||
|
z.object({
|
||||||
|
configName: z.string(),
|
||||||
|
appId: z.string(),
|
||||||
|
limit: z.number(),
|
||||||
|
offset: z.number(),
|
||||||
|
})
|
||||||
|
)
|
||||||
|
.query(async ({ input }) => {
|
||||||
|
const config = getConfig(input.configName);
|
||||||
|
|
||||||
|
const app = config.apps.find((x) => x.id === input.appId);
|
||||||
|
|
||||||
|
if (!app || (app.integration?.type !== 'nzbGet' && app.integration?.type !== 'sabnzbd')) {
|
||||||
|
throw new Error(`App with ID "${input.appId}" could not be found.`);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (app.integration?.type === 'nzbGet') {
|
||||||
|
const url = new URL(app.url);
|
||||||
|
const options = {
|
||||||
|
host: url.hostname,
|
||||||
|
port: url.port || (url.protocol === 'https:' ? '443' : '80'),
|
||||||
|
login: app.integration.properties.find((x) => x.field === 'username')?.value ?? undefined,
|
||||||
|
hash: app.integration.properties.find((x) => x.field === 'password')?.value ?? undefined,
|
||||||
|
};
|
||||||
|
|
||||||
|
const nzbGet = NzbgetClient(options);
|
||||||
|
|
||||||
|
const nzbgetHistory: NzbgetHistoryItem[] = await new Promise((resolve, reject) => {
|
||||||
|
nzbGet.history(false, (err: any, result: NzbgetHistoryItem[]) => {
|
||||||
|
if (!err) {
|
||||||
|
resolve(result);
|
||||||
|
} else {
|
||||||
|
reject(err);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!nzbgetHistory) {
|
||||||
|
throw new Error('Error while getting NZBGet history');
|
||||||
|
}
|
||||||
|
|
||||||
|
const nzbgetItems: UsenetHistoryItem[] = nzbgetHistory.map((item: NzbgetHistoryItem) => ({
|
||||||
|
id: item.NZBID.toString(),
|
||||||
|
name: item.Name,
|
||||||
|
// Convert from MB to bytes
|
||||||
|
size: item.DownloadedSizeMB * 1000000,
|
||||||
|
time: item.DownloadTimeSec,
|
||||||
|
}));
|
||||||
|
|
||||||
|
return {
|
||||||
|
items: nzbgetItems,
|
||||||
|
total: nzbgetItems.length,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
const { origin } = new URL(app.url);
|
||||||
|
|
||||||
|
const apiKey = app.integration.properties.find((x) => x.field === 'apiKey')?.value;
|
||||||
|
if (!apiKey) {
|
||||||
|
throw new Error(`API Key for app "${app.name}" is missing`);
|
||||||
|
}
|
||||||
|
|
||||||
|
const history = await new Client(origin, apiKey).history(input.offset, input.limit);
|
||||||
|
|
||||||
|
const items: UsenetHistoryItem[] = history.slots.map((slot) => ({
|
||||||
|
id: slot.nzo_id,
|
||||||
|
name: slot.name,
|
||||||
|
size: slot.bytes,
|
||||||
|
time: slot.download_time,
|
||||||
|
}));
|
||||||
|
|
||||||
|
return {
|
||||||
|
items,
|
||||||
|
total: history.noofslots,
|
||||||
|
};
|
||||||
|
}),
|
||||||
});
|
});
|
||||||
|
|
||||||
export interface UsenetInfoResponse {
|
export interface UsenetInfoResponse {
|
||||||
|
|||||||
Reference in New Issue
Block a user