net.nitrado.api.order.Pricing Maven / Gradle / Ivy

Go to download
package net.nitrado.api.order;

import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import net.nitrado.api.Nitrapi;
import net.nitrado.api.common.http.Parameter;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;

public abstract class Pricing {
    Nitrapi nitrapi;
    protected String product;
    int locationId;


    protected HashMap additionals;


    /**
     * Cache of prices by location id
     */
    private HashMap prices;




    public Pricing(Nitrapi nitrapi, int locationId) {
        this.nitrapi = nitrapi;
        this.locationId = locationId;
        this.prices = new HashMap();
        this.additionals = new HashMap();
    }

    /**
     *
     * @return a list of locations this service is available in.
     */
    public Location[] getLocations() {
        if (product == null) {
            throw new IllegalStateException("You cannot use Pricing.getLocations(). Please use the product class.");
        }
        JsonObject data = nitrapi.dataGet("order/order/locations", null);
        Location[] locations = nitrapi.fromJson(data.get("locations"), Location[].class);

        ArrayList available = new ArrayList();
        for (Location loc:locations) {
            if (loc.hasService(product)) { // TODO: dependent
                available.add(loc);
            }
        }
        return available.toArray(new Location[available.size()]);
    }

    public void setLocationId(int id) {
        this.locationId = id;
    }

    public PriceList getPrices() {
        return getPrices(-1);
    }

    /**
     * Get full price list for a specified product
     * @param service service id or -1
     * @return the price list for this product
     */
    public PriceList getPrices(int service) {
        String cacheName = "" + locationId;
        if (service != -1) {
            cacheName += "/" + service;
        }
        if (prices.containsKey(cacheName)) {  // if we requested this before, return it
            return prices.get(cacheName);
        }

        Parameter[] parameters;
        if (service != -1) {
            parameters = new Parameter[] {
                    new Parameter("location", locationId),
                    new Parameter("sale_service", service)
            };
        } else {
            parameters = new Parameter[] {
                    new Parameter("location", locationId)
            };
        }

        JsonObject data = nitrapi.dataGet("order/pricing/" + product, parameters);
        prices.put(cacheName, nitrapi.fromJson(data.get("prices"), PriceList.class));

        return prices.get(cacheName);
    }

    /**
     * Returns the price for extending a specific service.
     * @param service id of the service
     * @return the prices for extending this service
     */
    public ExtensionPrice[] getExtendPricesForService(int service) {
        JsonObject data = nitrapi.dataGet("order/pricing/" + product, new Parameter[]{
                new Parameter("method", "extend"),
                new Parameter("service_id", service)
        }).get("extend").getAsJsonObject().get("prices").getAsJsonObject();
        ExtensionPrice[] list = new ExtensionPrice[data.entrySet().size()];
        int i = 0;
        for (Map.Entry entry: data.entrySet()) {
            list[i] = new ExtensionPrice(Integer.parseInt(entry.getKey()), entry.getValue().getAsInt());
            i++;
        }
        return list;
    }

    /**
     * Extends the specific service about the specific rental time
     * @param service id of the service
     * @param rentalTime time to rent
     * @param price price calculated from getExtendPricesForService
     */
    public void doExtendService(int service, int rentalTime, int price) {
        // int price = getExtendPriceForService(service, rentalTime);
        nitrapi.dataPost("order/order/" + product, new Parameter[] {
                new Parameter("price", price),
                new Parameter("rental_time", rentalTime),
                new Parameter("service_id", service),
                new Parameter("method", "extend")
        });
    }

    public int getPrice(int rentalTime) {
        return getPrice(-1, rentalTime);
    }
    public abstract int getPrice(int service, int rentalTime);
    public abstract void orderService(int rentalTime);
    public int getSwitchPrice(int service, int rentalTime) {
        return getPrice(service, rentalTime);
    }
    public abstract void switchService(int service, int rentalTime);


}