com.redhat.victims.mock.MockService.java Source code

Introduction

Here is the source code for com.redhat.victims.mock.MockService.java

Source

package com.redhat.victims.mock;

/*
 * #%L
 * This file is part of victims-lib.
 * %%
 * Copyright (C) 2013 The Victims Project
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 * #L%
 */

import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.net.HttpURLConnection;
import java.net.InetSocketAddress;

import org.apache.commons.io.FileUtils;

import com.sun.net.httpserver.Headers;
import com.sun.net.httpserver.HttpServer;
import com.sun.net.httpserver.HttpHandler;
import com.sun.net.httpserver.HttpExchange;

@SuppressWarnings("restriction")
public class MockService {
    private static Integer PORT = 1337;
    private static String DEFAULT_RESPONSE = "[]";
    private static HttpServer httpd;
    private static boolean running = false;

    public static void start(final File updateResponse, final File removeResponse) throws IOException {
        if (!running) {
            httpd = HttpServer.create(new InetSocketAddress(PORT), 0);
            HttpHandler empty = new GetHandler(DEFAULT_RESPONSE);
            HttpHandler update = updateResponse != null ? new GetHandler(updateResponse) : empty;
            HttpHandler remove = removeResponse != null ? new GetHandler(removeResponse) : empty;
            httpd.createContext("/service/update/", update);
            httpd.createContext("/service/remove/", remove);
            httpd.start();
            running = true;
        }
    }

    public static void stop() {
        httpd.stop(0);
        running = false;
    }

    public static String uri() {
        return String.format("http://localhost:%d/", PORT);
    }

    public static boolean isRunning() {
        return running;
    }

    static class GetHandler implements HttpHandler {

        private byte[] json;

        public GetHandler(File response) {
            try {
                json = FileUtils.readFileToByteArray(response);
            } catch (IOException e) {
                json = "".getBytes();
            }
        }

        public GetHandler(String response) {
            json = response.getBytes();
        }

        public void handle(HttpExchange exchange) {
            try {
                Headers headers = exchange.getResponseHeaders();
                headers.add("Content-Type", "application/json");

                exchange.sendResponseHeaders(HttpURLConnection.HTTP_OK, json.length);
                OutputStream os = exchange.getResponseBody();
                os.write(json);
                os.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
}