io.yawp.commons.http.RequestContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of yawp-core Show documentation

YAWP! core

There is a newer version: 2.08alpha

package io.yawp.commons.http;

import io.yawp.commons.utils.JsonUtils;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;

public class RequestContext {

    private HttpServletResponse resp;

    private HttpServletRequest req;

    protected String method;

    protected String uri;

    protected String json;

    protected Map params;

    public RequestContext() {
    }

    public RequestContext(HttpServletRequest req, HttpServletResponse resp) {
        this.req = req;
        this.resp = resp;

        this.method = parseMethod();
        this.uri = parseUri();
        this.json = parseJson();
        this.params = parseParams();
    }

    public HttpServletRequest req() {
        return req;
    }

    public HttpServletResponse resp() {
        return resp;
    }

    public String getMethod() {
        return method;
    }

    public HttpVerb getHttpVerb() {
        return HttpVerb.fromString(method);
    }

    public String getUri() {
        return uri;
    }

    public String getJson() {
        return json;
    }

    public Map getParams() {
        return params;
    }

    public boolean hasParam(String key) {
        return params.containsKey(key);
    }

    public String getParam(String key) {
        return params.get(key);
    }

    private String parseMethod() {
        return req.getMethod();
    }

    private String parseUri() {
        return req.getRequestURI().substring(req.getServletPath().length());
    }

    private String parseJson() {
        try {
            return JsonUtils.readJson(req.getReader());
        } catch (IOException e) {
            throw new RuntimeException();
        }
    }

    private Map parseParams() {
        Map map = new HashMap<>();

        Enumeration e = req.getParameterNames();
        while (e.hasMoreElements()) {
            String name = e.nextElement();
            map.put(name, req.getParameter(name));
        }

        return map;
    }

}