FEAT: add tenant creation api

Author: bharathkeyvalue

src/authentication/authKey.guard.ts

@@ -0,0 +1,20 @@
+import { Injectable, CanActivate, ExecutionContext } from '@nestjs/common';
+import { ConfigService } from '@nestjs/config';
+import { GqlExecutionContext } from '@nestjs/graphql';
+
+@Injectable()
+export class AuthKeyGuard implements CanActivate {
+  constructor(private configService: ConfigService) {}
+
+  canActivate(context: ExecutionContext): boolean {
+    const ctx = GqlExecutionContext.create(context).getContext();
+    if (ctx) {
+      const authKeyInHeader = ctx.headers['x-api-key'];
+      if (authKeyInHeader) {
+        const secretKey = this.configService.get('AUTH_KEY') as string;
+        return secretKey === authKeyInHeader;
+      }
+    }
+    return false;
+  }
+}

src/authorization/authorization.module.ts

@@ -51,6 +51,11 @@ import { UserService } from './service/user.service';
 import { UserServiceInterface } from './service/user.service.interface';
 import { UserCacheService } from './service/usercache.service';
 import { UserCacheServiceInterface } from './service/usercache.service.interface';
+import Tenant from './entity/tenant.entity';
+import { TenantResolver } from './resolver/tenant.resolver';
+import TenantService from './service/tenant.service';
+import { TenantRepository } from './repository/tenant.repository';
+import { TenantServiceInterface } from './service/tenant.service.interface';
 
 @Module({
   imports: [
@@ -66,6 +71,7 @@ import { UserCacheServiceInterface } from './service/usercache.service.interface
       Role,
       GroupRole,
       RolePermission,
+      Tenant,
     ]),
     RedisCacheModule,
   ],
@@ -91,6 +97,8 @@ import { UserCacheServiceInterface } from './service/usercache.service.interface
     UserGroupRepository,
     EntityPermissionRepository,
     LoggerService,
+    TenantResolver,
+    TenantRepository,
     {
       provide: EntityServiceInterface,
       useClass: EntityService,
@@ -127,6 +135,10 @@ import { UserCacheServiceInterface } from './service/usercache.service.interface
       provide: UserCacheServiceInterface,
       useClass: UserCacheService,
     },
