AxiosCodeGenerator.ts

declare function require(module: string): any;

declare function registerCodeGenerator(callback: Object): void;

declare interface AxiosRequest {
    name: string;
    url: string;
    method: string;
    headers: Object;
    body: any;
    // httpBasicAuth: Object;
    // followRedirects: boolean;
    // sendCookies: boolean;
    // storeCookies: boolean;
    // timeout: number;
}

class ParsedURL {
    public schema = 'http';
    public host = 'localhost';
    public port = 80;
    public path = '/';

    constructor(url: string) {
        const regexp = /(https?):\/\/([^\/:]+):?(\d*)(\/?.*)/;
        const match = url.match(regexp);

        if (match) {
            this.schema = match[1];
            this.host = match[2];
            this.port =
                match[3].length > 0
                    ? +match[3]
                    : (() => {
                          if (this.schema == 'https') return 443;
                          return 80;
                      })();
            this.path = match[4];
        }
    }
}

class AxiosCodeGenerator {
    static title = 'Axios';
    static fileExtension = 'js';
    static languageHighlighter = 'js';
    static identifier = 'cloud.lumin.AxiosCodeGenerator';

    private multipleRequestNotice(request: AxiosRequest[]) {
        if (request.length > 1) {
            return '// Warning: requests below are going to be executed in parallel\n\n';
        }
        return '';
    }

    public generate(context: any, requests: AxiosRequest[], options) {
        if (!Array.isArray(requests)) {
            requests = [context.getCurrentRequest()];
        }
        return this.multipleRequestNotice(requests) + requests.map(this.generateRequest).join('\n');
    }

    private generateRequest(request: AxiosRequest) {
        const headers = request.headers;
        for (var key in headers) {
            headers[key] = headers[key].trim();
        }

        const parsedUrl = new ParsedURL(request.url);

        return `// request ${request.name} 

        axios({
            method: '${request.method.toLowerCase()}',
            url: '${parsedUrl.schema}://${parsedUrl.host}${parsedUrl.path}',
            ${Object.keys(request.body).length > 0 ? 'data: ' + request.body + ',' : ''}
            ${headers ? 'headers: ' + JSON.stringify(headers) + ',' : ''}
        })
`;
    }
}

registerCodeGenerator(AxiosCodeGenerator);