KnowledgeBase/server/src/app.module.ts

import { Module } from '@nestjs/common';
import { ConfigModule, ConfigService } from '@nestjs/config';
import { TypeOrmModule } from '@nestjs/typeorm';
import { ScheduleModule } from '@nestjs/schedule';
import { ServeStaticModule } from '@nestjs/serve-static';
import { join } from 'path';
import { APP_GUARD } from '@nestjs/core';
import { AppController } from './app.controller';
import { AppService } from './app.service';
import { ApiModule } from './api/api.module';
import { ElasticsearchModule } from './elasticsearch/elasticsearch.module';
import { UploadModule } from './upload/upload.module';
import { ChatModule } from './chat/chat.module';
import { AuthModule } from './auth/auth.module';
import { I18nModule } from './i18n/i18n.module';
import { JwtAuthGuard } from './auth/jwt-auth.guard';
import { KnowledgeBaseModule } from './knowledge-base/knowledge-base.module';
import { ModelConfigModule } from './model-config/model-config.module';
import { UserModule } from './user/user.module';
import { UserSettingModule } from './user-setting/user-setting.module';
import { TikaModule } from './tika/tika.module';
import { VisionModule } from './vision/vision.module';
import { LibreOfficeModule } from './libreoffice/libreoffice.module';
import { Pdf2ImageModule } from './pdf2image/pdf2image.module';
import { VisionPipelineModule } from './vision-pipeline/vision-pipeline.module';
import { KnowledgeGroupModule } from './knowledge-group/knowledge-group.module';
import { SearchHistoryModule } from './search-history/search-history.module';
import { NoteModule } from './note/note.module';
import { PodcastModule } from './podcasts/podcast.module';
import { ImportTaskModule } from './import-task/import-task.module';
import { User } from './user/user.entity';
import { UserSetting } from './user-setting/user-setting.entity';
import { ModelConfig } from './model-config/model-config.entity';
import { KnowledgeBase } from './knowledge-base/knowledge-base.entity';
import { KnowledgeGroup } from './knowledge-group/knowledge-group.entity';
import { SearchHistory } from './search-history/search-history.entity';
import { ChatMessage } from './search-history/chat-message.entity';
import { Note } from './note/note.entity';
import { PodcastEpisode } from './podcasts/entities/podcast-episode.entity';
import { ImportTask } from './import-task/import-task.entity';

@Module({
  imports: [
    ConfigModule.forRoot({
      isGlobal: true,
      ignoreEnvFile: false,
    }),
    ServeStaticModule.forRoot({
      rootPath: join(process.cwd(), 'uploads'),
      serveRoot: '/uploads',
    }),
    ScheduleModule.forRoot(),
    TypeOrmModule.forRootAsync({
      imports: [ConfigModule],
      inject: [ConfigService],
      useFactory: (configService: ConfigService) => ({
        type: 'better-sqlite3',
        database: configService.get<string>('DATABASE_PATH'),
        entities: [
          User,
          UserSetting,
          ModelConfig,
          KnowledgeBase,
          KnowledgeGroup,
          SearchHistory,
          ChatMessage,
          Note,
          PodcastEpisode,
          ImportTask,
        ],
        synchronize: true, // Auto-create database schema. Disable in production.
      }),
    }),
    AuthModule,
    I18nModule,
    UserModule,
    UserSettingModule,
    ModelConfigModule,
    KnowledgeBaseModule,
    KnowledgeGroupModule,
    SearchHistoryModule,
    NoteModule,
    PodcastModule,
    TikaModule,
    VisionModule,
    LibreOfficeModule,
    Pdf2ImageModule,
    VisionPipelineModule,
    ApiModule,
    ElasticsearchModule,
    UploadModule,
    ChatModule,
    ImportTaskModule,
  ],
  controllers: [AppController],
  providers: [
    AppService,
    {
      provide: APP_GUARD,
      useClass: JwtAuthGuard,
    },
  ],
})
export class AppModule { }