upcore-tcp 0.0.1

package/lib/routeContext.js

@@ -0,0 +1,191 @@
+const fs = require("fs");
+const checkDirectory = require('./checkDirectory');
+const extension = require('./extension');
+
+module.exports = routeContext = class{
+    setup = ['next', 'all', 'websocket', 'directory', 'routes', 'get','post','put','patch','delete','options', 'head']
+    constructor(app, domain, core = []){
+        this.app = app;
+        this.domain = domain;
+        this.core = core;
+    }
+
+    next(...cb){
+        let core = [];
+
+        for(let x of cb){
+            if(typeof x === 'function'){
+                let id = `__cb_${++this.app.CALLBACK_ID}`;
+
+                this.app.CALLBACK_TREE[id] = x;
+                core.push(id);
+                
+            }else if(typeof x === 'string'){
+                core.push(x);
+            }
+        }
+        
+        return new routeContext(
+            this.app,
+            this.domain,
+            [...this.core, ...core]
+        );
+    }
+
+    all(path, ...cb){
+        this.app.set(this.domain, '*', path, ...[...this.core, ...cb]);
+        return this;
+    }
+    websocket(path, ...cb){
+        let mode = {
+            start:cb[0] || (()=>{}),
+            data:cb[1] || (()=>{}),
+            close:cb[2] || (()=>{}),
+            ping:cb[3] || (()=>{}),
+            pong:cb[4] || (()=>{}),
+        };
+        
+        ['start', 'data', 'close', 'ping', 'pong'];
+        this.app.set(this.domain, 'websocket', path, ...[...this.core, cb.length == 1 ? cb : (req, m)=>{
+            if(mode[m]){
+                mode[m](req);
+            }
+        }]);
+        return this;
+    }
+    directory(dir, options){
+        let {pathname, type, callback, error} = options || {};
+
+        if(typeof pathname !== 'string'){
+            pathname = '/';
+        }else if(pathname[pathname.length-1] != '/'){
+            pathname += '/';
+        }
+
+        if(dir != '' && dir[dir.length-1] != '/'){
+            dir += '/';
+        }
+
+        if(typeof callback !== 'function'){
+            callback = (req, {load})=>{
+                req.file({
+                    file:load,
+                    error:(code)=>{
+                        error(req, code, '');
+                    }
+                });
+            }
+        }
+
+        if(typeof error !== 'function'){
+            error = (req, code, text)=>{
+                req.code(code).send(text);
+            }
+        }
+
+        let checkType = false;
+        if(Array.isArray(type)){
+            checkType = true;
+        }
+
+        this.app.set(this.domain, 'get', `${pathname}*`, ...[...this.core, (req)=>{
+            let file = checkDirectory(req.wildcard);
+            let load = `${dir}${file}`;
+            let ext = extension(file);
+
+            if(ext === false || (checkType && !type.includes(ext))){
+                error(req, 400, 'The file extension is incorrect.');
+                return false;
+            }
+            if(file !== false && fs.existsSync(load)){
+                callback(req, {load, file});
+            }else{
+                error(req, 404, 'File not found.');
+            }
+        }]);
+
+        return this;
+    }
+    routes(dir){
+        if(fs.lstatSync(dir).isDirectory()){
+            let path = `${dir}/`;
+            let list = {};
+            fs.readdirSync(path).forEach((file)=>{
+                if(fs.existsSync(path + file)){
+                    let ext = file.split('.').pop();
+                    if(ext == 'js'){
+                        let pathFile = path + file;
+                        list[file.slice(0,-3)] = require(pathFile);
+                    }
+                }
+            });
+
+            for(let x in list){
+                let db = x.split('@');
+                let method = "get";
+                let path = '';
+
+                if(db.length == 1){
+                    path = db[0];
+                }else if(db.length == 2){
+                    method = db[0];
+                    path = db[1];
+                }else{
+                    break;
+                }
+
+                path = path.replace(/[\-\$\"]/g,(a)=>{
+                    let list={
+                        '-':'/',
+                        '$':':'
+                    };
+                    if(list.hasOwnProperty(a)){
+                        return list[a];
+                    }else{
+                        return '';
+                    }
+                });
+
+                if(this[method]){
+                    let r = [];
+                    if(Array.isArray(list[x].next)){
+                        r.push(...list[x].next);
+                    }else if(typeof list[x].next == 'string'){
+                        r.push(list[x].next);
+                    }
+                    r.push(list[x]);
+                    this[method](`/${path}`, ...r);
+                }
+            }
+        }
+        return this;
+    }
+    get(path, ...cb){
+        this.app.set(this.domain, 'get', path, ...[...this.core, ...cb]);
+        return this;
+    }
+    post(path, ...cb){
+        this.app.set(this.domain, 'post', path, ...[...this.core, ...cb]);
+        return this;
+    }
+    put(path, ...cb){
+        this.app.set(this.domain, 'put', path, ...[...this.core, ...cb]);
+        return this;
+    }
+    patch(path, ...cb){
+        this.app.set(this.domain, 'patch', path, ...[...this.core, ...cb]);
+        return this;
+    }
+    delete(path, ...cb){
+        this.app.set(this.domain, 'delete', path, ...[...this.core, ...cb]);
+        return this;
+    }
+    options(path, ...cb){
+        this.app.set(this.domain, 'options', path, ...[...this.core, ...cb]);
+        return this;
+    }
+    head(path, ...cb){
+        this.app.set(this.domain, 'head', path, ...[...this.core, ...cb]);
+        return this;
+    }
+}