Philipinho
42 lines
1.3 KiB
TypeScript
42 lines
1.3 KiB
TypeScript
import { Injectable, NestMiddleware, NotFoundException } from '@nestjs/common';
|
|
import { FastifyRequest, FastifyReply } from 'fastify';
|
|
import { WorkspaceRepository } from '../core/workspace/repositories/workspace.repository';
|
|
import { EnvironmentService } from '../integrations/environment/environment.service';
|
|
|
|
@Injectable()
|
|
export class DomainMiddleware implements NestMiddleware {
|
|
constructor(
|
|
private workspaceRepository: WorkspaceRepository,
|
|
private environmentService: EnvironmentService,
|
|
) {}
|
|
async use(
|
|
req: FastifyRequest['raw'],
|
|
res: FastifyReply['raw'],
|
|
next: () => void,
|
|
) {
|
|
if (this.environmentService.isSelfHosted()) {
|
|
const workspace = await this.workspaceRepository.findFirst();
|
|
if (!workspace) {
|
|
throw new NotFoundException('Workspace not found');
|
|
}
|
|
|
|
(req as any).workspaceId = workspace.id;
|
|
} else if (this.environmentService.isCloud()) {
|
|
const header = req.headers.host;
|
|
const subdomain = header.split('.')[0];
|
|
|
|
const workspace = await this.workspaceRepository.findOneBy({
|
|
hostname: subdomain,
|
|
});
|
|
|
|
if (!workspace) {
|
|
throw new NotFoundException('Workspace not found');
|
|
}
|
|
|
|
(req as any).workspaceId = workspace.id;
|
|
}
|
|
|
|
next();
|
|
}
|
|
}
|