tianji/src/server/model/user.ts

import { prisma } from './_client';
import bcryptjs from 'bcryptjs';
import { ROLES } from '../utils/const';

async function hashPassword(password: string) {
  return await bcryptjs.hash(password, 10);
}

/**
 * Create User
 */
export async function createAdminUser(username: string, password: string) {
  const count = await prisma.user.count();

  if (count > 0) {
    throw new Error('Create Admin User Just Only allow in non people exist');
  }

  const user = await prisma.user.create({
    data: {
      username,
      password: await hashPassword(password),
      role: ROLES.admin,
      workspaces: {
        create: [
          {
            role: ROLES.owner,
            workspace: {
              create: {
                name: username,
              },
            },
          },
        ],
      },
    },
    include: {
      workspaces: true,
    },
  });

  if (user.workspaces[0]) {
    prisma.user.update({
      where: {
        id: user.id,
      },
      data: {
        currentWorkspaceId: user.workspaces[0].workspaceId,
      },
    });
  }
}

export async function createUser(username: string, password: string) {
  const existCount = await prisma.user.count({
    where: {
      username,
    },
  });

  if (existCount > 0) {
    throw new Error('User already exists');
  }

  const user = await prisma.user.create({
    data: {
      username,
      password: await hashPassword(password),
      role: ROLES.user,
      workspaces: {
        create: [
          {
            role: ROLES.owner,
            workspace: {
              create: {
                name: username,
              },
            },
          },
        ],
      },
    },
    include: {
      workspaces: true,
    },
  });

  if (user.workspaces[0]) {
    prisma.user.update({
      where: {
        id: user.id,
      },
      data: {
        currentWorkspaceId: user.workspaces[0].workspaceId,
      },
    });
  }
}

export async function authUser(username: string, password: string) {
  const user = await prisma.user.findUniqueOrThrow({
    where: {
      username,
      password: await hashPassword(password),
    },
    select: {
      id: true,
      username: true,
      role: true,
      createdAt: true,
      updatedAt: true,
      deletedAt: true,
    },
  });

  return user;
}

export async function findUser(userId: string) {
  const user = await prisma.user.findUnique({
    where: {
      id: userId,
    },
    select: {
      id: true,
      username: true,
      role: true,
    },
  });

  return user;
}