com.rightscale.provider.rest.ServersResource.java Source code

Introduction

Here is the source code for com.rightscale.provider.rest.ServersResource.java

Source

// Dashboard: an Android front-end to the RightScale dashboard
// Copyright (C) 2009 Tony Spataro <code@tracker.xeger.net>
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU 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 General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

package com.rightscale.provider.rest;

import java.util.ArrayList;

import net.xeger.rest.ProtocolError;
import net.xeger.rest.RestException;
import net.xeger.rest.Session;

import org.apache.http.message.BasicNameValuePair;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;

import com.rightscale.provider.*;

import android.database.Cursor;
import android.database.MatrixCursor;

public class ServersResource extends DashboardResource {
    public static final String MIME_TYPE = "vnd.rightscale.server";

    public static final String ID = Dashboard.ID;
    public static final String DEPLOYMENT_ID = "deployment_id";
    public static final String SERVER_TEMPLATE_ID = "server_template_id";

    public static final String HREF = Dashboard.HREF;
    public static final String NICKNAME = "nickname";
    public static final String STATE = "state";

    public static final String[] COLUMNS = { ID, HREF, DEPLOYMENT_ID, SERVER_TEMPLATE_ID, NICKNAME, STATE };

    public ServersResource(Session session, String accountId) {
        super(session, accountId);
    }

    public Cursor index() throws RestException {
        try {
            return buildCursor(getJsonArray("servers.js"));
        } catch (JSONException e) {
            throw new ProtocolError(e);
        }
    }

    public Cursor indexForDeployment(String deploymentId) throws RestException {
        try {
            JSONObject deployment = getJsonObject("deployments/" + deploymentId + ".js");
            JSONArray servers = deployment.getJSONArray("servers");
            return buildCursor(servers);
        } catch (JSONException e) {
            throw new ProtocolError(e);
        }
    }

    public Cursor show(String id) throws RestException {
        try {
            JSONObject deployment = getJsonObject("servers/" + id + ".js");
            return buildCursor(deployment);
        } catch (JSONException e) {
            throw new ProtocolError(e);
        }
    }

    public void launch(String id) throws RestException {
        post("servers/" + id + "/start");
    }

    public void terminate(String id) throws RestException {
        post("servers/" + id + "/stop");
    }

    public void reboot(String id) throws RestException {
        post("servers/" + id + "/reboot");
    }

    public void runScript(String id, String scriptId) throws RestException {
        ArrayList<BasicNameValuePair> params = new ArrayList<BasicNameValuePair>();
        params.add(new BasicNameValuePair("right_script", Integer.valueOf(scriptId).toString()));
        post("servers/" + id + "/run_script", params);
    }

    private void buildRow(MatrixCursor.RowBuilder row, JSONObject object) throws JSONException {
        String href = object.getString("href");
        String deployment_href = object.getString("deployment_href");
        String server_template_href = object.getString("server_template_href");

        int id = Integer.valueOf(href.substring(href.lastIndexOf('/') + 1)).intValue(); //TODO error handling
        int deployment_id = Integer.valueOf(deployment_href.substring(deployment_href.lastIndexOf('/') + 1))
                .intValue(); //TODO error handling
        int server_template_id = Integer
                .valueOf(server_template_href.substring(server_template_href.lastIndexOf('/') + 1)).intValue(); //TODO error handling

        String nickname = object.getString("nickname");
        String state = object.getString("state");
        row.add(id);
        row.add(href);
        row.add(deployment_id);
        row.add(server_template_id);
        row.add(nickname);
        row.add(state);
    }

    private Cursor buildCursor(JSONObject object) throws JSONException {
        MatrixCursor result = new MatrixCursor(COLUMNS);
        MatrixCursor.RowBuilder row = result.newRow();
        buildRow(row, object);
        return result;
    }

    private Cursor buildCursor(JSONArray array) throws JSONException {
        MatrixCursor result = new MatrixCursor(COLUMNS);

        for (Integer i : sortJsonArray(array, "nickname")) {
            JSONObject object = array.getJSONObject(i);

            /* 
             * The API is glitchy for non-EC2 servers and returns an incomplete JSON object.
             * If the object lacks an href property, it is one of these and we skip it...
             */
            if (object.has("href")) {
                MatrixCursor.RowBuilder row = result.newRow();
                buildRow(row, object);
            }
        }

        return result;
    }
}