Add command to disable all projects for a given user

TailorDev · Apr 26, 2018 · 1f670e3 · 1f670e3
1 parent f46c4bd
commit 1f670e3
Show file tree

Hide file tree

Showing 3 changed files with 56 additions and 10 deletions.
diff --git a/bin/assignees b/bin/assignees
@@ -13,6 +13,7 @@ const findReviewersTask = require('../tasks/findReviewers');
 const listUserFeaturesTask = require('../tasks/listUserFeatures');
 const updateUserFeatureTask = require('../tasks/updateUserFeature');
 const disableProjectTask = require('../tasks/disableProject');
+const disableProjectsTask = require('../tasks/disableProjects');
 
 const updateUserFeatureAction = operation => async (username, feature) => {
   const updateUserFeature = updateUserFeatureTask.configure({ logger });
@@ -98,7 +99,24 @@ program
     const disableProject = disableProjectTask.configure({ logger });
 
     try {
-      await disableProject(owner, repo);
+      await disableProject({ owner, repo });
+    } catch(e) {
+      logger.error(chalk.red(e.stack));
+      process.exitCode = 1;
+    } finally {
+      mongoose.connection.close();
+    }
+  })
+;
+
+program
+  .command('project:disable-all [owner]')
+  .description('disable all projects of a user or organization completely')
+  .action(async (owner, repo) => {
+    const disableProjects = disableProjectsTask.configure({ logger });
+
+    try {
+      await disableProjects(owner);
     } catch(e) {
       logger.error(chalk.red(e.stack));
       process.exitCode = 1;

diff --git a/tasks/disableProject.js b/tasks/disableProject.js
@@ -7,15 +7,17 @@ const User = require('../models/User');
  *   logger: { info: Function, error: Function },
  * }
  */
-exports.configure = config => async (owner, repo) => {
-  const repository = await Repository.findOne({
-    name: repo,
-    owner,
-  })
+exports.configure = config => async ({ owner, repo, repository }) => {
+  if (!repository) {
+    repository = await Repository.findOne({
+      name: repo,
+      owner,
+    })
     .catch(() => null);
+  }
 
   if (!repository) {
-    config.logger.error(`No project found for owner = "${owner}" and repo = "${repo}".`);
+    config.logger.error(`No project found for owner = "${repository.owner}" and repo = "${repository.name}".`);
     return;
   }
 
@@ -38,8 +40,8 @@ exports.configure = config => async (owner, repo) => {
     }
 
     await gh.auth(user).repos.deleteHook({
-      owner,
-      repo,
+      owner: repository.owner,
+      repo: repository.name,
       id: repository.github_hook_id,
     });
 
@@ -52,5 +54,5 @@ exports.configure = config => async (owner, repo) => {
     return;
   }
 
-  config.logger.info(`${owner}/${repo} successfully disabled.`);
+  config.logger.info(`${repository.owner}/${repository.name} successfully disabled.`);
 };
diff --git a/tasks/disableProjects.js b/tasks/disableProjects.js
@@ -0,0 +1,26 @@
+const Repository = require('../models/Repository');
+const disableProjectTask = require('./disableProject');
+
+/**
+ * config = {
+ *   logger: { info: Function, error: Function },
+ * }
+ */
+exports.configure = config => async (owner) => {
+  config.logger.info(`Disabling all enabled projects for "${owner}"`);
+
+  const disableProject = disableProjectTask.configure(config);
+
+  const repositories = await Repository.find({
+    owner,
+    enabled: true,
+  });
+
+  await Promise.all(
+    repositories.map(async (repository) => {
+      config.logger.info(`-> disabling "${repository.name}"`);
+
+      await disableProject({ repository });
+    })
+  );
+};