@rdyl/node-koa-controller 0.1.6 → 0.1.8

package/CHANGELOG.md

@@ -1,4 +1,4 @@
-## 0.1.6 (2025-04-10)
+## 0.1.8 (2025-04-11)
 
 
 

package/dist/controller/gateway.js

@@ -0,0 +1,157 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __generator = (this && this.__generator) || function (thisArg, body) {
+    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
+    return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
+    function verb(n) { return function (v) { return step([n, v]); }; }
+    function step(op) {
+        if (f) throw new TypeError("Generator is already executing.");
+        while (g && (g = 0, op[0] && (_ = 0)), _) try {
+            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+            if (y = 0, t) op = [op[0] & 2, t.value];
+            switch (op[0]) {
+                case 0: case 1: t = op; break;
+                case 4: _.label++; return { value: op[1], done: false };
+                case 5: _.label++; y = op[1]; op = [0]; continue;
+                case 7: op = _.ops.pop(); _.trys.pop(); continue;
+                default:
+                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
+                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
+                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
+                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
+                    if (t[2]) _.ops.pop();
+                    _.trys.pop(); continue;
+            }
+            op = body.call(thisArg, _);
+        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
+        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
+    }
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.JwtSecretGateway = void 0;
+var crypto_js_1 = __importDefault(require("crypto-js"));
+var JwtTokens = /** @class */ (function () {
+    function JwtTokens(KEY, duration) {
+        this.KEY = KEY;
+        this.duration = duration;
+        this.records = [];
+    }
+    JwtTokens.prototype.encode = function (payload) {
+        var KEY = this.KEY;
+        var text = JSON.stringify(payload);
+        return crypto_js_1.default.AES.encrypt(text, KEY).toString();
+    };
+    JwtTokens.prototype.decode = function (value) {
+        try {
+            var KEY = this.KEY;
+            var bytes = crypto_js_1.default.AES.decrypt(value, KEY);
+            var result = bytes.toString(crypto_js_1.default.enc.Utf8);
+            return JSON.parse(result);
+        }
+        catch (error) {
+            return null;
+        }
+    };
+    JwtTokens.prototype.removeBy = function (t) {
+        var records = this.records;
+        for (var i = 0; i < records.length; i++) {
+            var ele = records[i];
+            if (ele.token == t) {
+                this.records.splice(i, 1);
+                break;
+            }
+        }
+    };
+    JwtTokens.prototype.find = function (t) {
+        var records = this.records;
+        var result = null;
+        for (var i = 0; i < records.length; i++) {
+            var ele = records[i];
+            if (ele.token == t) {
+                result = ele;
+                break;
+            }
+        }
+        return result;
+    };
+    JwtTokens.prototype.get = function (token) {
+        return __awaiter(this, void 0, void 0, function () {
+            var result, time;
+            return __generator(this, function (_a) {
+                result = this.find(token);
+                if (!result) {
+                    return [2 /*return*/, {
+                            stat: "none",
+                            value: null,
+                        }];
+                }
+                else {
+                    time = new Date().getTime();
+                    if (result.expired <= time) {
+                        this.removeBy(token);
+                        return [2 /*return*/, {
+                                stat: "expired",
+                                value: result.data,
+                            }];
+                    }
+                    return [2 /*return*/, {
+                            stat: "ok",
+                            value: result.data,
+                        }];
+                }
+                return [2 /*return*/];
+            });
+        });
+    };
+    JwtTokens.prototype.authorize = function (data) {
+        var token = this.encode(data);
+        var created = new Date().getTime();
+        var item = {
+            token: token,
+            data: data,
+            created: created,
+            expired: created + this.duration,
+        };
+        this.records.push(item);
+        return token;
+    };
+    return JwtTokens;
+}());
+var JwtSecretGateway = /** @class */ (function () {
+    function JwtSecretGateway(headerName, opts) {
+        this.headerName = headerName;
+        var KEY = opts.KEY, _a = opts.duration, duration = _a === void 0 ? 7 * 24 * 60 * 60 * 1000 : _a;
+        this._jwtToken = new JwtTokens(KEY, duration);
+    }
+    JwtSecretGateway.prototype.use = function (ctx) {
+        return __awaiter(this, void 0, void 0, function () {
+            var token;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        token = ctx.request.headers[this.headerName.toLowerCase()] || "";
+                        if (!(typeof token === "string")) return [3 /*break*/, 2];
+                        return [4 /*yield*/, this._jwtToken.get(token)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                    case 2: return [2 /*return*/, {
+                            stat: "none",
+                            value: null,
+                        }];
+                }
+            });
+        });
+    };
+    return JwtSecretGateway;
+}());
+exports.JwtSecretGateway = JwtSecretGateway;

