com.github.stephanarts.cas.ticket.registry.RegistryClient.java Source code

Introduction

Here is the source code for com.github.stephanarts.cas.ticket.registry.RegistryClient.java

Source

/*
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.github.stephanarts.cas.ticket.registry;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.ArrayList;
import java.util.Collection;

import java.io.ByteArrayOutputStream;
import java.io.ObjectOutputStream;
import java.io.ByteArrayInputStream;
import java.io.ObjectInputStream;
import javax.xml.bind.DatatypeConverter;

import org.json.JSONObject;
import org.json.JSONArray;
//import org.json.JSONException;

import org.jasig.cas.ticket.Ticket;

import com.github.stephanarts.cas.ticket.registry.support.JSONRPCException;
import com.github.stephanarts.cas.ticket.registry.support.JSONRPCClient;
import com.github.stephanarts.cas.ticket.registry.support.PaceMaker;

/**
 * RegistryClient Class.
 */
public class RegistryClient extends JSONRPCClient implements RegistryClientMBean {

    /**
     * Logging Class.
     */
    protected final Logger logger = LoggerFactory.getLogger(getClass());

    /**
     * Create RegistryClient object.
     *
     * @param connectUri   URI to connect the client to.
     */
    public RegistryClient(final String connectUri) {
        super(connectUri);
        this.connect();
    }

    /**
     * Create RegistryClient object.
     *
     * @param connectUri   URI to connect the client to.
     * @param pacemaker    Pacemaker instance,
     */
    public RegistryClient(final String connectUri, final PaceMaker pacemaker) {
        super(connectUri, pacemaker);
        this.connect();
    }

    /**
     * get ProviderId.
     *
     * @return providerId string
     *
     * @throws JSONRPCException when something goes wrong.
     */
    public final String getProviderId() throws JSONRPCException {

        JSONObject params = new JSONObject();
        JSONObject result;

        String providerId = null;

        result = this.call("cas.getProviderId", params);

        if (result.has("provider-id")) {
            providerId = result.getString("provider-id");
        }

        return providerId;
    }

    /**
     * addTicket Method.
     *
     * @param ticket CAS Ticket object
     *
     * @throws JSONRPCException Throws JSONRPCException containing any error.
     */
    public final void addTicket(final Ticket ticket) throws JSONRPCException {

        byte[] serializedTicket = {};
        JSONObject params = new JSONObject();

        /* Check if it's not null */
        if (ticket == null) {
            throw new JSONRPCException(-32501, "Could not encode Ticket");
        }

        try {
            ByteArrayOutputStream bo = new ByteArrayOutputStream();
            ObjectOutputStream so = new ObjectOutputStream(bo);
            so.writeObject(ticket);
            so.flush();
            serializedTicket = bo.toByteArray();
        } catch (final Exception e) {
            throw new JSONRPCException(-32501, "Could not encode Ticket");
        }

        params.put("ticket-id", ticket.getId());
        params.put("ticket", DatatypeConverter.printBase64Binary(serializedTicket));

        this.call("cas.addTicket", params);
    }

    /**
     * Delete a ticket from the ticketregistry.
     *
     * @param ticketId       Ticket-object to delete from Registry
     *
     * @throws JSONRPCException Throws JSONRPCException containing any error.
     */
    public final void deleteTicket(final String ticketId) throws JSONRPCException {

        JSONObject params = new JSONObject();

        params.put("ticket-id", ticketId);

        this.call("cas.deleteTicket", params);
    }

    /**
     * Get a ticket from the ticketregistry.
     *
     * @param ticketId       id of ticket-object to get from Registry
     *
     * @return Ticket Object
     *
     * @throws JSONRPCException Throws JSONRPCException containing any error.
     */
    public final Ticket getTicket(final String ticketId) throws JSONRPCException {

        JSONObject params = new JSONObject();
        JSONObject result;

        Ticket ticket = null;
        String serializedTicket;

        params.put("ticket-id", ticketId);

        result = this.call("cas.getTicket", params);

        if (result.has("ticket")) {
            serializedTicket = result.getString("ticket");

            try {
                ByteArrayInputStream bi = new ByteArrayInputStream(
                        DatatypeConverter.parseBase64Binary(serializedTicket));
                ObjectInputStream si = new ObjectInputStream(bi);

                ticket = (Ticket) si.readObject();
            } catch (final Exception e) {
                throw new JSONRPCException(-32501, "Could not decode Ticket");
            }
        }

        if (ticket == null) {
            logger.debug("Failed to get Ticket: {}", ticketId);
        }

        return ticket;
    }

    /**
     * updateTicket Method.
     *
     * @param ticket CAS Ticket object
     *
     * @return Response Object
     *
     * @throws JSONRPCException Throws JSONRPCException containing any error.
     */
    public final JSONObject updateTicket(final Ticket ticket) throws JSONRPCException {

        byte[] serializedTicket = {};
        JSONObject params = new JSONObject();

        try {
            ByteArrayOutputStream bo = new ByteArrayOutputStream();
            ObjectOutputStream so = new ObjectOutputStream(bo);
            so.writeObject(ticket);
            so.flush();
            serializedTicket = bo.toByteArray();
        } catch (final Exception e) {
            throw new JSONRPCException(-32501, "Could not decode Ticket");
        }

        params.put("ticket-id", ticket.getId());
        params.put("ticket", DatatypeConverter.printBase64Binary(serializedTicket));

        return this.call("cas.updateTicket", params);
    }

    /**
     * Get a ticket from the ticketregistry.
     *
     * @return Ticket Objects
     *
     * @throws JSONRPCException Throws JSONRPCException containing any error.
     */
    public final Collection<Ticket> getTickets() throws JSONRPCException {

        JSONObject params = new JSONObject();
        JSONObject result;
        JSONArray resultTickets;

        Ticket ticket;
        ArrayList<Ticket> tickets = new ArrayList<Ticket>();

        result = this.call("cas.getTickets", params);

        if (result.has("tickets")) {
            logger.debug("Number of Tickets: " + result.getJSONArray("tickets").length());
            resultTickets = result.getJSONArray("tickets");
            for (int i = 0; i < resultTickets.length(); ++i) {
                try {
                    String serializedTicket = resultTickets.getString(i);
                    ByteArrayInputStream bi = new ByteArrayInputStream(
                            DatatypeConverter.parseBase64Binary(serializedTicket));
                    ObjectInputStream si = new ObjectInputStream(bi);

                    ticket = (Ticket) si.readObject();

                    tickets.add(ticket);
                } catch (final Exception e) {
                    throw new JSONRPCException(-32501, "Could not decode Ticket");
                }
            }
        }

        return tickets;
    }

    /**
     * MBean Interface for provider availability.
     *
     * @return true if the provider is available;
     */
    public final boolean getProviderAvailable() {
        return this.getAvailable();
    }

}