From fb6d7f3b9d34032ec7ed32295b13dd556088989f Mon Sep 17 00:00:00 2001
From: Vitaly Takmazov
Date: Fri, 15 Mar 2019 14:11:47 +0300
Subject: Allow parameters in sape links

---
 src/main/java/ru/sape/Sape.java          |  4 +++-
 src/main/java/ru/sape/SapePageLinks.java | 15 ++++++++-------
 2 files changed, 11 insertions(+), 8 deletions(-)

(limited to 'src/main/java/ru/sape')

diff --git a/src/main/java/ru/sape/Sape.java b/src/main/java/ru/sape/Sape.java
index 38577c45..a94bcc62 100644
--- a/src/main/java/ru/sape/Sape.java
+++ b/src/main/java/ru/sape/Sape.java
@@ -3,6 +3,8 @@
  */
 package ru.sape;
 
+import java.net.URI;
+
 public class Sape {
 
     private final String sapeUser;
@@ -17,7 +19,7 @@ public class Sape {
     }
     public boolean debug = false;
 
-    public SapePageLinks getPageLinks(String requestUri, String cookie) {
+    public SapePageLinks getPageLinks(URI requestUri, String cookie) {
         return new SapePageLinks(sapePageLinkConnection, sapeUser, requestUri, cookie, debug);
     }
 }
diff --git a/src/main/java/ru/sape/SapePageLinks.java b/src/main/java/ru/sape/SapePageLinks.java
index e89b4e71..77715aea 100644
--- a/src/main/java/ru/sape/SapePageLinks.java
+++ b/src/main/java/ru/sape/SapePageLinks.java
@@ -1,17 +1,18 @@
 package ru.sape;
 
+import org.apache.commons.lang3.StringUtils;
+
+import java.net.URI;
 import java.util.*;
 
 public class SapePageLinks {
 
     private boolean showCode;
 
-    public SapePageLinks(SapeConnection sapeConnection, String sapeUser, String requestUri, String sapeCookie) {
-        this(sapeConnection, sapeUser, requestUri, sapeCookie, false);
-    }
-
     @SuppressWarnings("unchecked")
-    public SapePageLinks(SapeConnection sapeConnection, String sapeUser, String requestUri, String sapeCookie, boolean showCode) {
+    public SapePageLinks(SapeConnection sapeConnection, String sapeUser, URI request, String sapeCookie, boolean showCode) {
+        String req = StringUtils.isNotEmpty(request.getQuery()) ? request.getPath() + "?" + request.getQuery()
+                : request.getPath();
         if (sapeUser.equals(sapeCookie)) {
             showCode = true;
         }
@@ -22,8 +23,8 @@ public class SapePageLinks {
             linkDelimiter = (String) data.get("__sape_delimiter__");
         }
 
-        if (data.containsKey(requestUri)) {
-            pageLinks = new ArrayList<>(((Map<Object, String>) data.get(requestUri)).values());
+        if (data.containsKey(req)) {
+            pageLinks = new ArrayList<>(((Map<Object, String>) data.get(req)).values());
         }
 
         if (data.containsKey("__sape_new_url__")) {
-- 
cgit v1.2.3