@leverege/build-tools 2.96.1 → 2.96.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@leverege/build-tools",
-  "version": "2.96.1",
+  "version": "2.96.2",
   "description": "A collection of build / support tools for Leverege developers",
   "main": "index.js",
   "repository": {
@@ -89,6 +89,7 @@
     "execa": "^9.6.1",
     "find-yarn-workspace-root": "^2.0.0",
     "glob": "^13.0.0",
+    "googleapis": "^170.1.0",
     "handlebars": "^4.7.8",
     "ignore": "^7.0.5",
     "inquirer": "^13.2.0",

package/src/gcp-snapshots.mjs

@@ -0,0 +1,182 @@
+#!/usr/bin/env node
+
+import { program } from 'commander'
+import { google } from 'googleapis'
+
+import {
+  debug,
+  errorExit,
+  log,
+  err,
+} from './Utils.mjs'
+
+program
+  .name( 'gcp-snapshots' )
+  .description( 'Simple GCP wrapper to ease the pain of snapshot management' )
+  .requiredOption( '--project <id>', 'GCP project id (required)' )
+  .option( '--age <days>', 'Number of days (default: 180)', '180' )
+  .option( '--list', 'List snapshots older than --age days' )
+  .option( '--delete', 'Delete snapshots older than --age days' )
+  .parse()
+
+const options = program.opts()
+
+const parseAgeDays = ( value ) => {
+  const n = Number( value )
+  if ( !Number.isFinite( n ) || n <= 0 ) {
+    errorExit( `--age must be a positive number. Got: ${value}` )
+  }
+  return n
+}
+
+const cutoffDateFromAgeDays = ( days ) => {
+  return new Date( Date.now() - days * 24 * 60 * 60 * 1000 )
+}
+
+const bytesToGb = ( bytes ) => {
+  if ( !Number.isFinite( bytes ) || bytes <= 0 ) return 0
+  return Math.round( ( bytes / ( 1024 ** 3 ) ) * 10 ) / 10 // 1 decimal GB
+}
+
+const formatRow = ( cols, widths ) => {
+  return cols.map( ( c, i ) => {
+    const s = String( c ?? '' )
+    return s.length > widths[i] ?
+      `${s.slice( 0, widths[i] - 1 )}…` : s.padEnd( widths[i] )
+  } ).join( '  ' )
+}
+
+const getComputeClient = async () => {
+  const auth = await google.auth.getClient( {
+    scopes : [ 'https://www.googleapis.com/auth/cloud-platform' ],
+  } )
+
+  return google.compute( { version : 'v1', auth } )
+}
+
+const fetchSnapshotsOlderThan = async ( { compute, projectId, cutoff } ) => {
+  const snapshots = []
+  let pageToken
+
+  debug( `Listing snapshots for project=${projectId}` )
+  debug( `Cutoff date=${cutoff.toISOString()}` )
+
+  do {
+    // eslint-disable-next-line no-await-in-loop
+    const res = await compute.snapshots.list( {
+      project : projectId,
+      maxResults : 500,
+      pageToken,
+    } )
+
+    const items = res.data.items ?? []
+    for ( const s of items ) {
+      if ( !s.creationTimestamp ) continue
+      const created = new Date( s.creationTimestamp )
+      if ( created < cutoff ) snapshots.push( s )
+    }
+
+    pageToken = res.data.nextPageToken
+  } while ( pageToken )
+
+  // Oldest -> newest
+  snapshots.sort( ( a, b ) => (
+    new Date( a.creationTimestamp ) - new Date( b.creationTimestamp )
+  ) )
+
+  return snapshots
+}
+
+const printSnapshotTable = async ( { snapshots, ageDays, cutoff } ) => {
+  log(
+    `Snapshots older than ${ageDays} days ` +
+    `(cutoff ${cutoff.toISOString().slice( 0, 10 )}) — ${snapshots.length} found`
+  )
+
+  const widths = [ 40, 20, 10, 8, 32 ]
+  log( formatRow(
+    [ 'NAME', 'CREATED', 'SIZE_GB', 'DISK_GB', 'SOURCE_DISK' ],
+    widths
+  ) )
+
+  for ( const s of snapshots ) {
+    const sourceDisk = s.sourceDisk ? s.sourceDisk.split( '/' ).pop() : ''
+
+    log( formatRow(
+      [
+        s.name,
+        s.creationTimestamp.replace( 'T', ' ' ).replace( 'Z', '' ),
+        bytesToGb( Number( s.storageBytes ) ),
+        s.diskSizeGb ?? '',
+        sourceDisk,
+      ],
+      widths
+    ) )
+  }
+}
+
+const deleteSnapshots = async ( { compute, projectId, snapshots } ) => {
+  if ( snapshots.length === 0 ) {
+    log( 'Nothing to delete.' )
+    return
+  }
+
+  log( `Deleting ${snapshots.length} snapshot(s)...` )
+
+  let ok = 0
+  let failed = 0
+
+  for ( const s of snapshots ) {
+    const name = s.name
+    if ( !name ) continue
+
+    try {
+      log( `- deleting: ${name}` )
+      // eslint-disable-next-line no-await-in-loop
+      await compute.snapshots.delete( {
+        project : projectId,
+        snapshot : name,
+      } )
+      ok++
+    } catch ( e ) {
+      failed++
+      err( `! failed: ${name} — ${e?.message ?? e}` )
+    }
+  }
+
+  log( `Done. Deleted: ${ok}. Failed: ${failed}.` )
+}
+
+const main = async () => {
+  if ( !options.list && !options.delete ) {
+    program.help( { error : false } )
+    return
+  }
+
+  const ageDays = parseAgeDays( options.age )
+  const cutoff = cutoffDateFromAgeDays( ageDays )
+
+  const compute = await getComputeClient()
+
+  debug( `project=${options.project}` )
+  debug( `cutoff=${cutoff.toISOString()}` )
+
+  const snapshots = await fetchSnapshotsOlderThan( {
+    compute,
+    projectId : options.project,
+    cutoff,
+  } )
+
+  // Always show what we're about to act on.
+  await printSnapshotTable( { snapshots, ageDays, cutoff } )
+
+  if ( options.delete ) {
+    await deleteSnapshots( {
+      compute,
+      projectId : options.project,
+      snapshots,
+    } )
+  }
+}
+
+await main()