Core: Generators

[kendarorg]

[kendarorg]

CURL GENERATOR

@Override
public String generate(Request request, Response response) {
    var result = new ArrayList<String>();
    result.add("#!/bin/sh");

    var dataFile = UUID.randomUUID() +".data";
    var hasData = false;
    if(request.isBinaryRequest() && request.getRequestBytes()!=null && request.getRequestBytes().length>0){
        var base64Bytes = new String(Base64.getEncoder().encode(request.getRequestBytes()));
        result.add("echo '"+base64Bytes+"' |base64 -d -o "+dataFile);
        hasData = true;
    }else if(request.getRequestText()!=null && !request.getRequestText().isEmpty()){
        var straightText = request.getRequestText();
        result.add("echo '"+straightText+"' |base64 -d -o "+dataFile);
        hasData = true;
    }

    var port = request.getPort()>0?":"+request.getPort():"";
    var target = request.getProtocol()+"://"+request.getHost()+port+request.getPath();
    target+="?"+ request.getQuery().entrySet().stream().map(e->e.getKey()+"="+e.getValue()).collect(Collectors.joining("&"));
    result.add("curl --insecure -X "+request.getMethod().toUpperCase(Locale.ROOT)+" \\");
    for (var head :
            request.getHeaders().entrySet()){
        result.add("\t-H '"+head.getKey()+":"+head.getValue()+"' \\");
    }
    if(hasData){
        result.add("\t-F 'file=@/"+dataFile+"' \\");
    }
    for (var postParameter :
            request.getPostParameters().entrySet()){
        result.add("\t-F \""+postParameter.getKey()+"="+postParameter.getValue()+"\" \\");
    }
    result.add("\t"+target);


    result.add("");
    return String.join("\n",result);
}