com.twosigma.beaker.core.rest.PublishRest.java Source code

Introduction

Here is the source code for com.twosigma.beaker.core.rest.PublishRest.java

Source

/*
 *  Copyright 2014 TWO SIGMA OPEN SOURCE, LLC
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package com.twosigma.beaker.core.rest;

import com.twosigma.beaker.core.module.config.BeakerConfig;
import com.google.inject.Inject;
import com.google.inject.Singleton;
import java.io.IOException;
import javax.ws.rs.FormParam;
import javax.ws.rs.Path;
import javax.ws.rs.POST;
import javax.ws.rs.Produces;
import javax.ws.rs.WebApplicationException;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import org.apache.commons.lang3.exception.ExceptionUtils;
import org.apache.http.HttpStatus;
import org.apache.http.client.fluent.Request;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.entity.ContentType;
import org.apache.http.HttpVersion;
import org.json.simple.JSONObject;
import org.json.simple.JSONValue;

/**
 * RESTful API for publishing to the web.
 */
@Path("publish")
@Produces(MediaType.APPLICATION_JSON)
@Singleton
public class PublishRest {

    private final String gistUrl;
    private final String sharingUrl;

    @Inject
    public PublishRest(BeakerConfig bkConfig) {
        this.gistUrl = bkConfig.getGistServerUrl();
        this.sharingUrl = bkConfig.getSharingServerUrl();
    }

    @POST
    @Path("github")
    @Produces(MediaType.APPLICATION_JSON)
    public String notebook(@FormParam("json") String json, @FormParam("type") String type)
            throws IOException, ClientProtocolException {
        String files = "{\"Beaker Share\":{\"content\":\"" + JSONObject.escape(json) + "\"}}";
        String body = "{\"description\":\"Beaker Share\",\"public\":true,\"files\":" + files + "}\n";
        String response = null;
        try {
            response = Request.Post(this.gistUrl).useExpectContinue().version(HttpVersion.HTTP_1_1)
                    .bodyString(body, ContentType.APPLICATION_JSON).execute().returnContent().asString();
        } catch (Throwable t) {
            throw new GistPublishException(ExceptionUtils.getStackTrace(t));
        }
        JSONObject parsed = (JSONObject) JSONValue.parse(response);
        String githubUrl = (String) parsed.get("html_url");
        int slash = githubUrl.lastIndexOf("/");
        if (slash < 0) {
            System.err.println("no slash found in github url: " + githubUrl);
            return githubUrl;
        }
        return this.sharingUrl + githubUrl.substring(slash);
    }

    private static class GistPublishException extends WebApplicationException {
        public GistPublishException(String stackTrace) {
            super(Response.status(HttpStatus.SC_INTERNAL_SERVER_ERROR)
                    .entity("<h1>Gist publish failed</h1><pre>" + stackTrace + "</pre>").type("text/plain")
                    .build());
        }
    }
}