Hub.java

package com.qiniu.pili;

import com.google.gson.Gson;
import com.qiniu.pili.utils.UrlSafeBase64;


public final class Hub {

    private RPC cli;
    private String hub;
    private String baseUrl;
    private Gson gson;

    private Hub() {
    }

    Hub(RPC cli, String hub) {
        this.cli = cli;
        this.hub = hub;
        this.baseUrl = Config.APIHTTPScheme + Config.APIHost + "/v2/hubs/" + hub;
        this.gson = new Gson();
    }

    /**
     * create stream
     *
     * @param streamKey
     * @return
     * @throws PiliException
     */
    public Stream create(String streamKey) throws PiliException {
        String path = this.baseUrl + "/streams";
        CreateArgs args = new CreateArgs(streamKey);
        String json = gson.toJson(args);

        try {
            cli.callWithJson(path, json);
            StreamInfo streamInfo = new StreamInfo(hub, streamKey);
            return new Stream(streamInfo, cli);
        } catch (PiliException e) {
            throw e;
        } catch (Exception e) {
            throw new PiliException(e);
        }
    }

    /**
     * get stream
     *
     * @param streamKey
     * @return
     * @throws PiliException
     */
    public Stream get(String streamKey) throws PiliException {

        try {
            String ekey = UrlSafeBase64.encodeToString(streamKey);
            ;
            String path = baseUrl + "/streams/" + ekey;

            String resp = cli.callWithGet(path);
            StreamInfo ret = gson.fromJson(resp, StreamInfo.class);
            ret.setMeta(hub, streamKey);
            return new Stream(ret, cli);
        } catch (PiliException e) {
            throw e;
        } catch (Exception e) {
            throw new PiliException(e);
        }

    }

    private ListRet list(boolean live, String prefix, int limit, String marker) throws PiliException {
        String path = String.format("%s/streams?liveonly=%s&prefix=%s&limit=%d&marker=%s", baseUrl, live, prefix, limit, marker);
        try {
            String resp = cli.callWithGet(path);

            ApiRet ret = gson.fromJson(resp, ApiRet.class);

            ListRet listRet = new ListRet();
            listRet.keys = new String[ret.items.length];
            for (int i = 0; i < ret.items.length; i++) {
                listRet.keys[i] = ret.items[i].key;
            }
            listRet.omarker = ret.marker;
            return listRet;
        } catch (PiliException e) {
            throw e;
        } catch (Exception e) {
            throw new PiliException(e);
        }

    }

    /**
     * list stream
     *
     * @param prefix
     * @param limit
     * @param marker
     * @return
     * @throws PiliException
     */
    public ListRet list(String prefix, int limit, String marker) throws PiliException {
        return list(false, prefix, limit, marker);
    }

    /**
     * list streams which is live
     *
     * @param prefix
     * @param limit
     * @param marker
     * @return
     * @throws PiliException
     */
    public ListRet listLive(String prefix, int limit, String marker) throws PiliException {
        return list(true, prefix, limit, marker);
    }

    /**
     * batch get live status
     *
     * @param streamTitles
     * @return
     * @throws PiliException
     */
    public BatchLiveStatus[] batchLiveStatus(String[] streamTitles) throws PiliException {
        String path = baseUrl + "/livestreams";
        String json = gson.toJson(new BatchLiveStatusOptions(streamTitles));
        try {
            String resp = cli.callWithJson(path, json);

            BatchLiveStatusRet ret = gson.fromJson(resp, BatchLiveStatusRet.class);
            return ret.items;
        } catch (PiliException e) {
            throw e;
        } catch (Exception e) {
            throw new PiliException(e);
        }
    }

    private class BatchLiveStatusOptions {
        String[] items;

        public BatchLiveStatusOptions(String[] items) {
            this.items = items;
        }
    }

    private class BatchLiveStatusRet {
        BatchLiveStatus[] items;
    }

    public class BatchLiveStatus extends Stream.LiveStatus {
        String key;
    }

    /*
        Create
     */
    private class CreateArgs {
        String key;

        public CreateArgs(String key) {
            this.key = key;
        }
    }

    /*
        List
     */
    public class ListRet {
        public String[] keys;
        public String omarker;
    }

    private class ListItem {
        String key;
    }

    private class ApiRet {
        ListItem[] items;
        String marker;
    }


}