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

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,
					})
				}),
			)
		}),
	)
}