endpoint /businesses (#7)

* added new features for new endpoint; bump version

* renamed method

* version increased

Author: Stepbus

build.gradle

@@ -2,7 +2,7 @@ apply plugin: 'java'
 apply plugin: 'maven-publish'
 
 group = 'com.github.outscraper'
-version = '2.1.1'
+version = '2.1.3'
 
 repositories {
   mavenCentral()

src/main/java/OutscraperClient.java

@@ -1,7 +1,9 @@
 import java.io.IOException;
 import java.net.URI;
+import java.net.URLEncoder;
 import java.net.http.HttpClient;
 import java.net.http.HttpRequest;
+import java.nio.charset.StandardCharsets;
 import java.net.http.HttpResponse;
 import java.util.ArrayList;
 import java.util.List;
@@ -69,6 +71,31 @@ private JSONObject getAPIRequest(String path, HashMap<String, Object> parameters
         return null;
     }
 
+    public JSONObject postAPIRequest(String path, HashMap<String, Object> parameters) {
+        String url = privateApiURL + path;
+
+        JSONObject jsonPayload = new JSONObject(parameters != null ? parameters : new HashMap<>());
+
+        HttpRequest request = HttpRequest.newBuilder()
+            .uri(URI.create(url))
+            .POST(HttpRequest.BodyPublishers.ofString(jsonPayload.toString()))
+            .header("accept", "application/json")
+            .header("content-type", "application/json")
+            .header("X-API-KEY", privateApiKey)
+            .header("client", "Java SDK")
+            .build();
+
+        try {
+            HttpResponse<String> response = client.send(request, HttpResponse.BodyHandlers.ofString());
+            String body = response.body();
+            return new JSONObject(body);
+        } catch (IOException | InterruptedException | JSONException e) {
+            e.printStackTrace();
+        }
+
+        return null;
+    }
+
     public JSONObject getRequestArchive(String requestId) {
         return getAPIRequest("/requests/" + requestId, new HashMap<String, Object>() {{}});
     }
@@ -335,4 +362,101 @@ public JSONArray yellowpagesSearch(HashMap<String, Object> parameters) {
 
         return getData(response);
     }
+
+    public JSONObject businessesSearch(HashMap<String, Object> parameters) {
+        if (parameters == null) parameters = new HashMap<>();
+        parameters.putIfAbsent("async", false);
+
+        JSONObject response = postAPIRequest("/businesses", parameters);
+        if (response == null) return null;
+
+        Object data = response.opt("data");
+        if (data instanceof JSONObject) {
+            return (JSONObject) data;
+        }
+        if (data instanceof JSONArray) {
+            return new JSONObject().put("items", (JSONArray) data).put("has_more", false).put("next_cursor", JSONObject.NULL);
+        }
+        return response;
+    }
+
+    public JSONArray businessesIterSearch(HashMap<String, Object> parameters) {
+        if (parameters == null) parameters = new HashMap<>();
+
+        JSONArray all = new JSONArray();
+        Object cursorObj = parameters.get("cursor");
+        String cursor = cursorObj != null ? String.valueOf(cursorObj) : null;
+
+        while (true) {
+            if (cursor != null) {
+                parameters.put("cursor", cursor);
+            } else {
+                parameters.remove("cursor");
+            }
+
+            JSONObject page = businessesSearch(parameters);
+            if (page == null) break;
+
+            JSONArray items = page.optJSONArray("items");
+            if (items == null) {
+                Object data = page.opt("data");
+                if (data instanceof JSONArray) items = (JSONArray) data;
+            }
+
+            if (items != null) {
+                for (int i = 0; i < items.length(); i++) {
+                    all.put(items.get(i));
+                }
+            }
+
+            boolean hasMore = page.optBoolean("has_more", false);
+
+            String nextCursor = null;
+            Object nextCursorObj = page.opt("next_cursor");
+            if (nextCursorObj != null && nextCursorObj != JSONObject.NULL) {
+                nextCursor = String.valueOf(nextCursorObj);
+                if (nextCursor.isBlank()) nextCursor = null;
+            }
+
+            if (!hasMore || nextCursor == null || items == null || items.length() == 0) {
+                break;
+            }
+
+            cursor = nextCursor;
+        }
+
+        return all;
+    }
+
+    public JSONObject businessesGet(String businessId, HashMap<String, Object> parameters) {
+        if (businessId == null || businessId.isBlank()) {
+            throw new IllegalArgumentException("businessId is required");
+        }
+        if (parameters == null) parameters = new HashMap<>();
+
+        Object fieldsObj = parameters.get("fields");
+        if (fieldsObj instanceof List) {
+            @SuppressWarnings("unchecked")
+            List<Object> fields = (List<Object>) fieldsObj;
+            List<String> strFields = new ArrayList<>();
+            for (Object f : fields) {
+                if (f != null) strFields.add(String.valueOf(f));
+            }
+            parameters.put("fields", String.join(",", strFields));
+        }
+
+        String encodedId = URLEncoder.encode(String.valueOf(businessId), StandardCharsets.UTF_8);
+        JSONObject response = getAPIRequest("/businesses/" + encodedId, parameters);
+        if (response == null) return null;
+
+        Object data = response.opt("data");
+        if (data instanceof JSONObject) return (JSONObject) data;
+        if (data instanceof JSONArray) {
+            JSONArray arr = (JSONArray) data;
+            if (arr.length() > 0 && arr.get(0) instanceof JSONObject) {
+                return arr.getJSONObject(0);
+            }
+        }
+        return response;
+    }
 }