cli.ts

#!/usr/bin/env bun

import { join } from 'node:path'
import { createCLI } from '@rubriclab/cli'
import { $ } from 'bun'
import { z } from 'zod/v4'

const cli = createCLI({
	commands: [
		{
			args: z.object({}),
			description: 'Prepares a submodule',
			handler: async () => {
				// Skip in CI environments
				if (process.env.CI) {
					console.log('CI detected, skipping prepare')
					return
				}

				try {
					const gitDir = (await $`git rev-parse --git-dir`.text()).replace('\n', '')

					try {
						await $`rm -r ${gitDir}/hooks`
					} catch (_e) {
						console.log('Hooks already removed')
					}

					await $`mkdir -p ${gitDir}/hooks`
					await $`git config core.hooksPath ${gitDir}/hooks`
					await $`bun x simple-git-hooks`
					console.log('Git hooks setup complete')
				} catch (error) {
					console.log(
						'Git hooks setup failed (this may be fine):',
						error.stderr?.toString() || error.message
					)
				}
			},
			name: 'prepare'
		},
		{
			args: z.object({
				'working-dir': z.string().optional().describe('Working directory for the package (defaults to current directory)')
			}),
			description: 'Runs the post-commit hook',
			handler: async (args) => {
				if (process.env.AMENDING) {
					process.exit(0)
				}
				// Use specified working directory or current working directory
				const workingDir = args['working-dir'] || process.cwd()
				const CHANGELOG_PATH = join(workingDir, 'CHANGELOG.md')
				const PACKAGE_PATH = join(workingDir, 'package.json')

				const packageJsonContent = Bun.file(PACKAGE_PATH)
				const packageJson = await packageJsonContent.json()

				console.log({ packageJson })

				const versionParts = packageJson.version.split('.').map(Number)
				versionParts[2] += 1
				packageJson.version = versionParts.join('.')

				await Bun.write(PACKAGE_PATH, `${JSON.stringify(packageJson, null, 2)}\n`)

				const commitHash = (await $`git rev-parse HEAD`).text().trim()
				const commitMessage = (await $`git log -1 --pretty=%B`).text().trim()
				const commitDate = (await $`git log -1 --pretty=%cd --date=short`).text().trim()
				let repositoryUrl = (await $`git config --get remote.origin.url`).text().trim()

				if (repositoryUrl.endsWith('.git')) {
					repositoryUrl = repositoryUrl.slice(0, -4)
				}
				if (repositoryUrl.startsWith('git@')) {
					repositoryUrl = repositoryUrl.replace('git@', 'https://').replace(':', '/')
				}

				const commitUrl = `${repositoryUrl}/commit/${commitHash}`
				const changelogEntry = `- [${commitDate}] [${commitMessage}](${commitUrl})\n`

				let changelogContent = ''
				const changelog = Bun.file(CHANGELOG_PATH)
				const changelogExists = await changelog.exists()
				if (changelogExists) {
					changelogContent = await changelog.text()
				} else {
					changelogContent = '# Changelog\n\n'
				}

				await Bun.write(CHANGELOG_PATH, `${changelogEntry}${changelogContent}`)

				await $`bun x biome format --write ./package.json`
				await $`git add package.json CHANGELOG.md`
				process.env.AMENDING = 'true'
				await $`git commit --amend --no-verify --no-edit`
				process.env.AMENDING = undefined
			},
			name: 'post-commit'
		},
		{
			args: z.object({}),
			description: 'Sets up a new package',
			handler: async () => {
				const projectRoot = process.cwd()
				const pkgJsonPath = join(projectRoot, 'package.json')

				const pkgJson = await Bun.file(pkgJsonPath).json()

				pkgJson['simple-git-hooks'] = { 'post-commit': 'rubriclab-package post-commit' }

				pkgJson.scripts = pkgJson.scripts || {}
				// pkgJson.scripts.prepare = 'rubriclab-package prepare'
				pkgJson.scripts.bleed = 'bun x npm-check-updates -u'
				pkgJson.scripts.clean = 'rm -rf .next && rm -rf node_modules'
				pkgJson.scripts.format = 'bun x biome format --write .'
				pkgJson.scripts.lint = 'bun x biome check .'
				pkgJson.scripts['lint:fix'] = 'bun x biome lint . --write --unsafe'
				pkgJson.publishConfig = { access: 'public' }

				const file = Bun.file(join(import.meta.dir, 'workflows/publish-package.yml'))

				await Promise.all([
					Bun.write(pkgJsonPath, JSON.stringify(pkgJson, null, 2)),
					Bun.write(join(projectRoot, '.github/workflows/publish.yml'), file)
				])

				console.log('Set up package successfully.')
			},
			name: 'setup-package'
		}
	],
	description: 'Package CLI tool',
	name: 'package cli',
	version: '0.0.0'
})

cli.parse()