All files / runner index.ts

96.77% Statements 30/31
77.78% Branches 7/9
100% Functions 9/9
96.77% Lines 30/31

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136                                                                                                                                                        1x 1x 1x 1x       2x 2x 2x       2x 2x       2x       2x       2x 2x   2x   2x 2x 2x       2x 2x     2x 2x 2x 2x 2x   2x 2x 1x 1x     2x         1x  
// Licensed to the Apache Software Foundation (ASF) under one or more
// contributor license agreements.  See the NOTICE file distributed with
// this work for additional information regarding copyright ownership.
// The ASF licenses this file to You under the Apache License, Version 2.0
// (the "License"); you may not use this file except in compliance with
// the License.  You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
 
interface Plugin {
    getName(): string
    parseConf(conf: string): string
    filter(conf: Object, request: Request, response: Response): Promise<void>
}
 
interface Request {
    // The id is for debug. It will be recycled when the number is exhausted
    id: number;
 
    // the path of the request.
    path: string;
 
    // The associated Headers object of the request.
    // See·https://developer.mozilla.org/en-US/docs/Web/API/Headers
    headers: Headers;
 
    srcIp: number[];
 
    // Request's method (GET, POST, etc.)
    method: string;
 
    // The associated Args object of the request.
    args: Args;
}
 
 
interface Args {
    keys(): Iterable<string>
    get(k: string): string
    set(k: string, v: string): Args
}
 
interface Headers {
    keys(): Iterable<string>
    get(k: string): string
    set(k: string, v: string): Headers
}
 
interface Response {
    // The status code of the response (200, 404, etc.)
    status?: number;
    // The associated Headers object of the response.
    // See https://developer.mozilla.org/en-US/docs/Web/API/Headers
    headers?: Headers;
    // The body of the response
    body?: Uint8Array | string;
}
 
class Runner {
    public confCacheTTL: number
    private plugins: Plugin[]
    private pluginMap: {
        [name: string]: Plugin
    }
    private cacheCount: number
    private cache: {
        [key: string]: any
    }
 
    constructor() {
        this.plugins = []
        this.pluginMap = {}
        this.cacheCount = 0
        this.cache = {}
    }
 
    registerPlugin(plugin: Plugin) {
        console.log(`Register JavaScript Plugin: ${plugin.getName()}`)
        this.pluginMap[plugin.getName()] = plugin
        this.plugins.push(plugin)
    }
 
    genCacheToken() {
        this.cacheCount++
        return this.cacheCount
    }
 
    setCache(k: string, val: Object) {
        this.cache[k] = val
    }
 
    getCache(k: string) {
        return this.cache[k]
    }
 
    prepareConf(confList: {name: string, value: string}[]): number {
        confList = confList.
            filter(({name}) => this.pluginMap[name]).
            map(({name, value}) => {
                return {name, value: this.pluginMap[name].parseConf(value)}
            })
        const token = this.genCacheToken()
        this.setCache(token.toString(), confList)
        return token
    }
 
    async httpReqCall(confToken: number, request: Request) {
        const cache = this.getCache(confToken.toString()) as {name: string, value: string}[]
        Iif (!cache) {
            throw new Error(`Cache ${confToken} not found`)
        }
        console.debug({cache})
        const response = {} as Response
        for (let {name, value} of cache)  {
            console.debug({name, value})
            await this.pluginMap[name].filter(value, request, response)
        }
        const isStop = response.body || response.status || response.headers
        if (isStop) {
            Eif (!response.status) {
                response.status = 200 // defaults to 200
            }
        }
        return {isStop, response, request}
    }
 
}
 
export default Runner
export {Request, Response, Headers, Plugin}