refactor: domain repositories (#4403)

server/src/domain/search/index.ts

@@ -1,4 +1,3 @@
 export * from './dto';
 export * from './response-dto';
-export * from './search.repository';
 export * from './search.service';

server/src/domain/search/search.repository.ts

@@ -1,97 +0,0 @@
-import { AlbumEntity, AssetEntity, AssetFaceEntity, AssetType } from '@app/infra/entities';
-
-export enum SearchCollection {
-  ASSETS = 'assets',
-  ALBUMS = 'albums',
-  FACES = 'faces',
-}
-
-export enum SearchStrategy {
-  CLIP = 'CLIP',
-  TEXT = 'TEXT',
-}
-
-export interface SearchFaceFilter {
-  ownerId: string;
-}
-
-export interface SearchFilter {
-  id?: string;
-  userId: string;
-  type?: AssetType;
-  isFavorite?: boolean;
-  isArchived?: boolean;
-  city?: string;
-  state?: string;
-  country?: string;
-  make?: string;
-  model?: string;
-  objects?: string[];
-  tags?: string[];
-  recent?: boolean;
-  motion?: boolean;
-  debug?: boolean;
-}
-
-export interface SearchResult<T> {
-  /** total matches */
-  total: number;
-  /** collection size */
-  count: number;
-  /** current page */
-  page: number;
-  /** items for page */
-  items: T[];
-  /** score */
-  distances: number[];
-  facets: SearchFacet[];
-}
-
-export interface SearchFacet {
-  fieldName: string;
-  counts: Array<{
-    count: number;
-    value: string;
-  }>;
-}
-
-export interface SearchExploreItem<T> {
-  fieldName: string;
-  items: Array<{
-    value: string;
-    data: T;
-  }>;
-}
-
-export type OwnedFaceEntity = Pick<AssetFaceEntity, 'assetId' | 'personId' | 'embedding'> & {
-  /** computed as assetId|personId */
-  id: string;
-  /** copied from asset.id */
-  ownerId: string;
-};
-
-export type SearchCollectionIndexStatus = Record<SearchCollection, boolean>;
-
-export const ISearchRepository = 'ISearchRepository';
-
-export interface ISearchRepository {
-  setup(): Promise<void>;
-  checkMigrationStatus(): Promise<SearchCollectionIndexStatus>;
-
-  importAlbums(items: AlbumEntity[], done: boolean): Promise<void>;
-  importAssets(items: AssetEntity[], done: boolean): Promise<void>;
-  importFaces(items: OwnedFaceEntity[], done: boolean): Promise<void>;
-
-  deleteAlbums(ids: string[]): Promise<void>;
-  deleteAssets(ids: string[]): Promise<void>;
-  deleteFaces(ids: string[]): Promise<void>;
-  deleteAllFaces(): Promise<number>;
-  updateCLIPField(num_dim: number): Promise<void>;
-
-  searchAlbums(query: string, filters: SearchFilter): Promise<SearchResult<AlbumEntity>>;
-  searchAssets(query: string, filters: SearchFilter): Promise<SearchResult<AssetEntity>>;
-  vectorSearch(query: number[], filters: SearchFilter): Promise<SearchResult<AssetEntity>>;
-  searchFaces(query: number[], filters: SearchFaceFilter): Promise<SearchResult<AssetFaceEntity>>;
-
-  explore(userId: string): Promise<SearchExploreItem<AssetEntity>[]>;
-}

server/src/domain/search/search.service.spec.ts

@@ -15,16 +15,18 @@ import {
   searchStub,
 } from '@test';
 import { plainToInstance } from 'class-transformer';
-import { IAlbumRepository } from '../album/album.repository';
 import { mapAsset } from '../asset';
-import { IAssetRepository } from '../asset/asset.repository';
 import { JobName } from '../job';
-import { IJobRepository } from '../job/job.repository';
-import { IPersonRepository } from '../person/person.repository';
-import { IMachineLearningRepository } from '../smart-info';
-import { ISystemConfigRepository } from '../system-config';
+import {
+  IAlbumRepository,
+  IAssetRepository,
+  IJobRepository,
+  IMachineLearningRepository,
+  IPersonRepository,
+  ISearchRepository,
+  ISystemConfigRepository,
+} from '../repositories';
 import { SearchDto } from './dto';
-import { ISearchRepository } from './search.repository';
 import { SearchService } from './search.service';
 
 jest.useFakeTimers();

server/src/domain/search/search.service.ts

@@ -1,25 +1,28 @@
 import { AlbumEntity, AssetEntity, AssetFaceEntity } from '@app/infra/entities';
 import { Inject, Injectable, Logger } from '@nestjs/common';
 import { mapAlbumWithAssets } from '../album';
-import { IAlbumRepository } from '../album/album.repository';
 import { AssetResponseDto, mapAsset } from '../asset';
-import { IAssetRepository } from '../asset/asset.repository';
 import { AuthUserDto } from '../auth';
 import { usePagination } from '../domain.util';
-import { IAssetFaceJob, IBulkEntityJob, IJobRepository, JOBS_ASSET_PAGINATION_SIZE, JobName } from '../job';
-import { AssetFaceId, IPersonRepository } from '../person';
-import { IMachineLearningRepository } from '../smart-info';
-import { FeatureFlag, ISystemConfigRepository, SystemConfigCore } from '../system-config';
-import { SearchDto } from './dto';
-import { SearchResponseDto } from './response-dto';
+import { IAssetFaceJob, IBulkEntityJob, JOBS_ASSET_PAGINATION_SIZE, JobName } from '../job';
 import {
+  AssetFaceId,
+  IAlbumRepository,
+  IAssetRepository,
+  IJobRepository,
+  IMachineLearningRepository,
+  IPersonRepository,
   ISearchRepository,
+  ISystemConfigRepository,
   OwnedFaceEntity,
   SearchCollection,
   SearchExploreItem,
   SearchResult,
   SearchStrategy,
-} from './search.repository';
+} from '../repositories';
+import { FeatureFlag, SystemConfigCore } from '../system-config';
+import { SearchDto } from './dto';
+import { SearchResponseDto } from './response-dto';
 
 interface SyncQueue {
   upsert: Set<string>;