+    {
+      provide: TenantServiceInterface,
+      useClass: TenantService,
+    },
   ],
   exports: [
     {

src/authorization/entity/tenant.entity.ts

@@ -1,14 +1,16 @@
-import { Column, Entity, Index, PrimaryGeneratedColumn } from 'typeorm';
+import { Column, Entity, PrimaryGeneratedColumn } from 'typeorm';
 import BaseEntity from './base.entity';
 
 @Entity()
-@Index('tenant_name_unique_idx', { synchronize: false })
 class Tenant extends BaseEntity {
   @PrimaryGeneratedColumn('uuid')
   public id!: string;
 
   @Column()
   public name!: string;
+
+  @Column({ unique: true })
+  public domain!: string;
 }
 
 export default Tenant;

src/authorization/exception/tenant.exception.ts

@@ -0,0 +1,7 @@
+import { NotFoundException } from '@nestjs/common';
+
+export class TenantNotFoundException extends NotFoundException {
+  constructor(tenantDomain: string) {
+    super(`Tenant with domain ${tenantDomain} not found`);
+  }
+}

src/authorization/graphql/tenant.graphql

@@ -0,0 +1,14 @@
+input NewTenantInput {
+  name: String!
+  domain: String!
+}
+
+type Tenant {
+  id: ID!
+  name: String!
+  domain: String!
+}
+
+type Mutation {
+  createTenant(input: NewTenantInput!): Tenant
+}

src/authorization/repository/tenant.repository.ts

@@ -0,0 +1,16 @@
+import { Injectable } from '@nestjs/common';
+import { DataSource } from 'typeorm';
+
+import { BaseRepository } from './base.repository';
+import Tenant from '../entity/tenant.entity';
+
+@Injectable()
+export class TenantRepository extends BaseRepository<Tenant> {
+  constructor(private dataSource: DataSource) {
+    super(Tenant, dataSource);
+  }
+
+  async getTenantByDomain(domain: string) {
+    return this.findOneBy({ domain });
+  }
+}

src/authorization/resolver/tenant.resolver.ts

@@ -0,0 +1,22 @@
+import { Args, Mutation, Resolver } from '@nestjs/graphql';
+import { NewTenantInput } from '../../schema/graphql.schema';
+import Tenant from '../entity/tenant.entity';
+import { Inject, UseGuards } from '@nestjs/common';
+import { AuthKeyGuard } from '../../authentication/authKey.guard';
+import { TenantServiceInterface } from '../service/tenant.service.interface';
+
+@Resolver('Tenant')
+export class TenantResolver {
+  constructor(
+    @Inject(TenantServiceInterface)
+    private tenantService: TenantServiceInterface,
+  ) {}
+
+  @UseGuards(AuthKeyGuard)
+  @Mutation()
+  async createTenant(
+    @Args('input') tenantInput: NewTenantInput,
+  ): Promise<Tenant> {
+    return this.tenantService.createTenant(tenantInput);
+  }
+}

src/authorization/service/tenant.service.interface.ts

@@ -0,0 +1,10 @@
+import { NewTenantInput } from 'src/schema/graphql.schema';
+import Tenant from '../entity/tenant.entity';
+
+export interface TenantServiceInterface {
+  getTenantByDomain(domain: string): Promise<Tenant>;
+
+  createTenant(tenant: NewTenantInput): Promise<Tenant>;
+}
+
+export const TenantServiceInterface = Symbol('TenantServiceInterface');

src/authorization/service/tenant.service.ts

@@ -0,0 +1,23 @@
+import { Injectable } from '@nestjs/common';
+
+import Tenant from '../entity/tenant.entity';
+import { TenantNotFoundException } from '../exception/tenant.exception';
+import { TenantRepository } from '../repository/tenant.repository';
+import { NewTenantInput } from '../../schema/graphql.schema';
+
+@Injectable()
+export default class TenantService {
+  constructor(private tenantRepository: TenantRepository) {}
+
+  async getTenantByDomain(domain: string): Promise<Tenant> {
+    const tenant = await this.tenantRepository.getTenantByDomain(domain);
+    if (!tenant) {
+      throw new TenantNotFoundException(domain);
+    }
+    return tenant;
+  }
+
+  async createTenant(tenant: NewTenantInput): Promise<Tenant> {
+    return this.tenantRepository.save(tenant);
+  }
+}

src/migrations/1733833844028-MultiTenantFeature.ts

@@ -5,10 +5,10 @@ export class MultiTenantFeature1733833844028 implements MigrationInterface {
 
   public async up(queryRunner: QueryRunner): Promise<void> {
     await queryRunner.query(
-      `CREATE TABLE "tenant" ("deleted_at" TIMESTAMP, "created_at" TIMESTAMP NOT NULL DEFAULT now(), "updated_at" TIMESTAMP NOT NULL DEFAULT now(), "id" uuid NOT NULL DEFAULT uuid_generate_v4(), "name" character varying NOT NULL, CONSTRAINT "PK_da8c6efd67bb301e810e56ac139" PRIMARY KEY ("id"))`,
+      `CREATE TABLE "tenant" ("deleted_at" TIMESTAMP, "created_at" TIMESTAMP NOT NULL DEFAULT now(), "updated_at" TIMESTAMP NOT NULL DEFAULT now(), "id" uuid NOT NULL DEFAULT uuid_generate_v4(), "name" character varying NOT NULL, "domain" character varying NOT NULL, CONSTRAINT "UQ_97b9c4dae58b30f5bd875f241ab" UNIQUE ("domain"), CONSTRAINT "PK_da8c6efd67bb301e810e56ac139" PRIMARY KEY ("id"))`,
     );
     const tenants = await queryRunner.query(
-      `INSERT INTO "tenant" ("name") VALUES ('Default Tenant') RETURNING id`,
+      `INSERT INTO "tenant" ("name", "domain") VALUES ('Default Tenant', 'default.domain') RETURNING id`,
     );
     const tenantId = tenants[0].id;
     await queryRunner.query(