package/dist/controller/index.js

@@ -102,36 +102,32 @@ function RegisterASTCacheRoutes(router, useJwt) {
             }
             // @ts-ignore
             router[method.toLowerCase()](path, function (_ctx) { return __awaiter(_this, void 0, void 0, function () {
-                var instance, token, jwt, _a;
-                return __generator(this, function (_b) {
-                    switch (_b.label) {
+                var instance, _a, stat, value, _b;
+                return __generator(this, function (_c) {
+                    switch (_c.label) {
                         case 0:
-                            _b.trys.push([0, 4, , 5]);
                             instance = new Instance(_ctx);
-                            if (!useJwt) return [3 /*break*/, 2];
-                            token = _ctx.request.headers[useJwt.name] || "";
-                            return [4 /*yield*/, useJwt.get(token)];
+                            _c.label = 1;
                         case 1:
-                            jwt = _b.sent();
-                            if (!jwt && !isPublic) {
+                            _c.trys.push([1, 5, , 6]);
+                            if (!useJwt) return [3 /*break*/, 3];
+                            return [4 /*yield*/, useJwt.use(_ctx)];
+                        case 2:
+                            _a = _c.sent(), stat = _a.stat, value = _a.value;
+                            if (stat === "none" || stat === "expired") {
                                 return [2 /*return*/, instance.status.unauthorized()];
                             }
-                            instance.jwt = jwt;
-                            _b.label = 2;
-                        case 2: return [4 /*yield*/, Promise.resolve(instance[name]())];
-                        case 3:
-                            _b.sent();
-                            return [3 /*break*/, 5];
+                            instance.jwt = value;
+                            _c.label = 3;
+                        case 3: return [4 /*yield*/, Promise.resolve(instance[name]())];
                         case 4:
-                            _a = _b.sent();
-                            _ctx.response.set("Content-Type", "application/json;charset=utf-8");
-                            _ctx.response.status = 500;
-                            _ctx.body = {
-                                code: 500,
-                                msg: "未知错误",
-                            };
-                            return [3 /*break*/, 5];
-                        case 5: return [2 /*return*/];
+                            _c.sent();
+                            return [3 /*break*/, 6];
+                        case 5:
+                            _b = _c.sent();
+                            instance.status.exception();
+                            return [3 /*break*/, 6];
+                        case 6: return [2 /*return*/];
                     }
                 });
             }); });
@@ -140,12 +136,12 @@ function RegisterASTCacheRoutes(router, useJwt) {
 }
 function useController(app, opts) {
     return __awaiter(this, void 0, void 0, function () {
-        var root, _a, prefix, _b, port, _c, logs, formidable, useJwt, router, rootDir, initAST, ip;
+        var root, _a, prefix, _b, port, _c, logs, formidable, jwt, router, rootDir, initAST, ip;
         var _this = this;
         return __generator(this, function (_d) {
             switch (_d.label) {
                 case 0:
-                    root = opts.root, _a = opts.prefix, prefix = _a === void 0 ? "" : _a, _b = opts.port, port = _b === void 0 ? 8080 : _b, _c = opts.logs, logs = _c === void 0 ? true : _c, formidable = opts.formidable, useJwt = opts.useJwt;
+                    root = opts.root, _a = opts.prefix, prefix = _a === void 0 ? "" : _a, _b = opts.port, port = _b === void 0 ? 8080 : _b, _c = opts.logs, logs = _c === void 0 ? true : _c, formidable = opts.formidable, jwt = opts.jwt;
                     exports.ASTCache.prefix = prefix;
                     router = new koa_router_1.default();
                     rootDir = (0, path_1.resolve)(process.cwd(), root);
@@ -197,7 +193,7 @@ function useController(app, opts) {
                 case 1:
                     _d.sent();
                     // 注册路由
-                    RegisterASTCacheRoutes(router, useJwt);
+                    RegisterASTCacheRoutes(router, jwt);
                     if (opts.static) {
                         app.use((0, koa_static_1.default)(process.cwd(), {
                             root: opts.static,
@@ -272,3 +268,4 @@ function useController(app, opts) {
     });
 }
 __exportStar(require("./meta"), exports);
+__exportStar(require("./gateway"), exports);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rdyl/node-koa-controller",
-  "version": "0.1.6",
+  "version": "0.1.8",
   "license": "MIT",
   "main": "dist/index.js",
   "types": "types/index.d.ts",
@@ -18,11 +18,13 @@
     "build": "tsc && copyfiles -u 1 src/**/*.d.ts src/*.d.ts types && npm run changelog && git add CHANGELOG.md"
   },
   "dependencies": {
+    "@types/crypto-js": "^4.2.2",
     "@types/koa": "^2.15.0",
     "@types/koa-router": "^7.4.8",
     "@types/koa-static": "^4.0.4",
     "boxen": "^8.0.1",
     "chalk": "^5.4.1",
+    "crypto-js": "^4.2.0",
     "koa": "^2.15.3",
     "koa-body": "^6.0.1",
     "koa-router": "^13.0.1",

package/types/controller/gateway.d.ts

@@ -0,0 +1,34 @@
+import { KoaCtx } from "./meta";
+declare class JwtTokens<T> {
+    KEY: string;
+    duration: number;
+    records: JwtTokenItem<T>[];
+    constructor(KEY: string, duration: number);
+    encode(payload: T): string;
+    decode(value: string): T | null;
+    removeBy(t: string): void;
+    find(t: string): JwtTokenItem<T> | null;
+    get(token: string): Promise<JwtResData<T>>;
+    authorize(data: T): string;
+}
+export declare class JwtSecretGateway<T extends object> {
+    headerName: string;
+    _jwtToken: JwtTokens<T>;
+    constructor(headerName: string, opts: {
+        KEY: string;
+        duration?: number;
+    });
+    use(ctx: KoaCtx): Promise<JwtResData<T>>;
+}
+export interface JwtTokenItem<T> {
+    token: string;
+    data: T;
+    expired: number;
+    created: number;
+}
+export type JwtStat = "ok" | "expired" | "none";
+export type JwtResData<T> = {
+    stat: JwtStat;
+    value: T;
+};
+export {};

package/types/controller/index.d.ts

@@ -1,5 +1,6 @@
 import Koa from "koa";
 import { ASTCacheProp, SyncUseValue, UseProps } from "./meta";
 export declare const ASTCache: ASTCacheProp;
-export declare function useController(app: Koa, opts: UseProps): Promise<SyncUseValue>;
+export declare function useController<U extends object = object>(app: Koa, opts: UseProps<U>): Promise<SyncUseValue>;
 export * from "./meta";
+export * from "./gateway";

package/types/controller/meta.d.ts

@@ -1,6 +1,8 @@
 import { ParameterizedContext } from "koa";
 import { IRouterParamContext } from "koa-router";
 import { ExtendedFormidableOptions } from "koa-body";
+import { JwtSecretGateway } from "./gateway";
+export type KoaCtx = ParameterizedContext<any, IRouterParamContext<any, {}>, any>;
 export declare enum HttpMethod {
     GET = "GET",
     POST = "POST",
@@ -9,9 +11,9 @@ export declare enum HttpMethod {
     DELETE = "DELETE"
 }
 export declare class BaseController {
-    _ctx: ParameterizedContext<any, IRouterParamContext<any, {}>, any>;
+    _ctx: KoaCtx;
     jwt: UseJwtValue;
-    constructor(_ctx: ParameterizedContext<any, IRouterParamContext<any, {}>, any>);
+    constructor(_ctx: KoaCtx);
     get pathname(): string;
     get query(): {
         values: import("querystring").ParsedUrlQuery;
@@ -50,17 +52,14 @@ export interface ResData<T> {
     msg: string;
     data?: T | null;
 }
-export interface UseProps {
+export interface UseProps<U extends object> {
     port: number;
     root: string;
     prefix?: string;
     static?: string;
     logs?: boolean;
     formidable?: ExtendedFormidableOptions;
-    useJwt?: {
-        name: string;
-        get(n: string): Promise<UseJwtValue | null>;
-    };
+    jwt?: JwtSecretGateway<U>;
 }
 export interface SyncUseValue {
     port: number;