Add separate handling of response status.

[jrw.git] / src / jrw / Request.java

package jrw;

import java.util.*;

public class Request {
    public final Map<Object, Object> env;
    public final Map<Object, Object> resp = new HashMap<>();
    public String status = "200 OK";
    public Object body = null;

    public Request(Map<Object, Object> env) {
	this.env = env;
    }

    public String ihead(String name, String def) {
	StringBuilder buf = new StringBuilder();
	buf.append("HTTP_");
	for(int i = 0; i < name.length(); i++) {
	    char c = name.charAt(i);
	    if(c == '-')
		buf.append('_');
	    else if((c >= 'a') && (c <= 'z'))
		buf.append((char)(c + ('A' - 'a')));
	    else
		buf.append(c);
	}
	Object ret = env.get(buf.toString());
	if(ret instanceof String)
	    return((String)ret);
	return(def);
    }

    @SuppressWarnings("unchecked")
    public Request ohead(String name, Object val, boolean repl) {
	name = "http." + name;
	if(repl) {
	    resp.put(name, val);
	} else {
	    Object cur = resp.get(name);
	    if(cur == null)
		resp.put(name, val);
	    else if(cur instanceof Collection)
		((Collection)cur).add(val);
	    else
		resp.put(name, new ArrayList<Object>(Arrays.asList(cur, val)));
	}
	return(this);
    }

    public Request status(String status) {this.status = status; return(this);}
    public Request body(Object body) {this.body = body; return(this);}

    public Map<Object, Object> response() {
	resp.put("http.status", status);
	resp.put("jagi.output", body);
	return(resp);
    }
}