com.dft.api.shopify.InventoryLevelAPI Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of shopify-admin-rest Show documentation
Shopify Admin REST API using JDK 11 and Reactive Programming
The newest version!
package com.dft.api.shopify;


import com.dft.api.shopify.model.ShopifyInventoryLevel;
import com.dft.api.shopify.model.ShopifyInventoryLevelsRoot;
import com.dft.api.shopify.model.ShopifyPage;
import lombok.SneakyThrows;

import javax.ws.rs.client.WebTarget;
import javax.ws.rs.core.Link;
import javax.ws.rs.core.Response;
import java.net.URI;
import java.util.HashMap;
import java.util.List;
import java.util.Set;

import static com.dft.api.shopify.ShopifySdk.PAGE_INFO_QUERY_PARAMETER;

public class InventoryLevelAPI {

    private static final String REL_NEXT_HEADER_KEY = "next";
    private static final String REL_PREVIOUS_HEADER_KEY = "previous";
    private static final String AMPERSAND = "&";
    private static final String EQUALS = "=";

    static final String VERSION_2023_01 = "api/2023-01";
    static final String INVENTORY_LEVELS = "inventory_levels";
    private final ShopifySdk shopifySdk;

    public InventoryLevelAPI(ShopifySdk shopifySdk) {
        this.shopifySdk = shopifySdk;
    }

    @SneakyThrows
    public ShopifyPage getInventoryLevels(HashMap parameters) {
        WebTarget webTarget = shopifySdk.getWebTarget().path(VERSION_2023_01).path(INVENTORY_LEVELS);

        for (String key : parameters.keySet()) {
            String value = parameters.get(key);
            webTarget = webTarget.queryParam(key, value);
        }

        final Response response = shopifySdk.get(webTarget);

        ShopifyInventoryLevelsRoot shopifyInventoryLevelRoot = response.readEntity(ShopifyInventoryLevelsRoot.class);
        return mapPagedResponse(shopifyInventoryLevelRoot.getInventoryLevels(), response);
    }

    @SneakyThrows
    private  ShopifyPage mapPagedResponse(final List items, final Response response) {

        final ShopifyPage shopifyPage = new ShopifyPage<>();
        shopifyPage.addAll(items);

        final Set links = response.getLinks();
        final String nextLink = links.stream().filter(link -> link.getRel().equals(REL_NEXT_HEADER_KEY))
            .map(link -> getQueryParam(link.getUri(), PAGE_INFO_QUERY_PARAMETER)).findFirst().orElse(null);
        final String previousLink = links.stream().filter(link -> link.getRel().equals(REL_PREVIOUS_HEADER_KEY))
            .map(link -> getQueryParam(link.getUri(), PAGE_INFO_QUERY_PARAMETER)).findFirst().orElse(null);
        shopifyPage.setNextPageInfo(nextLink);
        shopifyPage.setPreviousPageInfo(previousLink);

        return shopifyPage;
    }

    @SneakyThrows
    private String getQueryParam(final URI uri, final String key) {
        final String[] params = uri.getQuery().split(AMPERSAND);
        for (final String param : params) {
            final String name = param.split(EQUALS)[0];
            final String value = param.split(EQUALS)[1];
            if (name.equals(key)) {
                return value;
            }
        }
        return null;
    }
}