auto-refresh every hour: bin/refresh.js

iliakan · iliakan · commit 10c6b7f2a478 · 2018-07-19T17:50:01.000+03:00
diff --git a/bin/clean.js b/bin/clean.js
@@ -0,0 +1,6 @@
+
+const config = require("../config");
+const fse = require('fs-extra');
+
+fse.removeSync(config.repoRoot);
+fse.mkdirsSync(config.repoRoot);
diff --git a/bin/init.js b/bin/init.js
diff --git a/bin/refresh.js b/bin/refresh.js
@@ -0,0 +1,25 @@
+const config = require("../config");
+const fse = require('fs-extra');
+const path = require('path');
+const run = require('../lib/run');
+const execSync = require('child_process').execSync;
+const Stats = require('../lib/stats');
+const debug = require('debug')('init');
+const updateRepo = require('../lib/updateRepo');
+
+(async function () {
+  for (let repoName in config.secret.repos) {
+    let repoConfig = config.secret.repos[repoName];
+
+    await updateRepo(repoName);
+  }
+
+  for (let repoName in config.secret.repos) {
+    let repoConfig = config.secret.repos[repoName];
+
+    if (repoConfig.lang != 'en') {
+      await Stats.instance().count(repoName);
+    }
+  }
+
+})().catch(console.error);
diff --git a/handlers/hook.js b/handlers/hook.js
@@ -1,7 +1,7 @@
 const crypto = require('crypto');
 const config = require('../config');
 const debug = require('debug')('handlers:hook');
-const onUpdateHook = require('../lib/onUpdateHook');
+const updateRepo = require('../lib/updateRepo');
 
 exports.post = async function(ctx) {
 
@@ -56,6 +56,8 @@ exports.post = async function(ctx) {
   ctx.body = {ok: true};
 
   // don't wait for it! (async)
-  onUpdateHook(ctx.request.body);
+  await updateRepo(ctx.request.body.repository.name);
+
+  await Stats.instance().count(ctx.request.body.repository.name);
 
 };
diff --git a/lib/onUpdateHook.js b/lib/onUpdateHook.js
diff --git a/lib/run.js b/lib/run.js
@@ -0,0 +1,15 @@
+
+const debug = require('debug')('run');
+const exec = require('mz/child_process').exec;
+
+module.exports = async function run(cmd, options = {}) {
+  if (!options.encoding) {
+    options.encoding = 'utf-8';
+  }
+
+  debug(cmd, options);
+  let result = await exec(cmd, options);
+  debug(result);
+
+  return result;
+};
diff --git a/lib/updateRepo.js b/lib/updateRepo.js
@@ -0,0 +1,27 @@
+const config = require("../config");
+const path = require("path");
+const run = require('./run');
+const fs = require('fs');
+
+module.exports = async function (repoName) {
+  let repoDir = path.join(config.repoRoot, path.basename(repoName));
+
+  let repoConfig = config.secret.repos[repoName];
+
+  let branch = repoConfig.branch || 'master';
+
+  if (fs.existsSync(repoDir)) {
+
+    await run(`git fetch origin`, {cwd: repoDir});
+
+  } else {
+
+    await run(`git clone --depth 1 https://github.com/${repoName}.git "${repoDir}"`);
+
+  }
+
+  await run(`git checkout -f origin/${branch}`, {cwd: repoDir});
+
+};
+
+
