feat: add Consumet as provider for stream URL
Summary: Test Plan:
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
import { OpenAPIHono, createRoute, z } from "@hono/zod-openapi";
|
||||
|
||||
import { readEnvVariable } from "~/libs/readEnvVariable";
|
||||
import type { Env } from "~/types/env";
|
||||
import {
|
||||
ErrorResponse,
|
||||
@@ -7,7 +8,6 @@ import {
|
||||
SuccessResponseSchema,
|
||||
} from "~/types/schema";
|
||||
|
||||
import { getSourcesFromAnify } from "./anify";
|
||||
import { FetchUrlResponse as FetchUrlResponseSchema } from "./responseType";
|
||||
|
||||
const FetchUrlRequest = z.object({ id: z.string(), provider: z.string() });
|
||||
@@ -38,6 +38,14 @@ const route = createRoute({
|
||||
},
|
||||
description: "Returns a stream URL",
|
||||
},
|
||||
400: {
|
||||
content: {
|
||||
"application/json": {
|
||||
schema: ErrorResponseSchema,
|
||||
},
|
||||
},
|
||||
description: "Unknown provider",
|
||||
},
|
||||
404: {
|
||||
content: {
|
||||
"application/json": {
|
||||
@@ -63,9 +71,32 @@ app.openapi(route, async (c) => {
|
||||
const aniListId = Number(c.req.param("aniListId"));
|
||||
const { provider, id } = await c.req.json<typeof FetchUrlRequest._type>();
|
||||
|
||||
const isAnifyEnabled = readEnvVariable(c.env, "ENABLE_ANIFY");
|
||||
if (provider === "consumet" || !isAnifyEnabled) {
|
||||
try {
|
||||
const result = await import("./consumet").then(
|
||||
({ getSourcesFromConsumet }) => getSourcesFromConsumet(id),
|
||||
);
|
||||
if (!result) {
|
||||
return c.json({ success: false }, { status: 404 });
|
||||
}
|
||||
|
||||
return c.json({
|
||||
success: true,
|
||||
result,
|
||||
});
|
||||
} catch (e) {
|
||||
console.error("Failed to fetch download URL from Consumet", e);
|
||||
|
||||
return c.json(ErrorResponse, { status: 500 });
|
||||
}
|
||||
}
|
||||
|
||||
if (provider === "anify") {
|
||||
try {
|
||||
const result = await getSourcesFromAnify(provider, id, aniListId);
|
||||
const result = await import("./anify").then(({ getSourcesFromAnify }) =>
|
||||
getSourcesFromAnify(provider, id, aniListId),
|
||||
);
|
||||
if (!result) {
|
||||
return c.json({ success: false }, { status: 404 });
|
||||
}
|
||||
@@ -80,6 +111,8 @@ app.openapi(route, async (c) => {
|
||||
return c.json(ErrorResponse, { status: 500 });
|
||||
}
|
||||
}
|
||||
|
||||
return c.json(ErrorResponse, { status: 400 });
|
||||
});
|
||||
|
||||
export default app;
|
||||
|
||||
Reference in New Issue
Block a user