Refactored URLEncoing code

fernandezpablo85 · fernandezpablo85 · commit c0d24b556c1a · 2010-11-20T19:27:04.000-03:00
diff --git a/changelog.txt b/changelog.txt
@@ -11,3 +11,4 @@
 
  * FEATURE: Added setConnectTimeout and setReadTimeout for the Request Object
  * FIX: Fixed Evernote Api (uses GET for request and access tokens)
+ * REFACTOR: URLEncoding stuff. LinkedIn specific decoder is no longer needed
diff --git a/src/main/java/org/scribe/builder/api/LinkedInApi.java b/src/main/java/org/scribe/builder/api/LinkedInApi.java
@@ -1,11 +1,7 @@
 package org.scribe.builder.api;
 
-import org.scribe.extensions.linkedin.*;
-import org.scribe.extractors.*;
-
 public class LinkedInApi extends DefaultApi10a
 {
-
   @Override
   protected String getAccessTokenEndpoint()
   {
@@ -17,11 +13,4 @@ protected String getRequestTokenEndpoint()
   {
     return "https://api.linkedin.com/uas/oauth/requestToken";
   }
-  
-  @Override
-  protected BaseStringExtractor getBaseStringExtractor()
-  {
-    return new LinkedInBaseStringExtractorImpl();
-  }
-
 }
diff --git a/src/main/java/org/scribe/extensions/linkedin/LinkedInBaseStringExtractorImpl.java b/src/main/java/org/scribe/extensions/linkedin/LinkedInBaseStringExtractorImpl.java
diff --git a/src/main/java/org/scribe/utils/URLUtils.java b/src/main/java/org/scribe/utils/URLUtils.java
@@ -20,6 +20,17 @@ public class URLUtils
 
   private static final String ERROR_MSG = String.format("Cannot find specified encoding: %s", UTF_8);
 
+  private static final Set<EncodingRule> ENCODING_RULES;
+
+  static
+  {
+    Set<EncodingRule> rules = new HashSet<EncodingRule>();
+    rules.add(new EncodingRule("*","%2A"));
+    rules.add(new EncodingRule("+","%20"));
+    rules.add(new EncodingRule("%7E", "~"));
+    ENCODING_RULES = Collections.unmodifiableSet(rules);
+  }
+
   /**
    * Turns a map into a form-url-encoded string (key=value&key2=value2)
    * 
@@ -57,7 +68,12 @@ public static String percentEncode(String string)
     Preconditions.checkNotNull(string, "Cannot encode null string");
     try
     {
-      return URLEncoder.encode(string, UTF_8).replaceAll("\\+", "%20");
+      String encoded = URLEncoder.encode(string, UTF_8);
+      for(EncodingRule rule : ENCODING_RULES)
+      {
+        encoded = rule.apply(encoded);
+      }
+      return encoded;
     } 
     catch (UnsupportedEncodingException uee)
     {
@@ -77,9 +93,26 @@ public static String percentDecode(String string)
     try
     {
       return URLDecoder.decode(string, UTF_8);
-    } catch (UnsupportedEncodingException uee)
+    }
+    catch (UnsupportedEncodingException uee)
     {
       throw new OAuthException(ERROR_MSG, uee);
     }
   }
+
+  private static final class EncodingRule
+  {
+    private final String ch;
+    private final String toCh;
+
+    EncodingRule(String ch, String toCh)
+    {
+      this.ch = ch;
+      this.toCh = toCh;
+    }
+
+    String apply(String string) {
+      return string.replace(ch, toCh);
+    }
+  }
 }
diff --git a/src/test/java/org/scribe/examples/LinkedInExample.java b/src/test/java/org/scribe/examples/LinkedInExample.java
@@ -55,8 +55,6 @@ public static void main(String[] args)
 
     System.out.println();
     System.out.println("Thats it man! Go and build something awesome with Scribe! :)");
-    
-    System.out.println(service.getRequestToken());
   }
 
 }
diff --git a/src/test/java/org/scribe/utils/URLUtilsTest.java b/src/test/java/org/scribe/utils/URLUtilsTest.java
@@ -8,13 +8,6 @@
 
 public class URLUtilsTest
 {
-
-  @Before
-  public void setup()
-  {
-
-  }
-
   @Test
   public void shouldPercentEncodeMap()
   {
@@ -51,6 +44,23 @@ public void shouldPercentDecodeString()
     assertEquals(expected, URLUtils.percentDecode(toDecode));
   }
 
+  @Test
+  public void shouldEncodeAllSpecialCharacters()
+  {
+    String plain = "!*'();:@&=+$,/?#[]";
+    String encoded = "%21%2A%27%28%29%3B%3A%40%26%3D%2B%24%2C%2F%3F%23%5B%5D";
+    assertEquals(encoded, URLUtils.percentEncode(plain));
+    assertEquals(plain, URLUtils.percentDecode(encoded));
+  }
+
+  @Test
+  public void shouldNotEncodeReservedCharacters()
+  {
+    String plain = "abcde123456-._~";
+    String encoded = plain;
+    assertEquals(encoded, URLUtils.percentEncode(plain));
+  }
+
   @Test(expected = IllegalArgumentException.class)
   public void shouldThrowExceptionIfMapIsNull()
   {

Original file line number	Diff line number	Diff line change
`@@ -11,3 +11,4 @@`
`11`	`11`
`12`	`12`	`* FEATURE: Added setConnectTimeout and setReadTimeout for the Request Object`
`13`	`13`	`* FIX: Fixed Evernote Api (uses GET for request and access tokens)`
	`14`	`+ * REFACTOR: URLEncoding stuff. LinkedIn specific decoder is no longer needed`
Original file line number	Diff line number	Diff line change
`@@ -1,11 +1,7 @@`
`1`	`1`	`package org.scribe.builder.api;`
`2`	`2`
`3`		`-import org.scribe.extensions.linkedin.*;`
`4`		`-import org.scribe.extractors.*;`
`5`		`-`
`6`	`3`	`public class LinkedInApi extends DefaultApi10a`
`7`	`4`	`{`
`8`		`-`
`9`	`5`	`@Override`
`10`	`6`	`protected String getAccessTokenEndpoint()`
`11`	`7`	`{`
`@@ -17,11 +13,4 @@ protected String getRequestTokenEndpoint()`
`17`	`13`	`{`
`18`	`14`	`return "https://api.linkedin.com/uas/oauth/requestToken";`
`19`	`15`	`}`
`20`		`-`
`21`		`- @Override`
`22`		`- protected BaseStringExtractor getBaseStringExtractor()`
`23`		`- {`
`24`		`- return new LinkedInBaseStringExtractorImpl();`
`25`		`- }`
`26`		`-`
`27`	`16`	`}`
Original file line number	Diff line number	Diff line change
`@@ -55,8 +55,6 @@ public static void main(String[] args)`
`55`	`55`
`56`	`56`	`System.out.println();`
`57`	`57`	`System.out.println("Thats it man! Go and build something awesome with Scribe! :)");`
`58`		`-`
`59`		`- System.out.println(service.getRequestToken());`
`60`	`58`	`}`
`61`	`59`
`62`	`60`	`}`