Mercurial Hosting > luan
annotate core/src/luan/modules/url/LuanUrl.java @ 725:a741a3a33423
add url support for multipart/form-data
| author | Franklin Schmidt <fschmidt@gmail.com> |
|---|---|
| date | Wed, 08 Jun 2016 23:13:10 -0600 |
| parents | core/src/luan/modules/LuanUrl.java@eaf30d5aaf6a |
| children | 14f136a4641f |
| rev | line source |
|---|---|
|
725
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
1 package luan.modules.url; |
| 722 | 2 |
| 3 import java.io.InputStream; | |
| 4 import java.io.InputStreamReader; | |
| 5 import java.io.OutputStream; | |
| 6 import java.io.Reader; | |
| 7 import java.io.IOException; | |
| 8 import java.io.UnsupportedEncodingException; | |
| 9 import java.net.URL; | |
| 10 import java.net.URLConnection; | |
| 11 import java.net.HttpURLConnection; | |
| 12 import java.net.URLEncoder; | |
| 13 import java.util.Map; | |
| 14 import java.util.HashMap; | |
| 15 import java.util.List; | |
| 16 import java.util.Base64; | |
| 17 import luan.LuanState; | |
| 18 import luan.LuanTable; | |
| 19 import luan.LuanJavaFunction; | |
| 20 import luan.LuanException; | |
|
725
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
21 import luan.modules.IoLuan; |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
22 import luan.modules.Utils; |
| 722 | 23 |
| 24 | |
| 25 public final class LuanUrl extends IoLuan.LuanIn { | |
|
725
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
26 |
| 722 | 27 private static enum Method { GET, POST, DELETE } |
| 28 | |
| 29 private URL url; | |
| 30 private Method method = Method.GET; | |
| 31 private Map headers; | |
|
725
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
32 private String content = null; |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
33 private MultipartClient multipart = null; |
| 722 | 34 |
|
725
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
35 public LuanUrl(LuanState luan,URL url,LuanTable options) throws LuanException { |
| 722 | 36 this.url = url; |
| 37 if( options != null ) { | |
| 38 Map map = options.asMap(luan); | |
| 39 String methodStr = getString(map,"method"); | |
| 40 if( methodStr != null ) { | |
| 41 methodStr = methodStr.toUpperCase(); | |
| 42 try { | |
| 43 this.method = Method.valueOf(methodStr); | |
| 44 } catch(IllegalArgumentException e) { | |
| 45 throw new LuanException( "invalid method: "+methodStr ); | |
| 46 } | |
| 47 } | |
| 723 | 48 Map headerMap = getMap(luan,map,"headers"); |
| 49 if( headerMap != null ) { | |
| 50 headers = new HashMap(); | |
| 51 for( Object hack : headerMap.entrySet() ) { | |
| 52 Map.Entry entry = (Map.Entry)hack; | |
| 53 String key = (String)entry.getKey(); | |
| 54 Object val = entry.getValue(); | |
| 55 String name = toHttpHeaderName(key); | |
| 56 if( val instanceof String ) { | |
| 57 headers.put(name,val); | |
| 58 } else { | |
| 59 if( !(val instanceof LuanTable) ) | |
| 60 throw new LuanException( "header '"+key+"' must be string or table" ); | |
| 61 LuanTable t = (LuanTable)val; | |
| 62 if( !t.isList() ) | |
| 63 throw new LuanException( "header '"+key+"' table must be list" ); | |
| 64 headers.put(name,t.asList()); | |
| 65 } | |
| 66 } | |
| 67 } | |
| 722 | 68 Map auth = getMap(luan,map,"authorization"); |
| 69 if( auth != null ) { | |
| 723 | 70 if( headers!=null && headers.containsKey("Authorization") ) |
| 71 throw new LuanException( "can't define authorization with header 'Authorization' defined" ); | |
| 722 | 72 String user = getString(auth,"user"); |
| 73 String password = getString(auth,"password"); | |
| 74 if( !auth.isEmpty() ) | |
| 75 throw new LuanException( "unrecognized authorization options: "+auth ); | |
| 76 StringBuilder sb = new StringBuilder(); | |
| 77 if( user != null ) | |
| 78 sb.append(user); | |
| 79 sb.append(':'); | |
| 80 if( password != null ) | |
| 81 sb.append(password); | |
| 82 String val = "Basic " + Base64.getEncoder().encodeToString(sb.toString().getBytes()); | |
| 83 if( headers == null ) | |
| 84 headers = new HashMap(); | |
| 85 headers.put("Authorization",val); | |
| 86 } | |
| 87 Map params = getMap(luan,map,"parameters"); | |
| 88 if( params != null ) { | |
|
725
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
89 if( this.method==Method.POST && "multipart/form-data".equals(headers.get("Content-Type")) ) { |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
90 multipart = new MultipartClient(params); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
91 } else { |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
92 StringBuilder sb = new StringBuilder(); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
93 for( Object hack : params.entrySet() ) { |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
94 Map.Entry entry = (Map.Entry)hack; |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
95 String key = (String)entry.getKey(); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
96 Object val = entry.getValue(); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
97 String keyEnc = encode(key); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
98 if( val instanceof String ) { |
| 722 | 99 and(sb); |
|
725
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
100 sb.append( keyEnc ).append( '=' ).append( encode((String)val) ); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
101 } else { |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
102 if( !(val instanceof LuanTable) ) |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
103 throw new LuanException( "parameter '"+key+"' must be string or table" ); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
104 LuanTable t = (LuanTable)val; |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
105 if( !t.isList() ) |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
106 throw new LuanException( "parameter '"+key+"' table must be list" ); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
107 for( Object obj : t.asList() ) { |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
108 if( !(obj instanceof String) ) |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
109 throw new LuanException( "parameter '"+key+"' values must be strings" ); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
110 and(sb); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
111 sb.append( keyEnc ).append( '=' ).append( encode((String)obj) ); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
112 } |
| 722 | 113 } |
| 114 } | |
|
725
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
115 if( this.method==Method.DELETE ) |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
116 throw new LuanException( "the DELETE method cannot take parameters" ); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
117 if( this.method==Method.POST ) { |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
118 content = sb.toString(); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
119 } else { // GET |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
120 String urlS = this.url.toString(); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
121 if( urlS.indexOf('?') == -1 ) { |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
122 urlS += '?'; |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
123 } else { |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
124 urlS += '&'; |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
125 } |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
126 urlS += sb; |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
127 try { |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
128 this.url = new URL(urlS); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
129 } catch(IOException e) { |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
130 throw new RuntimeException(e); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
131 } |
| 722 | 132 } |
| 133 } | |
| 134 } | |
| 135 if( !map.isEmpty() ) | |
| 136 throw new LuanException( "unrecognized options: "+map ); | |
| 137 } | |
| 138 } | |
| 139 | |
| 723 | 140 public static String toHttpHeaderName(String luanName) { |
| 141 luanName = luanName.toLowerCase(); | |
| 142 StringBuilder buf = new StringBuilder(); | |
| 143 boolean capitalize = true; | |
| 144 char[] a = luanName.toCharArray(); | |
| 145 for( int i=0; i<a.length; i++ ) { | |
| 146 char c = a[i]; | |
| 147 if( c == '_' || c == '-' ) { | |
| 148 a[i] = '-'; | |
| 149 capitalize = true; | |
| 150 } else if( capitalize ) { | |
| 151 a[i] = Character.toUpperCase(c); | |
| 152 capitalize = false; | |
| 153 } | |
| 154 } | |
| 155 return String.valueOf(a); | |
| 156 } | |
| 157 | |
| 722 | 158 private static void and(StringBuilder sb) { |
| 159 if( sb.length() > 0 ) | |
| 160 sb.append('&'); | |
| 161 } | |
| 162 | |
| 163 private static String encode(String s) { | |
| 164 try { | |
| 165 return URLEncoder.encode(s,"UTF-8"); | |
| 166 } catch(UnsupportedEncodingException e) { | |
| 167 throw new RuntimeException(e); | |
| 168 } | |
| 169 } | |
| 170 | |
| 171 private static String getString(Map map,String key) throws LuanException { | |
| 172 Object val = map.remove(key); | |
| 173 if( val!=null && !(val instanceof String) ) | |
| 174 throw new LuanException( "parameter '"+key+"' must be a string" ); | |
| 175 return (String)val; | |
| 176 } | |
| 177 | |
| 178 private static LuanTable getTable(Map map,String key) throws LuanException { | |
| 179 Object val = map.remove(key); | |
| 180 if( val!=null && !(val instanceof LuanTable) ) | |
| 181 throw new LuanException( "parameter '"+key+"' must be a table" ); | |
| 182 return (LuanTable)val; | |
| 183 } | |
| 184 | |
| 185 private static Map getMap(LuanState luan,Map map,String key) throws LuanException { | |
| 186 LuanTable t = getTable(map,key); | |
| 187 return t==null ? null : t.asMap(luan); | |
| 188 } | |
| 189 | |
|
725
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
190 @Override public InputStream inputStream() throws IOException, LuanException { |
| 722 | 191 URLConnection con = url.openConnection(); |
| 192 if( headers != null ) { | |
| 193 for( Object hack : headers.entrySet() ) { | |
| 194 Map.Entry entry = (Map.Entry)hack; | |
| 195 String key = (String)entry.getKey(); | |
| 196 Object val = entry.getValue(); | |
| 197 if( val instanceof String ) { | |
| 198 con.addRequestProperty(key,(String)val); | |
| 199 } else { | |
| 200 List list = (List)val; | |
| 201 for( Object obj : list ) { | |
| 202 con.addRequestProperty(key,(String)obj); | |
| 203 } | |
| 204 } | |
| 205 } | |
| 206 } | |
| 207 if( method==Method.GET ) { | |
| 208 return con.getInputStream(); | |
| 209 } | |
| 210 | |
| 211 HttpURLConnection httpCon = (HttpURLConnection)con; | |
| 212 | |
| 213 if( method==Method.DELETE ) { | |
| 214 httpCon.setRequestMethod("DELETE"); | |
| 215 return httpCon.getInputStream(); | |
| 216 } | |
| 217 | |
| 218 // POST | |
| 219 | |
| 220 // httpCon.setRequestProperty("Content-Type","application/x-www-form-urlencoded"); | |
| 221 httpCon.setDoOutput(true); | |
| 222 httpCon.setRequestMethod("POST"); | |
| 223 | |
|
725
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
224 OutputStream out; |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
225 if( multipart != null ) { |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
226 out = multipart.write(httpCon); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
227 } else { |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
228 byte[] post = content.getBytes(); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
229 // httpCon.setRequestProperty("Content-Length",Integer.toString(post.length)); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
230 out = httpCon.getOutputStream(); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
231 out.write(post); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
232 out.flush(); |
|
a741a3a33423
add url support for multipart/form-data
Franklin Schmidt <fschmidt@gmail.com>
parents:
723
diff
changeset
|
233 } |
| 722 | 234 try { |
| 235 try { | |
| 236 return httpCon.getInputStream(); | |
| 237 } catch(IOException e) { | |
| 238 InputStream is = httpCon.getErrorStream(); | |
| 239 if( is == null ) | |
| 240 throw e; | |
| 241 Reader in = new InputStreamReader(is); | |
| 242 String msg = Utils.readAll(in); | |
| 243 in.close(); | |
| 244 throw new LuanException(msg,e); | |
| 245 } | |
| 246 } finally { | |
| 247 out.close(); | |
| 248 } | |
| 249 } | |
| 250 | |
| 251 @Override public String to_string() { | |
| 252 return url.toString(); | |
| 253 } | |
| 254 | |
| 255 @Override public String to_uri_string() { | |
| 256 return url.toString(); | |
| 257 } | |
| 258 /* | |
| 259 public String post(String postS) throws IOException { | |
| 260 return new UrlCall(url).post(postS); | |
| 261 } | |
| 262 | |
| 263 @Override public LuanTable table() { | |
| 264 LuanTable tbl = super.table(); | |
| 265 try { | |
| 266 tbl.rawPut( "post", new LuanJavaFunction( | |
| 267 LuanUrl.class.getMethod( "post", String.class ), this | |
| 268 ) ); | |
| 269 } catch(NoSuchMethodException e) { | |
| 270 throw new RuntimeException(e); | |
| 271 } | |
| 272 return tbl; | |
| 273 } | |
| 274 */ | |
| 275 } |