✨ Add Tdarr integration and widget (#1882)
This commit is contained in:
Janik Brüll
@@ -22,6 +22,7 @@ import { smartHomeEntityStateRouter } from './routers/smart-home/entity-state';
|
||||
import { usenetRouter } from './routers/usenet/router';
|
||||
import { userRouter } from './routers/user';
|
||||
import { weatherRouter } from './routers/weather';
|
||||
import { tdarrRouter } from '~/server/api/routers/tdarr';
|
||||
|
||||
/**
|
||||
* This is the primary router for your server.
|
||||
@@ -51,6 +52,7 @@ export const rootRouter = createTRPCRouter({
|
||||
notebook: notebookRouter,
|
||||
smartHomeEntityState: smartHomeEntityStateRouter,
|
||||
healthMonitoring: healthMonitoringRouter,
|
||||
tdarr: tdarrRouter,
|
||||
});
|
||||
|
||||
// export type definition of API
|
||||
|
||||
356
src/server/api/routers/tdarr.ts
Normal file
356
src/server/api/routers/tdarr.ts
Normal file
@@ -0,0 +1,356 @@
|
||||
import { TRPCError } from '@trpc/server';
|
||||
import axios from 'axios';
|
||||
import { z } from 'zod';
|
||||
import { checkIntegrationsType } from '~/tools/client/app-properties';
|
||||
import { getConfig } from '~/tools/config/getConfig';
|
||||
import { ConfigAppType } from '~/types/app';
|
||||
|
||||
import { createTRPCRouter, publicProcedure } from '../trpc';
|
||||
import { TdarrQueue, TdarrStatistics, TdarrWorker } from '~/types/api/tdarr';
|
||||
|
||||
const getStatisticsSchema = z.object({
|
||||
totalFileCount: z.number(),
|
||||
totalTranscodeCount: z.number(),
|
||||
totalHealthCheckCount: z.number(),
|
||||
table3Count: z.number(),
|
||||
table6Count: z.number(),
|
||||
table1Count: z.number(),
|
||||
table4Count: z.number(),
|
||||
pies: z.array(
|
||||
z.tuple([
|
||||
z.string(), // Library Name
|
||||
z.string(), // Library ID
|
||||
z.number(), // File count
|
||||
z.number(), // Number of transcodes
|
||||
z.number(), // Space saved (in GB)
|
||||
z.number(), // Number of health checks
|
||||
z.array(
|
||||
z.object({
|
||||
// Transcode Status (Pie segments)
|
||||
name: z.string(),
|
||||
value: z.number(),
|
||||
})
|
||||
),
|
||||
z.array(
|
||||
z.object({
|
||||
// Health Status (Pie segments)
|
||||
name: z.string(),
|
||||
value: z.number(),
|
||||
})
|
||||
),
|
||||
z.array(
|
||||
z.object({
|
||||
// Video files - Codecs (Pie segments)
|
||||
name: z.string(),
|
||||
value: z.number(),
|
||||
})
|
||||
),
|
||||
z.array(
|
||||
z.object({
|
||||
// Video files - Containers (Pie segments)
|
||||
name: z.string(),
|
||||
value: z.number(),
|
||||
})
|
||||
),
|
||||
z.array(
|
||||
z.object({
|
||||
// Video files - Resolutions (Pie segments)
|
||||
name: z.string(),
|
||||
value: z.number(),
|
||||
})
|
||||
),
|
||||
z.array(
|
||||
z.object({
|
||||
// Audio files - Codecs (Pie segments)
|
||||
name: z.string(),
|
||||
value: z.number(),
|
||||
})
|
||||
),
|
||||
z.array(
|
||||
z.object({
|
||||
// Audio files - Containers (Pie segments)
|
||||
name: z.string(),
|
||||
value: z.number(),
|
||||
})
|
||||
),
|
||||
])
|
||||
),
|
||||
});
|
||||
|
||||
const getNodesResponseSchema = z.record(
|
||||
z.string(),
|
||||
z.object({
|
||||
_id: z.string(),
|
||||
nodeName: z.string(),
|
||||
nodePaused: z.boolean(),
|
||||
workers: z.record(
|
||||
z.string(),
|
||||
z.object({
|
||||
_id: z.string(),
|
||||
file: z.string(),
|
||||
fps: z.number(),
|
||||
percentage: z.number(),
|
||||
ETA: z.string(),
|
||||
job: z.object({
|
||||
type: z.string(),
|
||||
}),
|
||||
status: z.string(),
|
||||
lastPluginDetails: z
|
||||
.object({
|
||||
number: z.string().optional(),
|
||||
})
|
||||
.optional(),
|
||||
originalfileSizeInGbytes: z.number(),
|
||||
estSize: z.number().optional(),
|
||||
outputFileSizeInGbytes: z.number().optional(),
|
||||
workerType: z.string(),
|
||||
})
|
||||
),
|
||||
})
|
||||
);
|
||||
|
||||
const getStatusTableSchema = z.object({
|
||||
array: z.array(
|
||||
z.object({
|
||||
_id: z.string(),
|
||||
HealthCheck: z.string(),
|
||||
TranscodeDecisionMaker: z.string(),
|
||||
file: z.string(),
|
||||
file_size: z.number(),
|
||||
container: z.string(),
|
||||
video_codec_name: z.string(),
|
||||
video_resolution: z.string(),
|
||||
})
|
||||
),
|
||||
totalCount: z.number(),
|
||||
});
|
||||
|
||||
export const tdarrRouter = createTRPCRouter({
|
||||
statistics: publicProcedure
|
||||
.input(z.object({
|
||||
appId: z.string(),
|
||||
configName: z.string(),
|
||||
}))
|
||||
.query(async ({ input }): Promise<TdarrStatistics> => {
|
||||
const app = getTdarrApp(input.appId, input.configName);
|
||||
const appUrl = new URL('api/v2/cruddb', app.url);
|
||||
|
||||
const body = {
|
||||
data: {
|
||||
collection: 'StatisticsJSONDB',
|
||||
mode: 'getById',
|
||||
docID: 'statistics',
|
||||
},
|
||||
};
|
||||
|
||||
const res = await axios.post(appUrl.toString(), body);
|
||||
const data: z.infer<typeof getStatisticsSchema> = res.data;
|
||||
|
||||
const zodRes = getStatisticsSchema.safeParse(data);
|
||||
if (!zodRes.success) {
|
||||
/*
|
||||
* Tdarr's API is not documented and had to be reverse engineered. To account for mistakes in the type
|
||||
* definitions, we assume the best case scenario and log any parsing errors to aid in fixing the types.
|
||||
*/
|
||||
console.error(zodRes.error);
|
||||
}
|
||||
|
||||
return {
|
||||
totalFileCount: data.totalFileCount,
|
||||
totalTranscodeCount: data.totalTranscodeCount,
|
||||
totalHealthCheckCount: data.totalHealthCheckCount,
|
||||
failedTranscodeCount: data.table3Count,
|
||||
failedHealthCheckCount: data.table6Count,
|
||||
stagedTranscodeCount: data.table1Count,
|
||||
stagedHealthCheckCount: data.table4Count,
|
||||
pies: data.pies.map((pie) => ({
|
||||
libraryName: pie[0],
|
||||
libraryId: pie[1],
|
||||
totalFiles: pie[2],
|
||||
totalTranscodes: pie[3],
|
||||
savedSpace: pie[4] * 1_000_000_000, // file_size is in GB, convert to bytes,
|
||||
totalHealthChecks: pie[5],
|
||||
transcodeStatus: pie[6],
|
||||
healthCheckStatus: pie[7],
|
||||
videoCodecs: pie[8],
|
||||
videoContainers: pie[9],
|
||||
videoResolutions: pie[10],
|
||||
audioCodecs: pie[11],
|
||||
audioContainers: pie[12],
|
||||
})),
|
||||
};
|
||||
}),
|
||||
|
||||
workers: publicProcedure
|
||||
.input(z.object({
|
||||
appId: z.string(),
|
||||
configName: z.string(),
|
||||
})).query(async ({ input }): Promise<TdarrWorker[]> => {
|
||||
const app = getTdarrApp(input.appId, input.configName);
|
||||
const appUrl = new URL('api/v2/get-nodes', app.url);
|
||||
|
||||
const res = await axios.get(appUrl.toString());
|
||||
const data: z.infer<typeof getNodesResponseSchema> = res.data;
|
||||
|
||||
const zodRes = getNodesResponseSchema.safeParse(data);
|
||||
if (!zodRes.success) {
|
||||
/*
|
||||
* Tdarr's API is not documented and had to be reverse engineered. To account for mistakes in the type
|
||||
* definitions, we assume the best case scenario and log any parsing errors to aid in fixing the types.
|
||||
*/
|
||||
console.error(zodRes.error);
|
||||
}
|
||||
|
||||
const nodes = Object.values(data);
|
||||
const workers = nodes.flatMap((node) => {
|
||||
return Object.values(node.workers);
|
||||
});
|
||||
|
||||
return workers.map((worker) => ({
|
||||
id: worker._id,
|
||||
filePath: worker.file,
|
||||
fps: worker.fps,
|
||||
percentage: worker.percentage,
|
||||
ETA: worker.ETA,
|
||||
jobType: worker.job.type,
|
||||
status: worker.status,
|
||||
step: worker.lastPluginDetails?.number ?? '',
|
||||
originalSize: worker.originalfileSizeInGbytes * 1_000_000_000, // file_size is in GB, convert to bytes,
|
||||
estimatedSize: worker.estSize ? worker.estSize * 1_000_000_000 : null, // file_size is in GB, convert to bytes,
|
||||
outputSize: worker.outputFileSizeInGbytes ? worker.outputFileSizeInGbytes * 1_000_000_000 : null, // file_size is in GB, convert to bytes,
|
||||
}));
|
||||
}),
|
||||
|
||||
queue: publicProcedure
|
||||
.input(z.object({
|
||||
appId: z.string(),
|
||||
configName: z.string(),
|
||||
showHealthChecksInQueue: z.boolean(),
|
||||
pageSize: z.number(),
|
||||
page: z.number(),
|
||||
}))
|
||||
.query(async ({ input }): Promise<TdarrQueue> => {
|
||||
const app = getTdarrApp(input.appId, input.configName);
|
||||
|
||||
const appUrl = new URL('api/v2/client/status-tables', app.url);
|
||||
|
||||
const { page, pageSize, showHealthChecksInQueue } = input;
|
||||
|
||||
const firstItemIndex = page * pageSize;
|
||||
|
||||
const transcodeQueueBody = {
|
||||
data: {
|
||||
start: firstItemIndex,
|
||||
pageSize: pageSize,
|
||||
filters: [],
|
||||
sorts: [],
|
||||
opts: {
|
||||
table: 'table1',
|
||||
},
|
||||
},
|
||||
};
|
||||
|
||||
const transcodeQueueRes = await axios.post(appUrl.toString(), transcodeQueueBody);
|
||||
const transcodeQueueData: z.infer<typeof getStatusTableSchema> = transcodeQueueRes.data;
|
||||
|
||||
const transcodeQueueZodRes = getStatusTableSchema.safeParse(transcodeQueueData);
|
||||
if (!transcodeQueueZodRes.success) {
|
||||
/*
|
||||
* Tdarr's API is not documented and had to be reverse engineered. To account for mistakes in the type
|
||||
* definitions, we assume the best case scenario and log any parsing errors to aid in fixing the types.
|
||||
*/
|
||||
console.error(transcodeQueueZodRes.error);
|
||||
}
|
||||
|
||||
const transcodeQueueResult = {
|
||||
array: transcodeQueueData.array.map((item) => ({
|
||||
id: item._id,
|
||||
healthCheck: item.HealthCheck,
|
||||
transcode: item.TranscodeDecisionMaker,
|
||||
filePath: item.file,
|
||||
fileSize: item.file_size * 1_000_000, // file_size is in MB, convert to bytes
|
||||
container: item.container,
|
||||
codec: item.video_codec_name,
|
||||
resolution: item.video_resolution,
|
||||
type: 'transcode' as const,
|
||||
})),
|
||||
totalCount: transcodeQueueData.totalCount,
|
||||
startIndex: firstItemIndex,
|
||||
endIndex: firstItemIndex + transcodeQueueData.array.length - 1,
|
||||
};
|
||||
|
||||
if (!showHealthChecksInQueue) {
|
||||
return transcodeQueueResult;
|
||||
}
|
||||
|
||||
const healthCheckQueueBody = {
|
||||
data: {
|
||||
start: Math.max(firstItemIndex - transcodeQueueData.totalCount, 0),
|
||||
pageSize: pageSize,
|
||||
filters: [],
|
||||
sorts: [],
|
||||
opts: {
|
||||
table: 'table4',
|
||||
},
|
||||
},
|
||||
};
|
||||
|
||||
const healthCheckQueueRes = await axios.post(appUrl.toString(), healthCheckQueueBody);
|
||||
const healthCheckQueueData: z.infer<typeof getStatusTableSchema> = healthCheckQueueRes.data;
|
||||
|
||||
const healthCheckQueueZodRes = getStatusTableSchema.safeParse(healthCheckQueueData);
|
||||
if (!healthCheckQueueZodRes.success) {
|
||||
/*
|
||||
* Tdarr's API is not documented and had to be reverse engineered. To account for mistakes in the type
|
||||
* definitions, we assume the best case scenario and log any parsing errors to aid in fixing the types.
|
||||
*/
|
||||
console.error(healthCheckQueueZodRes.error);
|
||||
}
|
||||
|
||||
const healthCheckResultArray = healthCheckQueueData.array.map((item) => ({
|
||||
id: item._id,
|
||||
healthCheck: item.HealthCheck,
|
||||
transcode: item.TranscodeDecisionMaker,
|
||||
filePath: item.file,
|
||||
fileSize: item.file_size * 1_000_000, // file_size is in MB, convert to bytes
|
||||
container: item.container,
|
||||
codec: item.video_codec_name,
|
||||
resolution: item.video_resolution,
|
||||
type: 'health check' as const,
|
||||
}));
|
||||
|
||||
const combinedArray = [...transcodeQueueResult.array, ...healthCheckResultArray].slice(
|
||||
0,
|
||||
pageSize
|
||||
);
|
||||
|
||||
return {
|
||||
array: combinedArray,
|
||||
totalCount: transcodeQueueData.totalCount + healthCheckQueueData.totalCount,
|
||||
startIndex: firstItemIndex,
|
||||
endIndex: firstItemIndex + combinedArray.length - 1,
|
||||
};
|
||||
}),
|
||||
});
|
||||
|
||||
function getTdarrApp(appId: string, configName: string): ConfigAppType {
|
||||
const config = getConfig(configName);
|
||||
|
||||
const app = config.apps.find((x) => x.id === appId);
|
||||
|
||||
if (!app) {
|
||||
throw new TRPCError({
|
||||
code: 'BAD_REQUEST',
|
||||
message: `[Tdarr integration] App with ID "${appId}" could not be found.`,
|
||||
});
|
||||
}
|
||||
|
||||
if (!checkIntegrationsType(app.integration, ['tdarr'])) {
|
||||
throw new TRPCError({
|
||||
code: 'BAD_REQUEST',
|
||||
message: `[Tdarr integration] App with ID "${appId}" is not using the Tdarr integration.`,
|
||||
});
|
||||
}
|
||||
|
||||
return app;
|
||||
}
|
||||
Reference in New Issue
Block a user