@rdyl/node-koa-controller 0.1.2 → 0.1.4

package/CHANGELOG.md

@@ -1,4 +1,4 @@
-## 0.1.2 (2025-04-07)
+## 0.1.4 (2025-04-10)
 
 
 

package/dist/controller/index.js

@@ -79,43 +79,59 @@ exports.ASTCache = void 0;
 exports.useController = useController;
 var koa_router_1 = __importDefault(require("koa-router"));
 var koa_static_1 = __importDefault(require("koa-static"));
+var koa_body_1 = require("koa-body");
 var fs_1 = __importDefault(require("fs"));
-var path_1 = require("path");
 var chalk_1 = __importDefault(require("chalk"));
+var path_1 = require("path");
 var utils_1 = require("./utils");
 exports.ASTCache = {
     prefix: "",
     routes: [],
     whitelist: [],
 };
-function RegisterASTCacheRoutes(router) {
+function RegisterASTCacheRoutes(router, useJwt) {
     var _this = this;
     exports.ASTCache.routes.forEach(function (route) {
         var handlers = route.handlers, prefix = route.prefix, Instance = route.Instance;
         handlers.forEach(function (handler) {
             var method = handler.method, url = handler.url, name = handler.name;
             var path = (0, path_1.join)("/" + exports.ASTCache.prefix, prefix, url);
-            var isPublic = route.isPublic || route.isPublic;
+            var isPublic = route.isPublic || handler.isPublic;
             if (isPublic) {
                 exports.ASTCache.whitelist.push(path);
             }
             // @ts-ignore
-            router[method.toLowerCase()](path, function (_ctx, next) { return __awaiter(_this, void 0, void 0, function () {
-                var instance, _a;
+            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) {
                         case 0:
-                            _b.trys.push([0, 2, , 3]);
+                            _b.trys.push([0, 4, , 5]);
                             instance = new Instance(_ctx);
-                            return [4 /*yield*/, Promise.resolve(instance[name]())];
+                            if (!useJwt) return [3 /*break*/, 2];
+                            token = _ctx.request.headers[useJwt.name] || "";
+                            return [4 /*yield*/, useJwt.get(token)];
                         case 1:
+                            jwt = _b.sent();
+                            if (!jwt && !isPublic) {
+                                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*/, 3];
-                        case 2:
+                            return [3 /*break*/, 5];
+                        case 4:
                             _a = _b.sent();
-                            console.log(path + ":配置错误!");
-                            return [3 /*break*/, 3];
-                        case 3: return [2 /*return*/];
+                            _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*/];
                     }
                 });
             }); });
@@ -124,12 +140,12 @@ function RegisterASTCacheRoutes(router) {
 }
 function useController(app, opts) {
     return __awaiter(this, void 0, void 0, function () {
-        var root, _a, prefix, _b, logs, router, rootDir, initAST;
+        var root, _a, prefix, _b, logs, formidable, useJwt, router, rootDir, initAST;
         var _this = this;
         return __generator(this, function (_c) {
             switch (_c.label) {
                 case 0:
-                    root = opts.root, _a = opts.prefix, prefix = _a === void 0 ? "" : _a, _b = opts.logs, logs = _b === void 0 ? true : _b;
+                    root = opts.root, _a = opts.prefix, prefix = _a === void 0 ? "" : _a, _b = opts.logs, logs = _b === void 0 ? true : _b, formidable = opts.formidable, useJwt = opts.useJwt;
                     exports.ASTCache.prefix = prefix;
                     router = new koa_router_1.default();
                     rootDir = (0, path_1.resolve)(process.cwd(), root);
@@ -181,16 +197,20 @@ function useController(app, opts) {
                 case 1:
                     _c.sent();
                     // 注册路由
-                    RegisterASTCacheRoutes(router);
+                    RegisterASTCacheRoutes(router, useJwt);
                     if (opts.static) {
                         app.use((0, koa_static_1.default)(process.cwd(), {
                             root: opts.static,
                         }));
                         exports.ASTCache.whitelist.unshift(new RegExp("/".concat(opts.static)));
                     }
+                    app.use((0, koa_body_1.koaBody)({
+                        multipart: true,
+                        formidable: formidable,
+                    }));
                     if (logs) {
                         app.use(function (ctx, next) { return __awaiter(_this, void 0, void 0, function () {
-                            var start, duration, status_1, method, statusColor, dateTime;
+                            var start, duration, status_1, method, statusColor, dateTime, dText;
                             return __generator(this, function (_a) {
                                 switch (_a.label) {
                                     case 0:
@@ -205,7 +225,7 @@ function useController(app, opts) {
                                     case 3:
                                         duration = Date.now() - start;
                                         status_1 = ctx.status;
-                                        method = ctx.method.padEnd(7);
+                                        method = ctx.method;
                                         statusColor = void 0;
                                         if (status_1 >= 500)
                                             statusColor = chalk_1.default.red;
@@ -218,7 +238,14 @@ function useController(app, opts) {
                                         else
                                             statusColor = chalk_1.default.gray;
                                         dateTime = (0, utils_1.formatDateTime)();
-                                        console.log("- [".concat(dateTime, " | ").concat(chalk_1.default.gray("".concat(duration, "ms")), "] ").concat(chalk_1.default.magenta(method)).concat(ctx.url, " ").concat(statusColor(status_1)));
+                                        dText = chalk_1.default.gray("".concat(duration, "ms"));
+                                        if (duration > 5000) {
+                                            dText = chalk_1.default.red("".concat(duration, "ms"));
+                                        }
+                                        else if (duration > 1000) {
+                                            dText = chalk_1.default.yellow("".concat(duration, "ms"));
+                                        }
+                                        console.log("- [".concat(dateTime, " | ").concat(dText, "] ").concat(chalk_1.default.magenta(method), " ").concat(ctx.url, " ").concat(statusColor(status_1)));
                                         return [7 /*endfinally*/];
                                     case 4: return [2 /*return*/];
                                 }

package/dist/controller/meta.js

@@ -50,6 +50,7 @@ var HttpMethod;
 var BaseController = /** @class */ (function () {
     function BaseController(_ctx) {
         this._ctx = _ctx;
+        this.jwt = {};
     }
     Object.defineProperty(BaseController.prototype, "pathname", {
         get: function () {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rdyl/node-koa-controller",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "license": "MIT",
   "main": "dist/index.js",
   "types": "types/index.d.ts",
@@ -23,6 +23,7 @@
     "@types/koa-static": "^4.0.4",
     "chalk": "^5.4.1",
     "koa": "^2.15.3",
+    "koa-body": "^6.0.1",
     "koa-router": "^13.0.1",
     "koa-static": "^5.0.0"
   },

package/types/controller/meta.d.ts

@@ -1,5 +1,6 @@
 import { ParameterizedContext } from "koa";
 import { IRouterParamContext } from "koa-router";
+import { ExtendedFormidableOptions } from "koa-body";
 export declare enum HttpMethod {
     GET = "GET",
     POST = "POST",
@@ -9,6 +10,7 @@ export declare enum HttpMethod {
 }
 export declare class BaseController {
     _ctx: ParameterizedContext<any, IRouterParamContext<any, {}>, any>;
+    jwt: UseJwtValue;
     constructor(_ctx: ParameterizedContext<any, IRouterParamContext<any, {}>, any>);
     get pathname(): string;
     get query(): {
@@ -25,7 +27,7 @@ export declare class BaseController {
     };
     get params(): Record<string, string>;
     get body(): any;
-    files(): Promise<any>;
+    files(): Promise<import("formidable").Files | undefined>;
     get status(): {
         ok(data?: unknown): void;
         okCreated(data: unknown, msg?: string): void;
@@ -53,6 +55,11 @@ export interface UseProps {
     prefix?: string;
     static?: string;
     logs?: boolean;
+    formidable?: ExtendedFormidableOptions;
+    useJwt?: {
+        name: string;
+        get(n: string): Promise<UseJwtValue>;
+    };
 }
 export interface ASTCacheHandler {
     url: string;

package/types/type.d.ts

@@ -0,0 +1 @@
+type UseJwtValue = Record<string, unknown>;