certd/packages/ui/certd-server/src/modules/pipeline/service/storage-service.ts

import { Provide, Scope, ScopeEnum } from '@midwayjs/core';
import { InjectEntityModel } from '@midwayjs/typeorm';
import { In, Repository } from 'typeorm';
import { BaseService } from '@certd/lib-server';
import { StorageEntity } from '../entity/storage.js';

/**
 */
@Provide()
@Scope(ScopeEnum.Singleton)
export class StorageService extends BaseService<StorageEntity> {
  @InjectEntityModel(StorageEntity)
  repository: Repository<StorageEntity>;

  //@ts-ignore
  getRepository() {
    return this.repository;
  }

  async get(where: { scope: any; namespace: any; userId: number; version: string; key: string }) {
    if (where.userId == null) {
      throw new Error('userId 不能为空');
    }
    return await this.repository.findOne({
      where,
    });
  }

  async set(entity: { id?: any; scope: any; namespace: any; userId: number; version: string; value: string; key: string }) {
    entity.id = null;
    const query = { ...entity };
    delete query.value;
    const ret = await this.get(query);
    if (ret != null) {
      entity.id = ret.id;
      if (ret.userId !== entity.userId) {
        throw new Error('您没有权限修改此数据');
      }
      await this.repository.save(entity);
    } else {
      await this.repository.insert(entity);
    }
    return;
  }

  async findPipelineVars(pipelineIds: number[]) {
    if (pipelineIds == null || pipelineIds.length === 0) {
      throw new Error('pipelineIds 不能为空');
    }
    return await this.repository.find({
      where: {
        scope: 'pipeline',
        namespace: In(pipelineIds),
        key: 'vars',
      },
    });
  }

  async getPipelinePrivateVars(pipelineId: number) {
    if (pipelineId == null) {
      throw new Error('pipelineId 不能为空');
    }
    const res = await this.repository.findOne({
      where: {
        scope: 'pipeline',
        namespace: pipelineId + '',
        key: 'privateVars',
      },
    });
    if (!res) {
      return {};
    }
    const value = JSON.parse(res.value);
    return value.value;
  }
}