From c5c2ed12adc44721b74c9a1544530de0ea8c85d7 Mon Sep 17 00:00:00 2001
From: 0xb00bface <0xboobface@gmail.com>
Date: Sat, 6 Nov 2021 17:41:34 +0100
Subject: [PATCH] Implement cherry.tv search

---
 .../java/ctbrec/sites/cherrytv/CherryTv.java  | 39 ++++++++++---------
 1 file changed, 21 insertions(+), 18 deletions(-)

diff --git a/common/src/main/java/ctbrec/sites/cherrytv/CherryTv.java b/common/src/main/java/ctbrec/sites/cherrytv/CherryTv.java
index b8174de5..02ed5af0 100644
--- a/common/src/main/java/ctbrec/sites/cherrytv/CherryTv.java
+++ b/common/src/main/java/ctbrec/sites/cherrytv/CherryTv.java
@@ -4,18 +4,16 @@ import ctbrec.Model;
 import ctbrec.io.HttpClient;
 import ctbrec.io.HttpException;
 import ctbrec.sites.AbstractSite;
-import okhttp3.HttpUrl;
 import okhttp3.Request;
 import okhttp3.Response;
+import org.json.JSONArray;
 import org.json.JSONObject;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 import java.io.IOException;
-import java.util.Collections;
-import java.util.List;
-import java.util.Locale;
-import java.util.Objects;
+import java.net.URLEncoder;
+import java.util.*;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
 
@@ -111,32 +109,37 @@ public class CherryTv extends AbstractSite {
 
     @Override
     public List<Model> search(String q) throws IOException, InterruptedException {
-        JSONObject variables = new JSONObject().put("slug", q).put("limit", 10);
-        JSONObject persistedQuery = new JSONObject().put("persistedQuery", new JSONObject().put("version", 1).put("sha256Hash", ""));
+        String url = "https://cherry.tv/graphql?operationName=findStreamersBySearch&variables="
+                + "{\"limit\":6,\"slug\":\"" + URLEncoder.encode(q, "utf-8") + "\"}&extensions={\"persistedQuery\":{\"version\":1,\"sha256Hash\":\"03d2f017fee32e1b6a1d3f816ce226c464a78d8dab28895c321980fbadffc1ef\"}}";
         Request req = new Request.Builder()
-                .url(new HttpUrl.Builder()
-                        .scheme("https")
-                        .host("cherry.tv")
-                        .addPathSegment("qraphql")
-                        .addQueryParameter("operationName", "Search")
-                        .addQueryParameter("variables", variables.toString())
-                        .addQueryParameter("extensions", persistedQuery.toString())
-                        .build()
-                )
+                .url(url)
                 .header(USER_AGENT, getConfig().getSettings().httpUserAgent)
                 .header(ACCEPT, MIMETYPE_APPLICATION_JSON)
                 .header(ACCEPT_LANGUAGE, Locale.ENGLISH.getLanguage())
                 .header(REFERER, getBaseUrl())
                 .header(X_REQUESTED_WITH, XML_HTTP_REQUEST)
                 .build();
+        LOG.debug("Search URL: {}", req.url());
+        List<Model> result = new LinkedList<>();
         try (Response response = getHttpClient().execute(req)) {
             if (response.isSuccessful()) {
-                LOG.debug("Response: {}", Objects.requireNonNull(response.body()).string());
+                JSONObject json = new JSONObject(Objects.requireNonNull(response.body()).string());
+                LOG.debug(json.toString(2));
+                JSONObject data = json.getJSONObject("data");
+                JSONObject searchResult = data.getJSONObject("searchResult");
+                JSONArray streamers = searchResult.getJSONArray("streamers");
+                for (int i = 0; i < streamers.length(); i++) {
+                    JSONObject hit = streamers.getJSONObject(i);
+                    CherryTvModel model = createModel(hit.getString("username"));
+                    model.setDescription(hit.getString("description"));
+                    model.setPreview(hit.getString("imageUrl"));
+                    result.add(model);
+                }
             } else {
                 throw new HttpException(response.code(), response.message());
             }
         }
-        return Collections.emptyList();
+        return result;
     }
 
     @Override