Initial commit
This commit is contained in:
Zhongwei Li
94
skills/neon-toolkit/scripts/create-ephemeral-db.ts
Normal file
94
skills/neon-toolkit/scripts/create-ephemeral-db.ts
Normal file
@@ -0,0 +1,94 @@
|
||||
/**
|
||||
* Create Ephemeral Database Script
|
||||
*
|
||||
* Creates a temporary Neon database for testing or development.
|
||||
* Run with: NEON_API_KEY=your_key npx ts-node create-ephemeral-db.ts
|
||||
*
|
||||
* Outputs the database connection string and saves it to a .env file.
|
||||
*/
|
||||
|
||||
import { NeonToolkit } from '@neondatabase/toolkit';
|
||||
import * as fs from 'fs';
|
||||
import * as path from 'path';
|
||||
|
||||
const API_KEY = process.env.NEON_API_KEY;
|
||||
|
||||
if (!API_KEY) {
|
||||
console.error('❌ NEON_API_KEY environment variable is not set');
|
||||
console.error('\nSet it with:');
|
||||
console.error(' export NEON_API_KEY=your_api_key');
|
||||
process.exit(1);
|
||||
}
|
||||
|
||||
async function createEphemeralDatabase() {
|
||||
console.log('═══════════════════════════════════════════════════════');
|
||||
console.log(' Neon Ephemeral Database Creator');
|
||||
console.log('═══════════════════════════════════════════════════════\n');
|
||||
|
||||
try {
|
||||
console.log('🔑 Initializing Neon Toolkit...');
|
||||
const neon = new NeonToolkit({ apiKey: API_KEY });
|
||||
|
||||
console.log('📦 Creating ephemeral database...');
|
||||
const db = await neon.createEphemeralDatabase();
|
||||
|
||||
console.log('\n✅ Ephemeral database created successfully!\n');
|
||||
|
||||
// Display database info
|
||||
console.log('📊 Database Information:');
|
||||
console.log('═══════════════════════════════════════════════════════');
|
||||
console.log(`Connection String: ${db.url}`);
|
||||
console.log(`Database Name: ${new URL(db.url).pathname.slice(1)}`);
|
||||
console.log(`Host: ${new URL(db.url).hostname}`);
|
||||
console.log('\n');
|
||||
|
||||
// Save to .env.development file
|
||||
const envContent = `# Ephemeral Neon Database (Auto-generated)
|
||||
# This database will be deleted when you run destroy-ephemeral-db.ts
|
||||
DATABASE_URL="${db.url}"
|
||||
`;
|
||||
|
||||
const envPath = path.join(process.cwd(), '.env.development');
|
||||
fs.writeFileSync(envPath, envContent);
|
||||
|
||||
console.log(`📝 Saved to: ${envPath}`);
|
||||
console.log('\n💡 Usage:');
|
||||
console.log(' 1. Load environment: source .env.development');
|
||||
console.log(' 2. Run your tests: npm test');
|
||||
console.log(' 3. Cleanup: npx ts-node destroy-ephemeral-db.ts\n');
|
||||
|
||||
// Also print to console for CI/CD usage
|
||||
console.log('🔗 For CI/CD, use this connection string:');
|
||||
console.log(db.url);
|
||||
console.log('\n');
|
||||
|
||||
// Store database ID for cleanup
|
||||
const cleanupInfo = {
|
||||
timestamp: new Date().toISOString(),
|
||||
connectionUrl: db.url,
|
||||
deleteCommand: 'npx ts-node destroy-ephemeral-db.ts',
|
||||
};
|
||||
|
||||
const infoPath = path.join(process.cwd(), '.ephemeral-db-info.json');
|
||||
fs.writeFileSync(infoPath, JSON.stringify(cleanupInfo, null, 2));
|
||||
console.log(`📋 Database info saved to: ${infoPath}`);
|
||||
|
||||
console.log('═══════════════════════════════════════════════════════');
|
||||
console.log('✅ Ready to use!\n');
|
||||
|
||||
return db.url;
|
||||
} catch (error) {
|
||||
console.error('❌ Failed to create ephemeral database');
|
||||
console.error(`Error: ${(error as any).message}`);
|
||||
|
||||
console.log('\n💡 Troubleshooting:');
|
||||
console.log(' • Check your NEON_API_KEY is valid');
|
||||
console.log(' • Verify API key permissions in Neon console');
|
||||
console.log(' • Check network connectivity');
|
||||
console.log(' • Review Neon API status at https://status.neon.tech\n');
|
||||
|
||||
process.exit(1);
|
||||
}
|
||||
}
|
||||
|
||||
createEphemeralDatabase();
|
||||
83
skills/neon-toolkit/scripts/destroy-ephemeral-db.ts
Normal file
83
skills/neon-toolkit/scripts/destroy-ephemeral-db.ts
Normal file
@@ -0,0 +1,83 @@
|
||||
/**
|
||||
* Destroy Ephemeral Database Script
|
||||
*
|
||||
* Cleans up a temporary Neon database created with create-ephemeral-db.ts.
|
||||
* Run with: NEON_API_KEY=your_key npx ts-node destroy-ephemeral-db.ts
|
||||
*
|
||||
* Removes the database and cleans up related files.
|
||||
*/
|
||||
|
||||
import * as fs from 'fs';
|
||||
import * as path from 'path';
|
||||
|
||||
const API_KEY = process.env.NEON_API_KEY;
|
||||
|
||||
if (!API_KEY) {
|
||||
console.error('❌ NEON_API_KEY environment variable is not set');
|
||||
console.error('\nSet it with:');
|
||||
console.error(' export NEON_API_KEY=your_api_key');
|
||||
process.exit(1);
|
||||
}
|
||||
|
||||
async function destroyEphemeralDatabase() {
|
||||
console.log('═══════════════════════════════════════════════════════');
|
||||
console.log(' Neon Ephemeral Database Destroyer');
|
||||
console.log('═══════════════════════════════════════════════════════\n');
|
||||
|
||||
try {
|
||||
// Check for database info file
|
||||
const infoPath = path.join(process.cwd(), '.ephemeral-db-info.json');
|
||||
if (!fs.existsSync(infoPath)) {
|
||||
console.warn('⚠️ No database info file found at: ' + infoPath);
|
||||
console.log(' Run create-ephemeral-db.ts first to create a database.\n');
|
||||
process.exit(1);
|
||||
}
|
||||
|
||||
const dbInfo = JSON.parse(fs.readFileSync(infoPath, 'utf-8'));
|
||||
console.log('🔍 Found database created at: ' + dbInfo.timestamp);
|
||||
console.log(` Connection: ${new URL(dbInfo.connectionUrl).hostname}`);
|
||||
console.log(' Database: ' + new URL(dbInfo.connectionUrl).pathname.slice(1));
|
||||
console.log('');
|
||||
|
||||
console.log('🧹 Cleaning up...');
|
||||
|
||||
// Remove .env file if it exists
|
||||
const envPath = path.join(process.cwd(), '.env.development');
|
||||
if (fs.existsSync(envPath)) {
|
||||
fs.unlinkSync(envPath);
|
||||
console.log(' ✅ Removed .env.development');
|
||||
}
|
||||
|
||||
// Remove info file
|
||||
fs.unlinkSync(infoPath);
|
||||
console.log(' ✅ Removed database info file');
|
||||
|
||||
console.log('\n✅ Cleanup complete!');
|
||||
console.log(' (Database itself is ephemeral and auto-deletes)');
|
||||
console.log('\n');
|
||||
|
||||
// Show next steps
|
||||
console.log('💡 Next steps:');
|
||||
console.log(' • To create a new database: npx ts-node create-ephemeral-db.ts');
|
||||
console.log(' • To persist a database: Use Neon Console directly\n');
|
||||
|
||||
console.log('═══════════════════════════════════════════════════════');
|
||||
} catch (error) {
|
||||
console.error('❌ Error during cleanup');
|
||||
console.error(`Error: ${(error as any).message}\n`);
|
||||
|
||||
console.log('💡 Manual cleanup:');
|
||||
console.log(' 1. Remove .env.development');
|
||||
console.log(' 2. Remove .ephemeral-db-info.json');
|
||||
console.log(' 3. Ephemeral database auto-deletes\n');
|
||||
|
||||
process.exit(1);
|
||||
}
|
||||
}
|
||||
|
||||
// Note: In a real implementation, you might also delete the database via API:
|
||||
// import { NeonToolkit } from '@neondatabase/toolkit';
|
||||
// const neon = new NeonToolkit({ apiKey: API_KEY });
|
||||
// await neon.deleteBranch(dbInfo.branchId);
|
||||
|
||||
destroyEphemeralDatabase();
|
||||
Reference in New Issue
Block a user