boredgame/src/lib/server/api/infrastructure/database/seeds/users.ts

import { eq } from 'drizzle-orm';
import { Argon2id } from 'oslo/password';
import { type db } from '$lib/server/api/infrastructure/database';
import * as schema from '$lib/server/api/infrastructure/database/tables';
import users from './data/users.json';
import { config } from '../../../common/config';

type JsonUser = {
	id: string;
	username: string;
	email: string;
	password: string;
	roles: {
		name: string;
		primary: boolean;
	}[];
};

type JsonRole = {
	name: string;
	primary: boolean;
};

export default async function seed(db: db) {
	const adminRole = await db.select().from(schema.roles).where(eq(schema.roles.name, 'admin'));
	const userRole = await db.select().from(schema.roles).where(eq(schema.roles.name, 'user'));

	console.log('Admin Role: ', adminRole);
	const adminUser = await db
		.insert(schema.usersTable)
		.values({
			username: `${config.ADMIN_USERNAME}`,
			email: '',
			first_name: 'Brad',
			last_name: 'S',
			verified: true,
		})
		.returning()
		.onConflictDoNothing();

	console.log('Admin user created.', adminUser);

	await db
		.insert(schema.credentialsTable)
		.values({
			user_id: adminUser[0].id,
			type: schema.CredentialsType.PASSWORD,
			secret_data: await new Argon2id().hash(`${config.ADMIN_PASSWORD}`),
		});

	await db
		.insert(schema.collections)
		.values({ user_id: adminUser[0].id })
		.onConflictDoNothing();

	await db
		.insert(schema.wishlists)
		.values({ user_id: adminUser[0].id })
		.onConflictDoNothing();

	await db
		.insert(schema.user_roles)
		.values({
			user_id: adminUser[0].id,
			role_id: adminRole[0].id,
			primary: true,
		})
		.onConflictDoNothing();

	console.log('Admin user given admin role.');

	await db
		.insert(schema.user_roles)
		.values({
			user_id: adminUser[0].id,
			role_id: userRole[0].id,
			primary: false,
		})
		.onConflictDoNothing();

	console.log('Admin user given user role.');
	await Promise.all(
		users.map(async (user) => {
			const [insertedUser] = await db
				.insert(schema.usersTable)
				.values({
					...user,
				})
				.returning();
			await db.insert(schema.credentialsTable).values({
				user_id: insertedUser?.id,
				type: schema.CredentialsType.PASSWORD,
				secret_data: await new Argon2id().hash(user.password),
			})
			await db.insert(schema.collections).values({ user_id: insertedUser?.id });
			await db.insert(schema.wishlists).values({ user_id: insertedUser?.id });
			await Promise.all(
				user.roles.map(async (role: JsonRole) => {
					const foundRole = await db.query.roles.findFirst({
						where: eq(schema.roles.name, role.name),
					});
					if (!foundRole) { throw new Error('Role not found'); };
					await db.insert(schema.user_roles).values({
						user_id: insertedUser?.id,
						role_id: foundRole?.id,
						primary: role?.primary,
					});
				}),
			);
		}),
	);
}