app.common.upload.shapeways.oauth.ModelUploadOauthRunner.java Source code

Introduction

Here is the source code for app.common.upload.shapeways.oauth.ModelUploadOauthRunner.java

Source

/*****************************************************************************
 *                        Shapeways, Inc Copyright (c) 2011
 *                               Java Source
 *
 * This source is licensed under the GNU LGPL v2.1
 * Please read http://www.gnu.org/copyleft/lgpl.html for more information
 *
 * This software comes with the standard NO WARRANTY disclaimer for any
 * purpose. Use it at your own risk. If there's a problem you get to fix it.
 *
 ****************************************************************************/

package app.common.upload.shapeways.oauth;

// External imports
import java.io.File;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

import org.scribe.builder.ServiceBuilder;
import org.scribe.model.*;
import org.scribe.oauth.*;
import org.apache.commons.codec.binary.*;
import org.apache.commons.io.FileUtils;
import com.google.gson.Gson;

// Internal imports
import app.common.upload.shapeways.oauth.api.ShapewaysApi;

/**
 * Uploads a model with OAuth authentication.
 *
 * @author Tony Wong
 */
public class ModelUploadOauthRunner {
    /** The upload url */
    private static final String UPLOAD_URL = "http://api.shapeways.com/models/v1";

    /** Access token and secret */
    private String CONSUMER_KEY = "fcc8983c4e72e553e19583fcfbedf8d55d5dfc24";
    private String CONSUMER_SECRET = "ae647dffab05dbabaf92f739c766cda3351664d1";
    private String ACCESS_TOKEN = "22cbc36539f31f8ecf8990a6386ebe76676bef87";
    private String ACCESS_SECRET = "8dc65aab672eedebeb691eb5bcc52e1b9a969aa9";

    private OAuthService service;
    private Token accessToken;

    public ModelUploadOauthRunner() {
        service = new ServiceBuilder().provider(ShapewaysApi.class).apiKey(CONSUMER_KEY).apiSecret(CONSUMER_SECRET)
                .build();
        accessToken = new Token(ACCESS_TOKEN, ACCESS_SECRET);
    }

    public ModelUploadOauthRunner(String apiKey, String apiSecret, String accessToken, String accessSecret) {
        CONSUMER_KEY = apiKey;
        CONSUMER_SECRET = apiSecret;
        ACCESS_TOKEN = accessToken;
        ACCESS_SECRET = accessSecret;

        service = new ServiceBuilder().provider(ShapewaysApi.class).apiKey(CONSUMER_KEY).apiSecret(CONSUMER_SECRET)
                .build();
        this.accessToken = new Token(ACCESS_TOKEN, ACCESS_SECRET);
    }

    /*
       public ModelUploadOauthRunner(String propertyFile) {
          Properties properties = new Properties();
          InputStream in = this.getClass().getResourceAsStream("/config/oauth.properties");
              
          try {
     properties.load(in);
         
     System.out.println(properties.toString());
          } catch (IOException ioe) {
     ioe.printStackTrace();
          }
       }
    */
    public String uploadModel(byte[] bytes, String fileName, Integer modelId, Float scale, String title,
            String description, Integer isPublic, Integer isForSale, Map<Integer, Object> materialsMap)
            throws Exception {

        String reason = null;
        Gson gson = new Gson();

        String encodedFile = null;
        encodedFile = encodeModel(bytes);

        Map<String, Object> parameters = new HashMap<String, Object>();
        parameters.put("file", encodedFile);
        parameters.put("fileName", fileName);
        parameters.put("ownOrAuthorizedModel", 1);
        parameters.put("acceptTermsAndConditions", 1);
        parameters.put("hasRightsToModel", 1);

        if (scale != null) {
            parameters.put("uploadScale", scale.doubleValue());
        }
        if (modelId != null) {
            parameters.put("modelId", modelId.intValue());
        }
        if (title != null) {
            parameters.put("title", title);
        }
        if (description != null) {
            parameters.put("description", description);
        }
        if (isPublic != null) {
            parameters.put("isPublic", isPublic.intValue());
        }
        if (isForSale != null) {
            parameters.put("isForSale", isForSale.intValue());
        }
        if (materialsMap != null) {
            parameters.put("materials", materialsMap);
        }

        OAuthRequest request = new OAuthRequest(Verb.POST, UPLOAD_URL);
        String encodedParams = gson.toJson(parameters);
        //System.out.println("json params: " + encodedParams);

        request.addPayload(encodedParams);
        service.signRequest(accessToken, request);
        Response response = request.send();

        if (!response.isSuccessful()) {
            reason = "Upload failed: " + UPLOAD_URL + ", response code: " + response.getCode();
            System.out.println(reason);
            throw new Exception(reason);
        }

        String responseBody = response.getBody();
        System.out.println(responseBody);

        return responseBody;
    }

    public String uploadModel(String modelFile, Integer modelId, Float scale, String title, String description,
            Integer isPublic, Integer isForSale, Map<Integer, Object> materialsMap) throws Exception {

        String reason = null;
        File file = new File(modelFile);

        try {
            byte[] fileBytes = FileUtils.readFileToByteArray(file);
            return uploadModel(fileBytes, file.getName(), modelId, scale, title, description, isPublic, isForSale,
                    materialsMap);
        } catch (IOException ioe) {
            reason = "Model not found: " + file.getAbsolutePath();
            System.out.println(reason);
            throw new Exception(reason);
        }
    }

    public boolean isSuccess(String response) {
        Gson gson = new Gson();
        Map responseMap = gson.fromJson(response, Map.class);

        String result = (String) responseMap.get("result");

        if (result.equals("failure")) {
            String reason = (String) responseMap.get("reason");
            System.out.println("Upload failed! Reason: " + reason);
            return false;
        } else if (result.equals("success")) {
            int modelId = ((Double) responseMap.get("modelId")).intValue();
            int movelVersion = ((Double) responseMap.get("modelVersion")).intValue();
            String fileName = (String) responseMap.get("fileName");
            System.out.println("Upload success... ");
            System.out.println("  model id: " + modelId);
            System.out.println("  model version: " + movelVersion);
            System.out.println("  file name: " + fileName);
            return true;
        } else {
            System.out.println("Missing result field in response!");
            return false;
        }
    }

    private String encodeModel(byte[] bytes) {
        return Base64.encodeBase64String(bytes);
    }

    private String encodeModel(File modelFile) throws IOException {
        byte[] fileBytes = FileUtils.readFileToByteArray(modelFile);
        return Base64.encodeBase64String(fileBytes);
    }

}