feat: implement ApiTokenGuard for authentication and apply it to relevant controllers

This commit is contained in:
Boris D
2025-10-06 11:06:02 +03:00
parent 1e84297e84
commit 39c40013b2
10 changed files with 85 additions and 39 deletions

View File

@ -0,0 +1,67 @@
import {
Body,
Controller,
Headers,
Inject,
Param,
Post,
Res,
UseGuards,
} from "@nestjs/common";
import { Response } from "express";
import { QueryHandlerService } from "./query.handler.service";
import { ApiTokenGuard } from "src/api/guards/api-token.guard";
import { QueryExecuterService } from "../executer/query.executer.service";
@Controller("query")
@UseGuards(ApiTokenGuard)
export class QueryController {
constructor(
@Inject(QueryHandlerService)
private readonly queryHandlerService: QueryHandlerService,
@Inject(QueryExecuterService)
private readonly queryExecuterService: QueryExecuterService
) {}
@Post("create")
async createQuery(
@Body() queryData: { projectToken: string; source: string }
) {
return this.queryHandlerService.createQuery(queryData);
}
@Post("update/:id")
async updateQuery(
@Body() updateData: Partial<{ source: string }>,
@Inject("id") id: string
) {
return this.queryHandlerService.updateQuery(id, updateData);
}
@Post("/run/:token")
async runQuery(
@Param("token") token: string,
@Body() query: Record<string, any>,
@Headers() headers: Record<string, any>,
@Res() res: Response
) {
const queryResult = await this.queryExecuterService.runQuery(
token,
query,
headers
);
res.status(queryResult.statusCode);
if (queryResult.statusCode === 302 && queryResult.headers["Location"]) {
res.redirect(queryResult.headers["Location"]);
return;
}
for (const [key, value] of Object.entries(queryResult.headers)) {
res.setHeader(key, value);
}
res.send(queryResult.response);
}
}