mirror of
https://github.com/movie-web/backend.git
synced 2025-09-13 10:23:25 +00:00
Compare commits
38 Commits
Author | SHA1 | Date | |
---|---|---|---|
|
befa173445 | ||
|
1c1d70fa05 | ||
|
6b83fcd158 | ||
|
f8870b5929 | ||
|
727a9cdd43 | ||
|
206844e758 | ||
|
b2e0841d0b | ||
|
886df2ffb4 | ||
|
6ea07ea685 | ||
|
1010eaf71e | ||
|
ebe00ecbd6 | ||
|
3bb427401f | ||
|
b2598e3d82 | ||
|
bf55be2978 | ||
|
8f52dad296 | ||
|
9a27b0b0b4 | ||
|
46baac466d | ||
|
26316c7755 | ||
|
d5851b522a | ||
|
7145d6c246 | ||
|
961ac7eeca | ||
|
59ab9b48bd | ||
|
07ecd445f9 | ||
|
5ebecd1476 | ||
|
f4e9a96666 | ||
|
60dda8ac57 | ||
|
d84cdc4239 | ||
|
c2cba27e68 | ||
|
c3259156ac | ||
|
9ef12d1c0f | ||
|
e173003f55 | ||
|
9f90ba7da2 | ||
|
05bf651939 | ||
|
cf0125755c | ||
|
4129b80828 | ||
|
e5c3cde51b | ||
|
4bf0285d06 | ||
|
53d5ca1461 |
@@ -12,6 +12,8 @@ services:
|
||||
POSTGRES_PASSWORD: postgres
|
||||
volumes:
|
||||
- 'postgres_data:/var/lib/postgresql/data'
|
||||
redis:
|
||||
image: redis
|
||||
|
||||
# custom services
|
||||
backend:
|
||||
|
10
package.json
10
package.json
@@ -1,6 +1,6 @@
|
||||
{
|
||||
"name": "backend",
|
||||
"version": "1.1.4",
|
||||
"version": "1.3.0",
|
||||
"private": true,
|
||||
"homepage": "https://github.com/movie-web/backend",
|
||||
"engines": {
|
||||
@@ -8,16 +8,16 @@
|
||||
},
|
||||
"scripts": {
|
||||
"dev": "nodemon -r tsconfig-paths/register src/main.ts",
|
||||
"build": "npm run build:pre && npm run build:compile",
|
||||
"build": "pnpm run build:pre && pnpm run build:compile",
|
||||
"start": "node dist/main.js",
|
||||
"lint": "eslint --ext .ts,.js,.json,.tsx src/",
|
||||
"lint:fix": "eslint --fix --ext .ts,.js,.json,.tsx src/",
|
||||
"build:pre": "rimraf dist/",
|
||||
"build:compile": "tsc && tsc-alias",
|
||||
"preinstall": "npx -y only-allow pnpm",
|
||||
"migration:create": "npx -y mikro-orm migration:create",
|
||||
"migration:up": "npx -y mikro-orm migration:up",
|
||||
"migration:down": "npx -y mikro-orm migration:down"
|
||||
"migration:create": "pnpm exec mikro-orm migration:create",
|
||||
"migration:up": "pnpm exec mikro-orm migration:up",
|
||||
"migration:down": "pnpm exec mikro-orm migration:down"
|
||||
},
|
||||
"mikro-orm": {
|
||||
"useTsNode": true,
|
||||
|
@@ -4,4 +4,8 @@ export const dockerFragment: FragmentSchema = {
|
||||
postgres: {
|
||||
connection: 'postgres://postgres:postgres@postgres:5432/postgres',
|
||||
},
|
||||
ratelimits: {
|
||||
enabled: true,
|
||||
redisUrl: 'redis://redis:6379',
|
||||
},
|
||||
};
|
||||
|
@@ -1,10 +1,19 @@
|
||||
import { devFragment } from '@/config/fragments/dev';
|
||||
import { dockerFragment } from '@/config/fragments/docker';
|
||||
import { createConfigLoader } from 'neat-config';
|
||||
import { z } from 'zod';
|
||||
|
||||
const fragments = {
|
||||
dev: devFragment,
|
||||
dockerdev: dockerFragment,
|
||||
};
|
||||
|
||||
export const ormConfigSchema = z.object({
|
||||
postgres: z.object({
|
||||
// connection URL for postgres database
|
||||
connection: z.string(),
|
||||
// whether to use SSL for the connection
|
||||
ssl: z.coerce.boolean().default(false),
|
||||
}),
|
||||
});
|
||||
|
||||
@@ -15,6 +24,8 @@ export const ormConf = createConfigLoader()
|
||||
prefix: 'MWB_',
|
||||
})
|
||||
.addFromFile('config.json')
|
||||
.setFragmentKey('usePresets')
|
||||
.addConfigFragments(fragments)
|
||||
.addZodSchema(ormConfigSchema)
|
||||
.freeze()
|
||||
.load();
|
||||
|
@@ -16,6 +16,9 @@ export const configSchema = z.object({
|
||||
// should it trust reverse proxy headers? (for ip gathering)
|
||||
trustProxy: z.coerce.boolean().default(false),
|
||||
|
||||
// should it trust cloudflare headers? (for ip gathering, cloudflare has priority)
|
||||
trustCloudflare: z.coerce.boolean().default(false),
|
||||
|
||||
// prefix for where the instance is run on. for example set it to /backend if you're hosting it on example.com/backend
|
||||
// if this is set, do not apply url rewriting before proxing
|
||||
basePath: z.string().default('/'),
|
||||
@@ -45,6 +48,9 @@ export const configSchema = z.object({
|
||||
// Enable debug logging for MikroORM - Outputs queries and entity management logs
|
||||
// Do NOT use in production, leaks all sensitive data
|
||||
debugLogging: z.coerce.boolean().default(false),
|
||||
|
||||
// Enable SSL for the postgres connection
|
||||
ssl: z.coerce.boolean().default(false),
|
||||
}),
|
||||
crypto: z.object({
|
||||
// session secret. used for signing session tokens
|
||||
|
@@ -268,143 +268,6 @@
|
||||
"checks": [],
|
||||
"foreignKeys": {}
|
||||
},
|
||||
{
|
||||
"columns": {
|
||||
"id": {
|
||||
"name": "id",
|
||||
"type": "uuid",
|
||||
"unsigned": false,
|
||||
"autoincrement": false,
|
||||
"primary": false,
|
||||
"nullable": false,
|
||||
"mappedType": "uuid"
|
||||
},
|
||||
"tmdb_id": {
|
||||
"name": "tmdb_id",
|
||||
"type": "varchar(255)",
|
||||
"unsigned": false,
|
||||
"autoincrement": false,
|
||||
"primary": false,
|
||||
"nullable": false,
|
||||
"mappedType": "string"
|
||||
},
|
||||
"type": {
|
||||
"name": "type",
|
||||
"type": "varchar(255)",
|
||||
"unsigned": false,
|
||||
"autoincrement": false,
|
||||
"primary": false,
|
||||
"nullable": false,
|
||||
"mappedType": "string"
|
||||
},
|
||||
"title": {
|
||||
"name": "title",
|
||||
"type": "varchar(255)",
|
||||
"unsigned": false,
|
||||
"autoincrement": false,
|
||||
"primary": false,
|
||||
"nullable": false,
|
||||
"mappedType": "string"
|
||||
},
|
||||
"season_id": {
|
||||
"name": "season_id",
|
||||
"type": "varchar(255)",
|
||||
"unsigned": false,
|
||||
"autoincrement": false,
|
||||
"primary": false,
|
||||
"nullable": true,
|
||||
"mappedType": "string"
|
||||
},
|
||||
"episode_id": {
|
||||
"name": "episode_id",
|
||||
"type": "varchar(255)",
|
||||
"unsigned": false,
|
||||
"autoincrement": false,
|
||||
"primary": false,
|
||||
"nullable": true,
|
||||
"mappedType": "string"
|
||||
},
|
||||
"created_at": {
|
||||
"name": "created_at",
|
||||
"type": "timestamptz(0)",
|
||||
"unsigned": false,
|
||||
"autoincrement": false,
|
||||
"primary": false,
|
||||
"nullable": false,
|
||||
"length": 0,
|
||||
"mappedType": "datetime"
|
||||
},
|
||||
"status": {
|
||||
"name": "status",
|
||||
"type": "varchar(255)",
|
||||
"unsigned": false,
|
||||
"autoincrement": false,
|
||||
"primary": false,
|
||||
"nullable": false,
|
||||
"mappedType": "string"
|
||||
},
|
||||
"provider_id": {
|
||||
"name": "provider_id",
|
||||
"type": "varchar(255)",
|
||||
"unsigned": false,
|
||||
"autoincrement": false,
|
||||
"primary": false,
|
||||
"nullable": false,
|
||||
"mappedType": "string"
|
||||
},
|
||||
"embed_id": {
|
||||
"name": "embed_id",
|
||||
"type": "varchar(255)",
|
||||
"unsigned": false,
|
||||
"autoincrement": false,
|
||||
"primary": false,
|
||||
"nullable": true,
|
||||
"mappedType": "string"
|
||||
},
|
||||
"error_message": {
|
||||
"name": "error_message",
|
||||
"type": "text",
|
||||
"unsigned": false,
|
||||
"autoincrement": false,
|
||||
"primary": false,
|
||||
"nullable": true,
|
||||
"mappedType": "text"
|
||||
},
|
||||
"full_error": {
|
||||
"name": "full_error",
|
||||
"type": "text",
|
||||
"unsigned": false,
|
||||
"autoincrement": false,
|
||||
"primary": false,
|
||||
"nullable": true,
|
||||
"mappedType": "text"
|
||||
},
|
||||
"hostname": {
|
||||
"name": "hostname",
|
||||
"type": "varchar(255)",
|
||||
"unsigned": false,
|
||||
"autoincrement": false,
|
||||
"primary": false,
|
||||
"nullable": false,
|
||||
"mappedType": "string"
|
||||
}
|
||||
},
|
||||
"name": "provider_metrics",
|
||||
"schema": "public",
|
||||
"indexes": [
|
||||
{
|
||||
"keyName": "provider_metrics_pkey",
|
||||
"columnNames": [
|
||||
"id"
|
||||
],
|
||||
"composite": false,
|
||||
"primary": true,
|
||||
"unique": true
|
||||
}
|
||||
],
|
||||
"checks": [],
|
||||
"foreignKeys": {}
|
||||
},
|
||||
{
|
||||
"columns": {
|
||||
"id": {
|
||||
@@ -620,6 +483,15 @@
|
||||
"primary": false,
|
||||
"nullable": true,
|
||||
"mappedType": "string"
|
||||
},
|
||||
"proxy_urls": {
|
||||
"name": "proxy_urls",
|
||||
"type": "text[]",
|
||||
"unsigned": false,
|
||||
"autoincrement": false,
|
||||
"primary": false,
|
||||
"nullable": true,
|
||||
"mappedType": "array"
|
||||
}
|
||||
},
|
||||
"name": "user_settings",
|
13
src/db/migrations/Migration20231221185725.ts
Normal file
13
src/db/migrations/Migration20231221185725.ts
Normal file
@@ -0,0 +1,13 @@
|
||||
import { Migration } from '@mikro-orm/migrations';
|
||||
|
||||
export class Migration20231221185725 extends Migration {
|
||||
|
||||
async up(): Promise<void> {
|
||||
this.addSql('drop table if exists "provider_metrics" cascade;');
|
||||
}
|
||||
|
||||
async down(): Promise<void> {
|
||||
this.addSql('create table "provider_metrics" ("id" uuid not null default null, "tmdb_id" varchar not null default null, "type" varchar not null default null, "title" varchar not null default null, "season_id" varchar null default null, "episode_id" varchar null default null, "created_at" timestamptz not null default null, "status" varchar not null default null, "provider_id" varchar not null default null, "embed_id" varchar null default null, "error_message" text null default null, "full_error" text null default null, "hostname" varchar not null default null, constraint "provider_metrics_pkey" primary key ("id"));');
|
||||
}
|
||||
|
||||
}
|
13
src/db/migrations/Migration20231229214215.ts
Normal file
13
src/db/migrations/Migration20231229214215.ts
Normal file
@@ -0,0 +1,13 @@
|
||||
import { Migration } from '@mikro-orm/migrations';
|
||||
|
||||
export class Migration20231229214215 extends Migration {
|
||||
|
||||
async up(): Promise<void> {
|
||||
this.addSql('alter table "user_settings" add column "proxy_urls" text[] null;');
|
||||
}
|
||||
|
||||
async down(): Promise<void> {
|
||||
this.addSql('alter table "user_settings" drop column "proxy_urls";');
|
||||
}
|
||||
|
||||
}
|
@@ -1,51 +0,0 @@
|
||||
import { Entity, PrimaryKey, Property } from '@mikro-orm/core';
|
||||
import { randomUUID } from 'crypto';
|
||||
|
||||
export const status = {
|
||||
failed: 'failed',
|
||||
notfound: 'notfound',
|
||||
success: 'success',
|
||||
} as const;
|
||||
type Status = keyof typeof status;
|
||||
|
||||
@Entity({ tableName: 'provider_metrics' })
|
||||
export class ProviderMetric {
|
||||
@PrimaryKey({ name: 'id', type: 'uuid' })
|
||||
id: string = randomUUID();
|
||||
|
||||
@Property({ name: 'tmdb_id' })
|
||||
tmdbId!: string;
|
||||
|
||||
@Property({ name: 'type' })
|
||||
type!: string;
|
||||
|
||||
@Property({ name: 'title' })
|
||||
title!: string;
|
||||
|
||||
@Property({ name: 'season_id', nullable: true })
|
||||
seasonId?: string;
|
||||
|
||||
@Property({ name: 'episode_id', nullable: true })
|
||||
episodeId?: string;
|
||||
|
||||
@Property({ name: 'created_at', type: 'date' })
|
||||
createdAt = new Date();
|
||||
|
||||
@Property({ name: 'status' })
|
||||
status!: Status;
|
||||
|
||||
@Property({ name: 'provider_id' })
|
||||
providerId!: string;
|
||||
|
||||
@Property({ name: 'embed_id', nullable: true })
|
||||
embedId?: string;
|
||||
|
||||
@Property({ name: 'error_message', nullable: true, type: 'text' })
|
||||
errorMessage?: string;
|
||||
|
||||
@Property({ name: 'full_error', nullable: true, type: 'text' })
|
||||
fullError?: string;
|
||||
|
||||
@Property({ name: 'hostname' })
|
||||
hostname!: string;
|
||||
}
|
@@ -1,4 +1,4 @@
|
||||
import { Entity, PrimaryKey, Property } from '@mikro-orm/core';
|
||||
import { ArrayType, Entity, PrimaryKey, Property } from '@mikro-orm/core';
|
||||
|
||||
@Entity({ tableName: 'user_settings' })
|
||||
export class UserSettings {
|
||||
@@ -13,6 +13,9 @@ export class UserSettings {
|
||||
|
||||
@Property({ name: 'default_subtitle_language', nullable: true })
|
||||
defaultSubtitleLanguage?: string | null;
|
||||
|
||||
@Property({ name: 'proxy_urls', type: ArrayType, nullable: true })
|
||||
proxyUrls?: string[] | null;
|
||||
}
|
||||
|
||||
export interface UserSettingsDTO {
|
||||
@@ -20,6 +23,7 @@ export interface UserSettingsDTO {
|
||||
applicationTheme?: string | null;
|
||||
applicationLanguage?: string | null;
|
||||
defaultSubtitleLanguage?: string | null;
|
||||
proxyUrls?: string[] | null;
|
||||
}
|
||||
|
||||
export function formatUserSettings(
|
||||
@@ -30,5 +34,6 @@ export function formatUserSettings(
|
||||
applicationTheme: userSettings.applicationTheme,
|
||||
applicationLanguage: userSettings.applicationLanguage,
|
||||
defaultSubtitleLanguage: userSettings.defaultSubtitleLanguage,
|
||||
proxyUrls: userSettings.proxyUrls,
|
||||
};
|
||||
}
|
||||
|
@@ -1,4 +1,4 @@
|
||||
import { ormConf } from '@/config/orm';
|
||||
import { makeOrmConfig } from '@/modules/mikro/orm';
|
||||
|
||||
export default makeOrmConfig(ormConf.postgres.connection);
|
||||
export default makeOrmConfig(ormConf.postgres.connection, ormConf.postgres.ssl);
|
||||
|
@@ -1,11 +1,9 @@
|
||||
import { challengeCodeJob } from '@/modules/jobs/list/challengeCode';
|
||||
import { sessionExpiryJob } from '@/modules/jobs/list/sessionExpiry';
|
||||
import { userDeletionJob } from '@/modules/jobs/list/userDeletion';
|
||||
import { providerMetricCleanupJob } from '@/modules/jobs/list/providerMetricCleanup';
|
||||
|
||||
export async function setupJobs() {
|
||||
challengeCodeJob.start();
|
||||
sessionExpiryJob.start();
|
||||
userDeletionJob.start();
|
||||
providerMetricCleanupJob.start();
|
||||
}
|
||||
|
@@ -1,27 +0,0 @@
|
||||
import { ProviderMetric } from '@/db/models/ProviderMetrics';
|
||||
import { job } from '@/modules/jobs/job';
|
||||
import ms from 'ms';
|
||||
|
||||
// every day at 12:00:00
|
||||
export const providerMetricCleanupJob = job(
|
||||
'provider-metric-cleanup',
|
||||
'0 12 * * *',
|
||||
async ({ em, log }) => {
|
||||
const now = new Date();
|
||||
const thirtyDaysAgo = new Date(now.getTime() - ms('30d'));
|
||||
|
||||
const deletedMetrics = await em
|
||||
.createQueryBuilder(ProviderMetric)
|
||||
.delete()
|
||||
.where({
|
||||
createdAt: {
|
||||
$lt: thirtyDaysAgo,
|
||||
},
|
||||
})
|
||||
.execute<{ affectedRows: number }>('run');
|
||||
|
||||
log.info(
|
||||
`Removed ${deletedMetrics.affectedRows} metrics that were older than 30 days`,
|
||||
);
|
||||
},
|
||||
);
|
@@ -9,17 +9,10 @@ const log = scopedLogger('metrics');
|
||||
|
||||
export type Metrics = {
|
||||
user: Counter<'namespace'>;
|
||||
providerMetrics: Counter<
|
||||
| 'title'
|
||||
| 'tmdb_id'
|
||||
| 'season_id'
|
||||
| 'episode_id'
|
||||
| 'status'
|
||||
| 'type'
|
||||
| 'provider_id'
|
||||
| 'embed_id'
|
||||
| 'hostname'
|
||||
>;
|
||||
captchaSolves: Counter<'success'>;
|
||||
providerHostnames: Counter<'hostname'>;
|
||||
providerStatuses: Counter<'provider_id' | 'status'>;
|
||||
watchMetrics: Counter<'title' | 'tmdb_full_id' | 'provider_id' | 'success'>;
|
||||
};
|
||||
|
||||
let metrics: null | Metrics = null;
|
||||
@@ -42,31 +35,39 @@ export async function setupMetrics(app: FastifyInstance) {
|
||||
|
||||
metrics = {
|
||||
user: new Counter({
|
||||
name: 'user_count',
|
||||
help: 'user_help',
|
||||
name: 'mw_user_count',
|
||||
help: 'mw_user_help',
|
||||
labelNames: ['namespace'],
|
||||
}),
|
||||
providerMetrics: new Counter({
|
||||
name: 'provider_metrics',
|
||||
help: 'provider_metrics',
|
||||
labelNames: [
|
||||
'episode_id',
|
||||
'provider_id',
|
||||
'season_id',
|
||||
'status',
|
||||
'title',
|
||||
'tmdb_id',
|
||||
'type',
|
||||
'embed_id',
|
||||
'hostname',
|
||||
],
|
||||
captchaSolves: new Counter({
|
||||
name: 'mw_captcha_solves',
|
||||
help: 'mw_captcha_solves',
|
||||
labelNames: ['success'],
|
||||
}),
|
||||
providerHostnames: new Counter({
|
||||
name: 'mw_provider_hostname_count',
|
||||
help: 'mw_provider_hostname_count',
|
||||
labelNames: ['hostname'],
|
||||
}),
|
||||
providerStatuses: new Counter({
|
||||
name: 'mw_provider_status_count',
|
||||
help: 'mw_provider_status_count',
|
||||
labelNames: ['provider_id', 'status'],
|
||||
}),
|
||||
watchMetrics: new Counter({
|
||||
name: 'mw_media_watch_count',
|
||||
help: 'mw_media_watch_count',
|
||||
labelNames: ['title', 'tmdb_full_id', 'provider_id', 'success'],
|
||||
}),
|
||||
};
|
||||
|
||||
const promClient = app.metrics.client;
|
||||
|
||||
promClient.register.registerMetric(metrics.user);
|
||||
promClient.register.registerMetric(metrics.providerMetrics);
|
||||
promClient.register.registerMetric(metrics.providerHostnames);
|
||||
promClient.register.registerMetric(metrics.providerStatuses);
|
||||
promClient.register.registerMetric(metrics.watchMetrics);
|
||||
promClient.register.registerMetric(metrics.captchaSolves);
|
||||
|
||||
const orm = getORM();
|
||||
const em = orm.em.fork();
|
||||
|
@@ -18,6 +18,7 @@ export async function setupMikroORM() {
|
||||
conf.postgres.connection,
|
||||
conf.postgres.debugLogging,
|
||||
(msg) => log.info(msg),
|
||||
conf.postgres.ssl,
|
||||
);
|
||||
|
||||
if (conf.postgres.syncSchema) {
|
||||
|
@@ -2,7 +2,10 @@ import { Options } from '@mikro-orm/core';
|
||||
import { MikroORM, PostgreSqlDriver } from '@mikro-orm/postgresql';
|
||||
import path from 'path';
|
||||
|
||||
export function makeOrmConfig(url: string): Options<PostgreSqlDriver> {
|
||||
export function makeOrmConfig(
|
||||
url: string,
|
||||
ssl: boolean,
|
||||
): Options<PostgreSqlDriver> {
|
||||
return {
|
||||
type: 'postgresql',
|
||||
clientUrl: url,
|
||||
@@ -13,6 +16,11 @@ export function makeOrmConfig(url: string): Options<PostgreSqlDriver> {
|
||||
pathTs: './migrations',
|
||||
path: './migrations',
|
||||
},
|
||||
driverOptions: {
|
||||
connection: {
|
||||
ssl,
|
||||
},
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
@@ -20,9 +28,10 @@ export async function createORM(
|
||||
url: string,
|
||||
debug: boolean,
|
||||
log: (msg: string) => void,
|
||||
ssl: boolean,
|
||||
) {
|
||||
return await MikroORM.init<PostgreSqlDriver>({
|
||||
...makeOrmConfig(url),
|
||||
...makeOrmConfig(url, ssl),
|
||||
logger: log,
|
||||
debug,
|
||||
});
|
||||
|
@@ -2,6 +2,7 @@ import Redis from 'ioredis';
|
||||
import RateLimiter from 'async-ratelimiter';
|
||||
import ms from 'ms';
|
||||
import { StatusError } from '@/services/error';
|
||||
import { IpReq, getIp } from '@/services/ip';
|
||||
|
||||
export interface LimiterOptions {
|
||||
redis: Redis;
|
||||
@@ -26,8 +27,8 @@ export class Limiter {
|
||||
this.redis = ops.redis;
|
||||
}
|
||||
|
||||
async bump(req: { ip: string }, ops: BucketOptions) {
|
||||
const ip = req.ip;
|
||||
async bump(req: IpReq, ops: BucketOptions) {
|
||||
const ip = getIp(req);
|
||||
if (!this.buckets[ops.id]) {
|
||||
this.buckets[ops.id] = {
|
||||
limiter: new RateLimiter({
|
||||
@@ -54,7 +55,7 @@ export class Limiter {
|
||||
};
|
||||
}
|
||||
|
||||
async assertAndBump(req: { ip: string }, ops: BucketOptions) {
|
||||
async assertAndBump(req: IpReq, ops: BucketOptions) {
|
||||
const { hasBeenLimited } = await this.bump(req, ops);
|
||||
if (hasBeenLimited) {
|
||||
throw new StatusError('Ratelimited', 429);
|
||||
|
@@ -1,8 +1,8 @@
|
||||
import { handle } from '@/services/handler';
|
||||
import { makeRouter } from '@/services/router';
|
||||
import { z } from 'zod';
|
||||
import { ProviderMetric, status } from '@/db/models/ProviderMetrics';
|
||||
import { getMetrics } from '@/modules/metrics';
|
||||
import { status } from '@/routes/statuses';
|
||||
|
||||
const metricsProviderSchema = z.object({
|
||||
tmdbId: z.string(),
|
||||
@@ -29,7 +29,7 @@ export const metricsRouter = makeRouter((app) => {
|
||||
body: metricsProviderInputSchema,
|
||||
},
|
||||
},
|
||||
handle(async ({ em, body, req, limiter }) => {
|
||||
handle(async ({ body, req, limiter }) => {
|
||||
await limiter?.assertAndBump(req, {
|
||||
id: 'provider_metrics',
|
||||
max: 300,
|
||||
@@ -37,43 +37,59 @@ export const metricsRouter = makeRouter((app) => {
|
||||
window: '30m',
|
||||
});
|
||||
|
||||
const hostname = req.headers.origin?.slice(0, 255) ?? 'unknown origin';
|
||||
|
||||
const entities = body.items.map((v) => {
|
||||
const errorMessage = v.errorMessage?.slice(0, 200);
|
||||
const truncatedFullError = v.fullError?.slice(0, 2000);
|
||||
|
||||
const metric = new ProviderMetric();
|
||||
em.assign(metric, {
|
||||
providerId: v.providerId,
|
||||
embedId: v.embedId,
|
||||
fullError: truncatedFullError,
|
||||
errorMessage: errorMessage,
|
||||
episodeId: v.episodeId,
|
||||
seasonId: v.seasonId,
|
||||
status: v.status,
|
||||
title: v.title,
|
||||
tmdbId: v.tmdbId,
|
||||
type: v.type,
|
||||
hostname,
|
||||
});
|
||||
return metric;
|
||||
const hostname = req.headers.origin?.slice(0, 255) ?? '<UNKNOWN>';
|
||||
getMetrics().providerHostnames.inc({
|
||||
hostname,
|
||||
});
|
||||
|
||||
entities.forEach((entity) => {
|
||||
getMetrics().providerMetrics.inc({
|
||||
episode_id: entity.episodeId,
|
||||
provider_id: entity.providerId,
|
||||
season_id: entity.seasonId,
|
||||
status: entity.status,
|
||||
title: entity.title,
|
||||
tmdb_id: entity.tmdbId,
|
||||
type: entity.type,
|
||||
hostname,
|
||||
body.items.forEach((item) => {
|
||||
getMetrics().providerStatuses.inc({
|
||||
provider_id: item.embedId ?? item.providerId,
|
||||
status: item.status,
|
||||
});
|
||||
});
|
||||
|
||||
await em.persistAndFlush(entities);
|
||||
const itemList = [...body.items];
|
||||
itemList.reverse();
|
||||
const lastSuccessfulItem = body.items.find(
|
||||
(v) => v.status === status.success,
|
||||
);
|
||||
const lastItem = itemList[0];
|
||||
|
||||
if (lastItem) {
|
||||
getMetrics().watchMetrics.inc({
|
||||
tmdb_full_id: lastItem.type + '-' + lastItem.tmdbId,
|
||||
provider_id: lastSuccessfulItem?.providerId ?? lastItem.providerId,
|
||||
title: lastItem.title,
|
||||
success: (!!lastSuccessfulItem).toString(),
|
||||
});
|
||||
}
|
||||
|
||||
return true;
|
||||
}),
|
||||
);
|
||||
|
||||
app.post(
|
||||
'/metrics/captcha',
|
||||
{
|
||||
schema: {
|
||||
body: z.object({
|
||||
success: z.boolean(),
|
||||
}),
|
||||
},
|
||||
},
|
||||
handle(async ({ body, req, limiter }) => {
|
||||
await limiter?.assertAndBump(req, {
|
||||
id: 'captcha_solves',
|
||||
max: 300,
|
||||
inc: 1,
|
||||
window: '30m',
|
||||
});
|
||||
|
||||
getMetrics().captchaSolves.inc({
|
||||
success: body.success.toString(),
|
||||
});
|
||||
|
||||
return true;
|
||||
}),
|
||||
);
|
||||
|
6
src/routes/statuses.ts
Normal file
6
src/routes/statuses.ts
Normal file
@@ -0,0 +1,6 @@
|
||||
export const status = {
|
||||
failed: 'failed',
|
||||
notfound: 'notfound',
|
||||
success: 'success',
|
||||
} as const;
|
||||
export type Status = keyof typeof status;
|
@@ -6,7 +6,6 @@ import {
|
||||
import { StatusError } from '@/services/error';
|
||||
import { handle } from '@/services/handler';
|
||||
import { makeRouter } from '@/services/router';
|
||||
import { randomUUID } from 'crypto';
|
||||
import { z } from 'zod';
|
||||
|
||||
const bookmarkDataSchema = z.object({
|
||||
|
@@ -6,6 +6,7 @@ import {
|
||||
import { StatusError } from '@/services/error';
|
||||
import { handle } from '@/services/handler';
|
||||
import { makeRouter } from '@/services/router';
|
||||
import { FilterQuery } from '@mikro-orm/core';
|
||||
import { randomUUID } from 'crypto';
|
||||
import { z } from 'zod';
|
||||
|
||||
@@ -18,6 +19,7 @@ const progressItemSchema = z.object({
|
||||
episodeId: z.string().optional(),
|
||||
seasonNumber: z.number().optional(),
|
||||
episodeNumber: z.number().optional(),
|
||||
updatedAt: z.string().datetime({ offset: true }).optional(),
|
||||
});
|
||||
|
||||
export const userProgressRouter = makeRouter((app) => {
|
||||
@@ -100,7 +102,9 @@ export const userProgressRouter = makeRouter((app) => {
|
||||
if (newItemIndex > -1) {
|
||||
const newItem = newItems[newItemIndex];
|
||||
if (existingItem.watched < newItem.watched) {
|
||||
existingItem.updatedAt = new Date();
|
||||
existingItem.updatedAt = defaultAndCoerceDateTime(
|
||||
newItem.updatedAt,
|
||||
);
|
||||
existingItem.watched = newItem.watched;
|
||||
}
|
||||
itemsUpserted.push(existingItem);
|
||||
@@ -123,7 +127,7 @@ export const userProgressRouter = makeRouter((app) => {
|
||||
tmdbId: newItem.tmdbId,
|
||||
userId: params.uid,
|
||||
watched: newItem.watched,
|
||||
updatedAt: new Date(),
|
||||
updatedAt: defaultAndCoerceDateTime(newItem.updatedAt),
|
||||
});
|
||||
}
|
||||
|
||||
@@ -161,22 +165,28 @@ export const userProgressRouter = makeRouter((app) => {
|
||||
if (auth.user.id !== params.uid)
|
||||
throw new StatusError('Cannot modify user other than yourself', 403);
|
||||
|
||||
const progressItem = await em.findOne(ProgressItem, {
|
||||
const query: FilterQuery<ProgressItem> = {
|
||||
userId: params.uid,
|
||||
tmdbId: params.tmdbid,
|
||||
episodeId: body.episodeId,
|
||||
seasonId: body.seasonId,
|
||||
});
|
||||
if (!progressItem) {
|
||||
};
|
||||
if (body.seasonId) query.seasonId = body.seasonId;
|
||||
if (body.episodeId) query.episodeId = body.episodeId;
|
||||
const progressItems = await em.find(ProgressItem, query);
|
||||
|
||||
if (progressItems.length === 0) {
|
||||
return {
|
||||
count: 0,
|
||||
tmdbId: params.tmdbid,
|
||||
episodeId: body.episodeId,
|
||||
seasonId: body.seasonId,
|
||||
};
|
||||
}
|
||||
|
||||
await em.removeAndFlush(progressItem);
|
||||
progressItems.forEach((v) => em.remove(v));
|
||||
await em.flush();
|
||||
|
||||
return {
|
||||
count: progressItems.length,
|
||||
tmdbId: params.tmdbid,
|
||||
episodeId: body.episodeId,
|
||||
seasonId: body.seasonId,
|
||||
@@ -207,3 +217,14 @@ export const userProgressRouter = makeRouter((app) => {
|
||||
}),
|
||||
);
|
||||
});
|
||||
|
||||
// 13th July 2021 - movie-web epoch
|
||||
const minEpoch = 1626134400000;
|
||||
|
||||
function defaultAndCoerceDateTime(dateTime: string | undefined) {
|
||||
const epoch = dateTime ? new Date(dateTime).getTime() : Date.now();
|
||||
|
||||
const clampedEpoch = Math.max(minEpoch, Math.min(epoch, Date.now()));
|
||||
|
||||
return new Date(clampedEpoch);
|
||||
}
|
||||
|
@@ -41,6 +41,7 @@ export const userSettingsRouter = makeRouter((app) => {
|
||||
applicationLanguage: z.string().nullable().optional(),
|
||||
applicationTheme: z.string().nullable().optional(),
|
||||
defaultSubtitleLanguage: z.string().nullable().optional(),
|
||||
proxyUrls: z.string().array().nullable().optional(),
|
||||
}),
|
||||
},
|
||||
},
|
||||
@@ -64,6 +65,7 @@ export const userSettingsRouter = makeRouter((app) => {
|
||||
settings.defaultSubtitleLanguage = body.defaultSubtitleLanguage;
|
||||
if (body.applicationTheme !== undefined)
|
||||
settings.applicationTheme = body.applicationTheme;
|
||||
if (body.proxyUrls !== undefined) settings.proxyUrls = body.proxyUrls;
|
||||
|
||||
await em.persistAndFlush(settings);
|
||||
return formatUserSettings(settings);
|
||||
|
27
src/services/ip.ts
Normal file
27
src/services/ip.ts
Normal file
@@ -0,0 +1,27 @@
|
||||
import { conf } from '@/config';
|
||||
import { IncomingHttpHeaders } from 'http';
|
||||
|
||||
export type IpReq = {
|
||||
ip: string;
|
||||
headers: IncomingHttpHeaders;
|
||||
};
|
||||
|
||||
const trustCloudflare = conf.server.trustCloudflare;
|
||||
|
||||
function getSingleHeader(
|
||||
headers: IncomingHttpHeaders,
|
||||
key: string,
|
||||
): string | undefined {
|
||||
const header = headers[key];
|
||||
if (Array.isArray(header)) return header[0];
|
||||
return header;
|
||||
}
|
||||
|
||||
export function getIp(req: IpReq) {
|
||||
const cfIp = getSingleHeader(req.headers, 'cf-connecting-ip');
|
||||
if (trustCloudflare && cfIp) {
|
||||
return cfIp;
|
||||
}
|
||||
|
||||
return req.ip;
|
||||
}
|
Reference in New Issue
Block a user