10 changed files with 1470 additions and 1097 deletions
File diff suppressed because it is too large
@ -0,0 +1,190 @@ |
|||||
|
// @ts-nocheck
|
||||
|
|
||||
|
import { walkDir, removeIndex, isIndexEnd } from "./util" |
||||
|
import * as Joi from "joi" |
||||
|
const path = require("path") |
||||
|
const fs = require("fs") |
||||
|
|
||||
|
class routePlugin { |
||||
|
public name: string = "routePlugin" |
||||
|
public version: string = "0.0.1" |
||||
|
public register(server: any, opts: any) { |
||||
|
const sourceDir = opts.sourceDir |
||||
|
const type = opts.type || "jwt" |
||||
|
const auth = opts.auth || [] |
||||
|
let array = [] |
||||
|
for (let i = 0; i < sourceDir.length; i++) { |
||||
|
const dir = sourceDir[i] |
||||
|
console.log(dir) |
||||
|
array.push(dir.dir + "对应路径：") |
||||
|
array = array.concat(this.registerRoute(server, dir.dir, dir.prefix || "", auth, type)) |
||||
|
} |
||||
|
fs.writeFileSync(path.resolve(process.cwd(), "route.txt"), array.join("\n"), { |
||||
|
encoding: "utf-8", |
||||
|
}) |
||||
|
} |
||||
|
registerRoute(server, sourceDir, prefix, auth, type) { |
||||
|
const files = walkDir(sourceDir) |
||||
|
const routes = [] |
||||
|
files.forEach(file => { |
||||
|
let filename = file.relativeFileNoExt |
||||
|
let array = filename.split(path.sep).slice(1) |
||||
|
let fileNoExt = removeIndex("/" + array.join("/")) |
||||
|
const moduleName = path.resolve(sourceDir, filename) |
||||
|
const obj = require(moduleName) |
||||
|
if (obj.default) { |
||||
|
const func = new (obj.default || obj)() |
||||
|
const prototype = Object.getPrototypeOf(func) |
||||
|
const keys = Reflect.ownKeys(prototype) |
||||
|
for (const key of keys) { |
||||
|
if (key !== "constructor") { |
||||
|
let ff = func[key] |
||||
|
let handler: () => void = undefined |
||||
|
// 默认方法
|
||||
|
const method = ff.$method || "GET" |
||||
|
// 路由收集规则
|
||||
|
let route = "" |
||||
|
if (ff.$route) { |
||||
|
if (isIndexEnd(fileNoExt)) { |
||||
|
route = ff.$route |
||||
|
} else { |
||||
|
route = fileNoExt + ff.$route |
||||
|
} |
||||
|
} else { |
||||
|
if (isIndexEnd(fileNoExt)) { |
||||
|
route = fileNoExt + key.toString() |
||||
|
} else { |
||||
|
route = fileNoExt + "/" + key.toString() |
||||
|
} |
||||
|
} |
||||
|
route = removeIndex(route) |
||||
|
route = prefix ? route[0] + prefix + "/" + route.slice(1) : route |
||||
|
// 配置规则
|
||||
|
const options = ff.$options ? ff.$options : {} |
||||
|
if (!options.auth) { |
||||
|
if (ff.$auth == undefined) { |
||||
|
if (auth && auth.length && auth.filter(v => route.startsWith(v)).length) { |
||||
|
options.auth = type |
||||
|
} else { |
||||
|
options.auth = false |
||||
|
} |
||||
|
} else if (ff.$auth) { |
||||
|
options.auth = |
||||
|
typeof ff.$auth === "boolean" |
||||
|
? type |
||||
|
: { |
||||
|
strategy: type, |
||||
|
mode: ff.$auth, |
||||
|
} |
||||
|
} else { |
||||
|
options.auth = false |
||||
|
} |
||||
|
} |
||||
|
if (!options.validate) { |
||||
|
let validateObj = ff.$validate || {} |
||||
|
if (options.auth && type === "jwt") { |
||||
|
if (validateObj.headers) { |
||||
|
validateObj.headers = validateObj.headers.keys({ |
||||
|
Authorization: Joi.string(), |
||||
|
}) |
||||
|
} else { |
||||
|
validateObj.headers = Joi.object({ |
||||
|
headers: Joi.object({ |
||||
|
Authorization: Joi.string(), |
||||
|
}).unknown(), // 注意加上这个
|
||||
|
}) |
||||
|
} |
||||
|
} |
||||
|
if (validateObj && !!Object.keys(validateObj).length) { |
||||
|
const failReason = validateObj.failReason |
||||
|
delete validateObj.failReason |
||||
|
if (validateObj.failAction === "log") { |
||||
|
if (!options.log) options.log = {} |
||||
|
options.log.collect = true |
||||
|
let errto = validateObj.$errto |
||||
|
handler = async function (...argus) { |
||||
|
const request = argus[0] |
||||
|
const h = argus[1] |
||||
|
if (request.logs && !!request.logs.length && errto) { |
||||
|
// request.yar.flash('error', request.logs.map((v: any)=>v.error.message));
|
||||
|
request.yar.flash("error", failReason || request.logs.map((v: any)=>v.error.message)) |
||||
|
return h.redirect(errto) |
||||
|
} |
||||
|
return await ff.call(this, ...argus) |
||||
|
} |
||||
|
} |
||||
|
if (validateObj.failAction === "function") { |
||||
|
let errto = validateObj.$errto |
||||
|
validateObj.failAction = async function (request, h, err) { |
||||
|
if (err.details) { |
||||
|
request.$joi_error = err.details.map(v => v.message) |
||||
|
} |
||||
|
return h.continue |
||||
|
} |
||||
|
handler = async function (...argus) { |
||||
|
const request = argus[0] |
||||
|
const h = argus[1] |
||||
|
if (request.$joi_error) { |
||||
|
loggerSite.debug("传输参数错误: ", request.$joi_error) |
||||
|
request.yar.flash("error", failReason || request.$joi_error) |
||||
|
delete request.$joi_error |
||||
|
return h.redirect(errto) |
||||
|
} |
||||
|
return await ff.call(this, ...argus) |
||||
|
} |
||||
|
} |
||||
|
options.validate = validateObj |
||||
|
} |
||||
|
} |
||||
|
// && route.startsWith("/api")
|
||||
|
if (ff.$swagger) { |
||||
|
options.description = ff.$swagger[0] |
||||
|
options.notes = ff.$swagger[1] |
||||
|
options.tags = ff.$swagger[2] |
||||
|
} |
||||
|
let str = route |
||||
|
if ( |
||||
|
(typeof options.auth === "string" && options.auth) || |
||||
|
(typeof options.auth === "object" && options.auth.mode === "required") |
||||
|
) { |
||||
|
str = " 需要权限 ： " + " " + full(method) + " " + str |
||||
|
} else if (typeof options.auth === "object" && options.auth.mode === "optional") { |
||||
|
str = " 不需权限(提供即需验证)： " + " " + full(method) + " " + str |
||||
|
} else if (typeof options.auth === "object" && options.auth.mode === "try") { |
||||
|
str = " 不需权限(提供无需验证)： " + " " + full(method) + " " + str |
||||
|
} else { |
||||
|
str = " 不需权限 ： " + " " + full(method) + " " + str |
||||
|
} |
||||
|
routes.push(str) |
||||
|
|
||||
|
if (options.validate && options.validate.$errto) { |
||||
|
delete options.validate.$errto |
||||
|
} |
||||
|
if (!handler) { |
||||
|
handler = ff |
||||
|
} |
||||
|
server.route({ |
||||
|
method: method, |
||||
|
path: route, |
||||
|
handler: handler, |
||||
|
options: options, |
||||
|
}) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
}) |
||||
|
return routes |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
function full(str: string, length = 10) { |
||||
|
let len = str.length |
||||
|
let need = length - len |
||||
|
if (need <= 0) return str |
||||
|
return str + [...Array(need)].map((v, i) => " ").join("") |
||||
|
} |
||||
|
|
||||
|
const plugin = new routePlugin() |
||||
|
|
||||
|
export { plugin } |
||||
|
export * from "./util/decorators" |
||||
@ -0,0 +1,40 @@ |
|||||
|
// @ts-nocheck
|
||||
|
/** |
||||
|
* 方法 |
||||
|
* @param opts 参数 |
||||
|
*/ |
||||
|
type TMethod = "GET" | "POST" | "PUT" | "DELETE" |
||||
|
export function method(opts?: TMethod | Array<TMethod>) { |
||||
|
return function (target, propertyKey: string, descriptor: PropertyDescriptor) { |
||||
|
target[propertyKey].$method = opts |
||||
|
} |
||||
|
} |
||||
|
export function route(route?: string) { |
||||
|
return function (target, propertyKey: string, descriptor: PropertyDescriptor) { |
||||
|
target[propertyKey].$route = route |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
export function config(options: Object) { |
||||
|
return function (target, propertyKey: string, descriptor: PropertyDescriptor) { |
||||
|
target[propertyKey].$options = options |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
export function auth(isAuth: boolean | "try" | "required" | "optional" = true) { |
||||
|
return function (target, propertyKey: string, descriptor: PropertyDescriptor) { |
||||
|
target[propertyKey].$auth = isAuth |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
export function validate(validate: Object) { |
||||
|
return function (target, propertyKey: string, descriptor: PropertyDescriptor) { |
||||
|
target[propertyKey].$validate = validate |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
export function swagger(desc, notes, tags) { |
||||
|
return function (target, propertyKey: string, descriptor: PropertyDescriptor) { |
||||
|
target[propertyKey].$swagger = [desc, notes, tags] |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,81 @@ |
|||||
|
// @ts-nocheck
|
||||
|
|
||||
|
const path = require("path") |
||||
|
const fs = require("fs") |
||||
|
|
||||
|
export function removeIndex(ss: any) { |
||||
|
const remove = (str: any) => { |
||||
|
if (str.endsWith("/index")) { |
||||
|
return str.slice(0, -6) |
||||
|
} |
||||
|
if (str.endsWith("index")) { |
||||
|
return str.slice(0, -5) |
||||
|
} |
||||
|
return str ? str : "/" |
||||
|
} |
||||
|
let r = true |
||||
|
let rr = ss |
||||
|
while (r) { |
||||
|
if (rr.endsWith("/index")) { |
||||
|
rr = remove(rr) |
||||
|
} else { |
||||
|
r = false |
||||
|
} |
||||
|
} |
||||
|
return rr ? rr : "/" |
||||
|
} |
||||
|
|
||||
|
export function isIndexEnd(str: any) { |
||||
|
return str.length == 1 && str.endsWith("/") |
||||
|
} |
||||
|
|
||||
|
export function walkDir( |
||||
|
filePath: any, |
||||
|
exclude = ["node_modules", "^_", ".git", ".idea", ".gitignore", "client", ".txt$", ".test.js$", ".test.ts$"], |
||||
|
) { |
||||
|
let files: any[] = [] |
||||
|
function Data(opts: any) { |
||||
|
this.relativeDir = opts.relativeDir |
||||
|
this.relativeFile = opts.relativeFile |
||||
|
this.filename = opts.filename |
||||
|
this.file = opts.file |
||||
|
this.absoluteFile = opts.absoluteFile |
||||
|
this.relativeFileNoExt = opts.relativeFileNoExt |
||||
|
this.absoluteDir = opts.absoluteDir |
||||
|
} |
||||
|
function readDir(filePath, dirname = ".") { |
||||
|
let res = fs.readdirSync(filePath) |
||||
|
res.forEach(filename => { |
||||
|
const filepath = path.resolve(filePath, filename) |
||||
|
const stat = fs.statSync(filepath) |
||||
|
const name = filepath.split(path.sep).slice(-1)[0] |
||||
|
if (typeof exclude === "string" && new RegExp(exclude).test(name)) { |
||||
|
return |
||||
|
} |
||||
|
if (Array.isArray(exclude)) { |
||||
|
for (let i = 0; i < exclude.length; i++) { |
||||
|
const excludeItem = exclude[i] |
||||
|
if (new RegExp(excludeItem).test(name)) { |
||||
|
return |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
if (!stat.isFile()) { |
||||
|
readDir(filepath, dirname + path.sep + name) |
||||
|
} else { |
||||
|
const data = new Data({ |
||||
|
relativeDir: dirname, |
||||
|
relativeFile: dirname + path.sep + path.parse(filepath).base, |
||||
|
relativeFileNoExt: dirname + path.sep + path.parse(filepath).name, |
||||
|
file: path.parse(filepath).base, |
||||
|
filename: path.parse(filepath).name, |
||||
|
absoluteFile: filepath, |
||||
|
absoluteDir: path.parse(filepath).dir, |
||||
|
}) |
||||
|
files.push(data) |
||||
|
} |
||||
|
}) |
||||
|
} |
||||
|
readDir(filePath) |
||||
|
return files |
||||
|
} |
||||
Loading…
Reference in new issue