2.0

Author: zmh-program

README.md

@@ -57,5 +57,5 @@ https://github.com/zmh-program/code-statistic/blob/a3841c2eefcda8484f94f56724d8d
 
 ### 🍇 Run
 ```shell
-node index
+ts-node index
 ```

index.ts

@@ -1,42 +1,3 @@
-import {isExistRepo} from "./utils";
+const server = require("./server");
 
-const logger = (require("log4js")).getLogger("Backend");
-const stats = require('./stats');
-const conf = require('./config');
-const utils = require('./utils');
-const app = require('express')();
-
-logger.level = "debug";
-app.set('views',__dirname + '/views');
-app.set("view engine","ejs");
-
-app.get('/', function(req: any, res: any) {
-    res.render('index');
-})
-
-app.get('/user/:user/', async function (req: any, res: any) {
-  res.type('svg');
-
-  try {
-    const username = req.params['user'];
-    if (await utils.isAuthenticated(username)) throw new Error();
-    res.render('user', await stats.getAccount(username, req.query['theme'] === 'dark'));
-  } catch (e) {
-    res.render('error', {dark: req.query['theme'] === 'dark'});
-  }
-});
-
-app.get('/repo/:user/:repo/', async function (req: any, res: any) {
-  res.type('svg');
-  try {
-    const username = req.params['user'], repo = req.params['repo'];
-
-    if (await isExistRepo(username, repo)) throw new Error();
-    res.render('repo', await stats.getRepository(username, repo, req.query['theme'] === 'dark'))
-  } catch (e) {
-    res.render('error', {dark: req.query['theme'] === 'dark'});
-  }
-});
-
-app.listen(conf.port, () =>
-  logger.info(`Starting deployment server at http://127.0.0.1:${conf.port}/.`));
+server.createServer();

server.ts

@@ -0,0 +1,64 @@
+const logger = (require("log4js")).getLogger("Backend");
+const stats = require('./stats');
+const conf = require('./config');
+const utils = require('./utils');
+const express = require('express');
+
+logger.level = "debug";
+
+
+function renderError(res: any, dark: boolean = false): void {
+    return res.render('error', {dark: dark});
+}
+
+async function renderUser(res: any, username: string, dark: boolean = false): Promise<void> {
+    if (! await utils.isAuthenticated(username)) {
+        renderError(res, dark);
+    } else {
+        const resp = await stats.analyseUser(username);
+        resp['dark'] = dark;
+        res.render('user', resp);
+    }
+}
+
+async function renderRepo(res: any, username: string, repo: string, dark: boolean = false): Promise<void> {
+    if (! await utils.isExistRepo(username, repo)) {
+        renderError(res, dark);
+    } else {
+        const resp = await stats.analyseRepo(username, repo);
+        resp['dark'] = dark;
+        res.render('repo', resp);
+    }
+}
+export function createServer(): void {
+    const app = express();
+
+    app.set('views', __dirname + '/views');
+    app.set("view engine", "ejs");
+
+    app.get('/', function (req: any, res: any) {
+        res.render('index');
+    });
+
+    app.get('/user/:user/', async function (req: any, res: any) {
+        res.type('svg');
+
+        const dark: boolean = req.query['theme'] === 'dark',
+            username: string = req.params['user'];
+
+        try { await renderUser(res, username, dark) } catch { renderError(res, dark) }
+    });
+
+    app.get('/repo/:user/:repo/', async function (req: any, res: any) {
+        res.type('svg');
+
+        const dark: boolean = req.query['theme'] === 'dark',
+            username = req.params['user'],
+            repo = req.params['repo'];
+
+        try { await renderRepo(res, username, repo, dark) } catch { renderError(res, dark) }
+    });
+
+    app.listen(conf.port, () =>
+        logger.info(`Starting deployment server at http://127.0.0.1:${conf.port}/.`));
+}

stats.ts

@@ -28,11 +28,10 @@ function formatter(langs: object): object[] {
     }});
 }
 
-export async function getAccount(username: string, dark: boolean = false) {
+async function _analyseUser(username: string) {
   const response = await utils.requestUser(username);
   const repos = await utils.listRepos(username);
   return {
-    dark: dark,
     org: response['type'] !== 'User',
     location: response['location'],  //@ts-ignore
     stars: utils.decConvert(utils.sum(repos.map(repo => repo['stargazers_count']))),  //@ts-ignore
@@ -48,11 +47,11 @@ export async function getAccount(username: string, dark: boolean = false) {
       })),
   };
 }
+export const analyseUser = cache.wrap(_analyseUser);
 
-export async function getRepository(username: string, repo: string, dark: boolean = false) {
+async function _analyseRepo(username: string, repo: string) {
   const res = await utils.requestRepo(username, repo);
   return {
-    dark : dark,
     username: username,
     repo: repo,
     size: utils.storeConvert(res['size'], 1),
@@ -64,3 +63,4 @@ export async function getRepository(username: string, repo: string, dark: boolea
     langs: formatter(await utils.requestLanguage(username, repo)),
   };
 }
+export const analyseRepo = cache.wrap(_analyseRepo);