Transfer repository from Gitlab

server/src/api-v1/device-info/device-info.controller.ts

@@ -0,0 +1,22 @@
+import { Controller, Get, Post, Body, Patch, Param, Delete, UseGuards, ValidationPipe } from '@nestjs/common';
+import { AuthUserDto, GetAuthUser } from '../../decorators/auth-user.decorator';
+import { JwtAuthGuard } from '../../modules/immich-jwt/guards/jwt-auth.guard';
+import { DeviceInfoService } from './device-info.service';
+import { CreateDeviceInfoDto } from './dto/create-device-info.dto';
+import { UpdateDeviceInfoDto } from './dto/update-device-info.dto';
+
+@UseGuards(JwtAuthGuard)
+@Controller('device-info')
+export class DeviceInfoController {
+  constructor(private readonly deviceInfoService: DeviceInfoService) {}
+
+  @Post()
+  async create(@Body(ValidationPipe) createDeviceInfoDto: CreateDeviceInfoDto, @GetAuthUser() authUser: AuthUserDto) {
+    return await this.deviceInfoService.create(createDeviceInfoDto, authUser);
+  }
+
+  @Patch()
+  async update(@Body(ValidationPipe) updateDeviceInfoDto: UpdateDeviceInfoDto, @GetAuthUser() authUser: AuthUserDto) {
+    return this.deviceInfoService.update(authUser.id, updateDeviceInfoDto);
+  }
+}

server/src/api-v1/device-info/device-info.module.ts

@@ -0,0 +1,12 @@
+import { Module } from '@nestjs/common';
+import { DeviceInfoService } from './device-info.service';
+import { DeviceInfoController } from './device-info.controller';
+import { TypeOrmModule } from '@nestjs/typeorm';
+import { DeviceInfoEntity } from './entities/device-info.entity';
+
+@Module({
+  imports: [TypeOrmModule.forFeature([DeviceInfoEntity])],
+  controllers: [DeviceInfoController],
+  providers: [DeviceInfoService],
+})
+export class DeviceInfoModule {}

server/src/api-v1/device-info/device-info.service.ts

@@ -0,0 +1,63 @@
+import { BadRequestException, HttpCode, Injectable, Logger, Res } from '@nestjs/common';
+import { InjectRepository } from '@nestjs/typeorm';
+import { Repository } from 'typeorm';
+import { AuthUserDto } from '../../decorators/auth-user.decorator';
+import { CreateDeviceInfoDto } from './dto/create-device-info.dto';
+import { UpdateDeviceInfoDto } from './dto/update-device-info.dto';
+import { DeviceInfoEntity } from './entities/device-info.entity';
+
+@Injectable()
+export class DeviceInfoService {
+  constructor(
+    @InjectRepository(DeviceInfoEntity)
+    private deviceRepository: Repository<DeviceInfoEntity>,
+  ) {}
+
+  async create(createDeviceInfoDto: CreateDeviceInfoDto, authUser: AuthUserDto) {
+    const res = await this.deviceRepository.findOne({
+      deviceId: createDeviceInfoDto.deviceId,
+      userId: authUser.id,
+    });
+
+    if (res) {
+      Logger.log('Device Info Exist', 'createDeviceInfo');
+      return res;
+    }
+
+    const deviceInfo = new DeviceInfoEntity();
+    deviceInfo.deviceId = createDeviceInfoDto.deviceId;
+    deviceInfo.deviceType = createDeviceInfoDto.deviceType;
+    deviceInfo.userId = authUser.id;
+
+    try {
+      return await this.deviceRepository.save(deviceInfo);
+    } catch (e) {
+      Logger.error('Error creating new device info', 'createDeviceInfo');
+    }
+  }
+
+  async update(userId: string, updateDeviceInfoDto: UpdateDeviceInfoDto) {
+    const deviceInfo = await this.deviceRepository.findOne({
+      where: { deviceId: updateDeviceInfoDto.deviceId, userId: userId },
+    });
+
+    if (!deviceInfo) {
+      throw new BadRequestException('Device Not Found');
+    }
+
+    const res = await this.deviceRepository.update(
+      {
+        id: deviceInfo.id,
+      },
+      updateDeviceInfoDto,
+    );
+
+    if (res.affected == 1) {
+      return await this.deviceRepository.findOne({
+        where: { deviceId: updateDeviceInfoDto.deviceId, userId: userId },
+      });
+    } else {
+      throw new BadRequestException('Bad Request');
+    }
+  }
+}

server/src/api-v1/device-info/dto/create-device-info.dto.ts

@@ -0,0 +1,13 @@
+import { IsNotEmpty, IsOptional } from 'class-validator';
+import { DeviceType } from '../entities/device-info.entity';
+
+export class CreateDeviceInfoDto {
+  @IsNotEmpty()
+  deviceId: string;
+
+  @IsNotEmpty()
+  deviceType: DeviceType;
+
+  @IsOptional()
+  isAutoBackup: boolean;
+}

server/src/api-v1/device-info/dto/update-device-info.dto.ts

@@ -0,0 +1,6 @@
+import { PartialType } from '@nestjs/mapped-types';
+import { IsOptional } from 'class-validator';
+import { DeviceType } from '../entities/device-info.entity';
+import { CreateDeviceInfoDto } from './create-device-info.dto';
+
+export class UpdateDeviceInfoDto extends PartialType(CreateDeviceInfoDto) {}

server/src/api-v1/device-info/entities/device-info.entity.ts

@@ -0,0 +1,32 @@
+import { Column, CreateDateColumn, Entity, PrimaryGeneratedColumn, Unique } from 'typeorm';
+
+@Entity('device_info')
+@Unique(['userId', 'deviceId'])
+export class DeviceInfoEntity {
+  @PrimaryGeneratedColumn()
+  id: number;
+
+  @Column()
+  userId: string;
+
+  @Column()
+  deviceId: string;
+
+  @Column()
+  deviceType: DeviceType;
+
+  @Column({ nullable: true })
+  notificationToken: string;
+
+  @CreateDateColumn()
+  createdAt: string;
+
+  @Column({ type: 'bool', default: false })
+  isAutoBackup: boolean;
+}
+
+export enum DeviceType {
+  IOS = 'IOS',
+  ANDROID = 'ANDROID',
+  WEB = 'WEB',
+}