Added support to delete codespaces just like repositories

index.js

@@ -16,27 +16,49 @@ async function main() {
       Config.save(token);
     }
 
-    const repositories = await UI.getRepositories();
-    if (!repositories) {
-      Config.deleteFile();
-      return await main();
-    }
+    if (process.argv[2] == 'codespaces') {
+      const codespaces = await UI.getCodespaces();
 
-    let res = await UI.promptSelectRepositories(repositories);
-    if (res.repos.length === 0) {
-      UI.printNoReposSelected();
+      let res = await UI.promptSelectCodespaces(codespaces);
 
-      return 0;
-    }
+      if (res.codespaces.length === 0) {
+        UI.printNoCodespaceSelected();
+
+        return 0;
+      }
 
-    const reposToDelete = res.repos;
-    const repoCount = reposToDelete.length;
-    res = await UI.promptConfirmDelete(repoCount);
+      const codespacesToDelete = res.codespaces;
+      const codespaceCount = codespacesToDelete.length;
+      res = await UI.promptConfirmDelete(codespaceCount, 'codespaces');
 
-    if (res.confirmDelete === 'Yes') {
-      await UI.deleteRepositories(reposToDelete);
+      if (res.confirmDelete === 'Yes') {
+        await UI.deleteCodespaces(codespacesToDelete);
+      } else {
+        UI.printNoReposDeleted();
+      }
     } else {
-      UI.printNoReposDeleted();
+      const repositories = await UI.getRepositories();
+      if (!repositories) {
+        Config.deleteFile();
+        return await main();
+      }
+
+      let res = await UI.promptSelectRepositories(repositories);
+      if (res.repos.length === 0) {
+        UI.printNoReposSelected();
+
+        return 0;
+      }
+
+      const reposToDelete = res.repos;
+      const repoCount = reposToDelete.length;
+      res = await UI.promptConfirmDelete(repoCount, 'repos');
+
+      if (res.confirmDelete === 'Yes') {
+        await UI.deleteRepositories(reposToDelete);
+      } else {
+        UI.printNoReposDeleted();
+      }
     }
   } catch (error) {
     if (error instanceof Github.AuthError || error instanceof Github.ScopesError) {

src/config.js

@@ -45,7 +45,9 @@ function getConfigDir(homeDir) {
     homeDir,
     process.platform === 'win32'
       ? path.join('AppData', 'Roaming', PACKAGE_AUTHOR, PACKAGE.name)
-      : path.join('Library', `com.${PACKAGE_AUTHOR}.${PACKAGE.name}`)
+      : process.platform == 'darwin'
+      ? path.join('Library', `com.${PACKAGE_AUTHOR}.${PACKAGE.name}`)
+      : path.join('.config', `com.${PACKAGE_AUTHOR}.${PACKAGE.name}`)
   );
 
   return configDir;

src/github.js

@@ -5,7 +5,7 @@ const CLIENT_ID_PROD = 'ed7c193c5b64ee06192a';
 
 const CLIENT_ID = process.env.DEV ? process.env.CLIENT_ID : CLIENT_ID_PROD;
 const CLIENT_TYPE = 'oauth-app';
-const CLIENT_SCOPES = ['delete_repo', 'repo'];
+const CLIENT_SCOPES = ['delete_repo', 'repo', 'codespace'];
 const API_PAGINATION = 100;
 const API_AFFILIATION = 'owner, collaborator';
 
@@ -40,6 +40,24 @@ async function getRepositories() {
   return repos;
 }
 
+async function getCodespaces() {
+  let page = 1;
+
+  const codespaces = [];
+
+  while (true) {
+    const res = await apiCall('GET', '/user/codespaces', page);
+    const codespacesCurrentPage = res.data.codespaces;
+
+    if (codespacesCurrentPage.length === 0) break;
+
+    codespaces.push(...codespacesCurrentPage);
+    page++;
+  }
+
+  return codespaces;
+}
+
 function checkPermissions(authScopes, clientScopes) {
   if (authScopes.length < clientScopes.length) {
     return false;
@@ -58,6 +76,14 @@ async function deleteRepository(repository) {
   return true;
 }
 
+async function deleteCodespace(codespace) {
+  const res = await apiCall('DELETE', `/user/codespaces/${codespace}`);
+
+  if (res.status !== 204) return false;
+
+  return true;
+}
+
 function getAuthHeader() {
   return `token ${process.env.GITHUB_TOKEN}`;
 }
@@ -110,8 +136,10 @@ class ScopesError extends Error {
 module.exports = {
   auth,
   getRepositories,
+  getCodespaces,
   checkPermissions,
   deleteRepository,
+  deleteCodespace,
   setToken,
   AuthError,
   ScopesError,

src/ui.js

@@ -69,6 +69,25 @@ async function promptSelectRepositories(repositories) {
   }
 }
 
+async function promptSelectCodespaces(codespaces) {
+  try {
+    if (codespaces.length === 0) throw error;
+
+    return await prompt({
+      type: 'autocomplete',
+      name: 'codespaces',
+      message: 'Select codespaces you want to delete:',
+      limit: 12,
+      multiple: true,
+      footer: '––—————––—————––—————––—————––————————————',
+      format: (value) => style.green(value),
+      choices: codespaces.map(({ name }) => name),
+    });
+  } catch (error) {
+    return { codespaces: [] };
+  }
+}
+
 async function getRepositories() {
   const strMessage = `Fetching repositories…`;
   const spinner = ora(strMessage).start();
@@ -90,12 +109,39 @@ async function getRepositories() {
   }
 }
 
+async function getCodespaces() {
+  const strMessage = `Fetching codespaces…`;
+  const spinner = ora(strMessage).start();
+
+  try {
+    const codespaces = await Github.getCodespaces();
+
+    const count = codespaces.length;
+    const strSucceed = printCodespacesFound(count);
+    spinner.succeed(style.dim(strSucceed));
+
+    return codespaces;
+  } catch (error) {
+    spinner.stop();
+
+    if (error instanceof Github.AuthError || error instanceof Github.ScopesError) {
+      throw error;
+    }
+  }
+}
+
 function printReposFound(count) {
   const strMessage = `${count} ${count > 1 ? 'repositories' : 'repository'} found.`;
 
   return strMessage;
 }
 
+function printCodespacesFound(count) {
+  const strMessage = `${count} ${count > 1 ? 'codespaces' : 'codespace'} found.`;
+
+  return strMessage;
+}
+
 async function deleteRepositories(repositories) {
   const deletedRepos = [];
 
@@ -115,13 +161,38 @@ async function deleteRepositories(repositories) {
   }
 
   if (deletedRepos.length > 0) {
-    printConfirmDelete(deletedRepos);
+    printConfirmDelete(deletedRepos, 'repos');
   } else {
     printNoReposDeleted();
   }
 }
 
-async function promptConfirmDelete(repoCount) {
+async function deleteCodespaces(codespaces) {
+  const deletedCodespaces = [];
+
+  for (const codespace of codespaces) {
+    const spinner = ora().start();
+
+    try {
+      await Github.deleteCodespace(codespace);
+      deletedCodespaces.push(codespace);
+
+      spinner.stopAndPersist({ symbol: '', text: style.strikethrough.dim(codespace) });
+    } catch (error) {
+      const message = error.response?.data?.message;
+
+      spinner.fail(style.dim(`${codespace} (Oops! ${message})`));
+    }
+  }
+
+  if (deletedCodespaces.length > 0) {
+    printConfirmDelete(deletedCodespaces, 'codespaces');
+  } else {
+    printNoCodespacesDeleted();
+  }
+}
+
+async function promptConfirmDelete(count, type) {
   return await prompt({
     type: 'select',
     name: 'confirmDelete',
@@ -131,7 +202,15 @@ async function promptConfirmDelete(repoCount) {
       {
         name: 'Yes',
         message: `${style.redBright(
-          `Yes, delete ${repoCount > 1 ? 'repositories' : 'repository'} (${repoCount})`
+          `Yes, delete ${
+            count > 1
+              ? type === 'repos'
+                ? 'repositories'
+                : 'codespaces'
+              : type === 'repos'
+              ? 'repository'
+              : 'codespace'
+          } (${count})`
         )}`,
         value: true,
       },
@@ -144,16 +223,23 @@ async function promptConfirmDelete(repoCount) {
   });
 }
 
-function printConfirmDelete(deletedRepos) {
-  const count = deletedRepos.length;
-
-  const strDeletedRepos = count > 1 ? deletedRepos.join(', ') : deletedRepos;
-  const strRepos = count > 1 ? 'repositories' : 'repository';
-  const strConfirm = `🔫 pew pew! ${count} ${strRepos} deleted successfully: ${strDeletedRepos}`;
+function printConfirmDelete(deletedItems, type) {
+  const count = deletedItems.length;
+
+  const strDeletedItems = count > 1 ? deletedItems.join(', ') : deletedItems;
+  const strItems =
+    count > 1
+      ? type === 'repos'
+        ? 'repositories'
+        : 'codespaces'
+      : type === 'repos'
+      ? 'repository'
+      : 'codespace';
+  const strConfirm = `🔫 pew pew! ${count} ${strItems} deleted successfully: ${strDeletedItems}`;
   const strRecover = `Recover repositories from github.com/settings/repositories`;
 
   console.log(strConfirm);
-  console.log(style.dim(strRecover));
+  type === 'repos' && console.log(style.dim(strRecover));
 
   return true;
 }
@@ -164,12 +250,24 @@ function printNoReposDeleted() {
   return console.log(style.dim(strMessage));
 }
 
+function printNoCodespacesDeleted() {
+  const strMessage = `Rest assured, no codespaces were deleted.`;
+
+  return console.log(style.dim(strMessage));
+}
+
 function printNoReposSelected() {
   const strMessage = `No repositories selected.`;
 
   return console.log(style.dim(strMessage));
 }
 
+function printNoCodespaceSelected() {
+  const strMessage = `No codespaces selected.`;
+
+  return console.log(style.dim(strMessage));
+}
+
 function printError(strError) {
   console.log();
   return console.log(style.redBright(strError));
@@ -179,10 +277,15 @@ module.exports = {
   printWelcome,
   promptAuth,
   getRepositories,
+  getCodespaces,
   promptSelectRepositories,
+  promptSelectCodespaces,
   deleteRepositories,
+  deleteCodespaces,
   promptConfirmDelete,
   printNoReposDeleted,
   printNoReposSelected,
+  printNoCodespacesDeleted,
+  printNoCodespaceSelected,
   printError,
 };