Skip to content

Commit 6dd87e4

Browse files
committed
refactor: guard ProvidersController via runMode
1 parent c201d11 commit 6dd87e4

3 files changed

Lines changed: 8 additions & 4 deletions

File tree

apps/backend/src/main.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ import helmet from "helmet";
66
import { NativeLogger } from "nestjs-pino";
77
import { toStructuredError } from "./common/logging.js";
88
import { createPinoExitLogger } from "./common/pino.config.js";
9+
import { parseRunMode } from "./config/env.parsers.js";
910

1011
/** Standalone pino logger used for pre-bootstrap and process-level error paths. */
1112
const exitLogger = createPinoExitLogger().child({ context: "Main" });
@@ -30,7 +31,7 @@ function logErrorAndExit(event: string, message: string, error: unknown): void {
3031
}
3132

3233
async function bootstrap() {
33-
const runMode = (process.env.DEALBOT_RUN_MODE || "both").toLowerCase();
34+
const runMode = parseRunMode(process.env);
3435
const isWorkerOnly = runMode === "worker";
3536
const rootModule = isWorkerOnly
3637
? (await import("./worker.module.js")).WorkerModule

apps/backend/src/providers/providers.module.ts

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,16 @@
11
import { Module } from "@nestjs/common";
22
import { TypeOrmModule } from "@nestjs/typeorm";
3+
import { parseRunMode } from "../config/env.parsers.js";
34
import { StorageProvider } from "../database/entities/storage-provider.entity.js";
45
import { ProvidersController } from "./providers.controller.js";
56
import { ProvidersService } from "./providers.service.js";
67
import { StorageProviderRepository } from "./repositories/storage-provider.repository.js";
78

9+
const isWorkerOnly = parseRunMode(process.env) === "worker";
10+
811
@Module({
912
imports: [TypeOrmModule.forFeature([StorageProvider])],
10-
controllers: [ProvidersController],
13+
controllers: isWorkerOnly ? [] : [ProvidersController],
1114
providers: [ProvidersService, StorageProviderRepository],
1215
exports: [ProvidersService, StorageProviderRepository],
1316
})

apps/backend/src/pull-check/pull-check.module.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import { Module } from "@nestjs/common";
22
import { ThrottlerModule } from "@nestjs/throttler";
33
import { TypeOrmModule } from "@nestjs/typeorm";
4+
import { parseRunMode } from "../config/env.parsers.js";
45
import { DatabaseModule } from "../database/database.module.js";
56
import { PullPiece } from "../database/entities/pull-piece.entity.js";
67
import { DataSourceModule } from "../dataSource/dataSource.module.js";
@@ -12,8 +13,7 @@ import { PieceSourceController } from "./pull-piece.controller.js";
1213
import { PullPieceRepository } from "./pull-piece.repository.js";
1314
import { PullPieceStreamTracker } from "./pull-piece-stream-tracker.service.js";
1415

15-
const runMode = process.env.DEALBOT_RUN_MODE?.toLowerCase() || "both";
16-
const isWorkerOnly = runMode === "worker";
16+
const isWorkerOnly = parseRunMode(process.env) === "worker";
1717

1818
@Module({
1919
imports: [

0 commit comments

Comments
 (0)