add auth, drizzle, prisma setup logic with template

This commit is contained in:
Aman Varshney
2025-03-04 09:33:31 +05:30
parent 54d63823bb
commit 792885b9c4
68 changed files with 2692 additions and 921 deletions

View File

@@ -1,195 +1,120 @@
import os from "node:os";
import path from "node:path";
import { cancel, confirm, isCancel, log, spinner, text } from "@clack/prompts";
import { $ } from "execa";
import { log, spinner } from "@clack/prompts";
import fs from "fs-extra";
import pc from "picocolors";
import { isTursoInstalled, isTursoLoggedIn } from "../utils/turso-cli";
import { setupTurso } from "./turso-setup";
interface TursoConfig {
dbUrl: string;
authToken: string;
}
async function loginToTurso() {
export async function setupDatabase(
projectDir: string,
databaseType: string,
orm: string,
setupTursoDb = true,
): Promise<void> {
const s = spinner();
try {
s.start("Logging in to Turso...");
await $`turso auth login`;
s.stop("Logged in to Turso successfully!");
} catch (error) {
s.stop(pc.red("Failed to log in to Turso"));
throw error;
}
}
const serverDir = path.join(projectDir, "packages/server");
async function installTursoCLI(isMac: boolean) {
const s = spinner();
try {
s.start("Installing Turso CLI...");
if (isMac) {
await $`brew install tursodatabase/tap/turso`;
} else {
const { stdout: installScript } =
await $`curl -sSfL https://get.tur.so/install.sh`;
await $`bash -c '${installScript}'`;
}
s.stop("Turso CLI installed successfully!");
} catch (error) {
if (error instanceof Error && error.message.includes("User force closed")) {
s.stop();
log.warn(pc.yellow("Turso CLI installation cancelled by user"));
throw new Error("Installation cancelled");
}
s.stop(pc.red("Failed to install Turso CLI"));
throw error;
}
}
async function createTursoDatabase(dbName: string): Promise<TursoConfig> {
try {
await $`turso db create ${dbName}`;
} catch (error) {
if (error instanceof Error && error.message.includes("already exists")) {
throw new Error("DATABASE_EXISTS");
}
throw error;
}
const { stdout: dbUrl } = await $`turso db show ${dbName} --url`;
const { stdout: authToken } = await $`turso db tokens create ${dbName}`;
return {
dbUrl: dbUrl.trim(),
authToken: authToken.trim(),
};
}
async function writeEnvFile(projectDir: string, config?: TursoConfig) {
const envPath = path.join(projectDir, "packages/server", ".env");
const envContent = config
? `TURSO_DATABASE_URL="${config.dbUrl}"
TURSO_AUTH_TOKEN="${config.authToken}"`
: `TURSO_DATABASE_URL=
TURSO_AUTH_TOKEN=`;
await fs.writeFile(envPath, envContent);
}
function displayManualSetupInstructions() {
log.info(`Manual Turso Setup Instructions:
1. Visit https://turso.tech and create an account
2. Create a new database from the dashboard
3. Get your database URL and authentication token
4. Add these credentials to the .env file in packages/server/.env
TURSO_DATABASE_URL=your_database_url
TURSO_AUTH_TOKEN=your_auth_token`);
}
export async function setupTurso(projectDir: string) {
const platform = os.platform();
const isMac = platform === "darwin";
const canInstallCLI = platform !== "win32";
if (!canInstallCLI) {
log.warn(pc.yellow("Automatic Turso setup is not supported on Windows."));
await writeEnvFile(projectDir);
displayManualSetupInstructions();
if (databaseType === "none") {
await fs.remove(path.join(serverDir, "src/db"));
log.info(pc.yellow("Database configuration removed"));
return;
}
try {
const isCliInstalled = await isTursoInstalled();
if (!isCliInstalled) {
const shouldInstall = await confirm({
message: "Would you like to install Turso CLI?",
});
if (isCancel(shouldInstall)) {
cancel(pc.red("Operation cancelled"));
process.exit(0);
if (databaseType === "sqlite") {
if (orm === "drizzle") {
await setupDrizzleDependencies(projectDir);
await setupTurso(projectDir, setupTursoDb);
} else if (orm === "prisma") {
await setupPrismaDependencies(projectDir);
await setupTurso(projectDir, setupTursoDb);
}
if (!shouldInstall) {
await writeEnvFile(projectDir);
displayManualSetupInstructions();
return;
}
const s = spinner();
s.start("Installing Turso CLI...");
try {
if (isMac) {
await $`brew install tursodatabase/tap/turso`;
} else {
const { stdout: installScript } =
await $`curl -sSfL https://get.tur.so/install.sh`;
await $`bash -c '${installScript}'`;
}
s.stop("Turso CLI installed successfully!");
} catch (error) {
s.stop(pc.red("Failed to install Turso CLI"));
throw error;
}
}
const isLoggedIn = await isTursoLoggedIn();
if (!isLoggedIn) {
const s = spinner();
s.start("Logging in to Turso...");
try {
await $`turso auth login`;
s.stop("Logged in to Turso successfully!");
} catch (error) {
s.stop(pc.red("Failed to log in to Turso"));
throw error;
}
}
let success = false;
let dbName = "";
let suggestedName = path.basename(projectDir);
while (!success) {
const dbNameResponse = await text({
message: "Enter a name for your database:",
defaultValue: suggestedName,
initialValue: suggestedName,
placeholder: suggestedName,
});
if (isCancel(dbNameResponse)) {
cancel(pc.red("Operation cancelled"));
process.exit(0);
}
dbName = dbNameResponse as string;
const s = spinner();
try {
s.start(`Creating Turso database "${dbName}"...`);
const config = await createTursoDatabase(dbName);
await writeEnvFile(projectDir, config);
s.stop("Turso database configured successfully!");
success = true;
} catch (error) {
if (error instanceof Error && error.message === "DATABASE_EXISTS") {
s.stop(pc.yellow(`Database "${pc.red(dbName)}" already exists`));
suggestedName = `${dbName}-${Math.floor(Math.random() * 1000)}`;
} else {
throw error;
}
} else if (databaseType === "postgres") {
log.info(
pc.blue(
"PostgreSQL setup is coming in a future update. Using SQLite configuration for now.",
),
);
if (orm === "drizzle") {
await setupDrizzleDependencies(projectDir);
await setupTurso(projectDir, setupTursoDb);
} else if (orm === "prisma") {
await setupPrismaDependencies(projectDir);
await setupTurso(projectDir, setupTursoDb);
}
}
} catch (error) {
log.error(pc.red(`Error during Turso setup: ${error}`));
await writeEnvFile(projectDir);
displayManualSetupInstructions();
log.success("Setup completed with manual configuration required.");
s.stop(pc.red("Failed to set up database"));
if (error instanceof Error) {
log.error(pc.red(error.message));
}
throw error;
}
}
async function setupDrizzleDependencies(projectDir: string): Promise<void> {
const serverDir = path.join(projectDir, "packages/server");
const packageJsonPath = path.join(serverDir, "package.json");
if (await fs.pathExists(packageJsonPath)) {
const packageJson = await fs.readJSON(packageJsonPath);
packageJson.dependencies = {
...packageJson.dependencies,
"drizzle-orm": "^0.38.4",
"@libsql/client": "^0.14.0",
};
packageJson.devDependencies = {
...packageJson.devDependencies,
"drizzle-kit": "^0.30.4",
};
packageJson.scripts = {
...packageJson.scripts,
"db:generate": "drizzle-kit generate",
"db:migrate": "drizzle-kit push",
"db:studio": "drizzle-kit studio",
};
await fs.writeJSON(packageJsonPath, packageJson, { spaces: 2 });
}
}
async function setupPrismaDependencies(projectDir: string): Promise<void> {
const serverDir = path.join(projectDir, "packages/server");
const packageJsonPath = path.join(serverDir, "package.json");
if (await fs.pathExists(packageJsonPath)) {
const packageJson = await fs.readJSON(packageJsonPath);
packageJson.dependencies = {
...packageJson.dependencies,
"@prisma/client": "^5.7.1",
"@prisma/adapter-libsql": "^5.7.1",
"@libsql/client": "^0.14.0",
};
packageJson.devDependencies = {
...packageJson.devDependencies,
prisma: "^5.7.1",
};
packageJson.scripts = {
...packageJson.scripts,
"prisma:generate": "prisma generate",
"prisma:push": "prisma db push",
"prisma:studio": "prisma studio",
};
await fs.writeJSON(packageJsonPath, packageJson, { spaces: 2 });
}
const envPath = path.join(serverDir, ".env");
if (await fs.pathExists(envPath)) {
const envContent = await fs.readFile(envPath, "utf8");
if (!envContent.includes("DATABASE_URL")) {
const databaseUrlLine = `\nDATABASE_URL="file:./dev.db"`;
await fs.appendFile(envPath, databaseUrlLine);
}
}
}