net.urllib: keep the query parameter order (#13405)

pull/13419/head
Vincenzo Palazzo 2022-02-09 16:36:12 +01:00 committed by GitHub
parent 4be3c92640
commit 0d1d259bb4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 134 additions and 49 deletions

View File

@ -230,6 +230,11 @@ fn parse_request_line(s string) ?(Method, urllib.URL, Version) {
} }
// Parse URL encoded key=value&key=value forms // Parse URL encoded key=value&key=value forms
//
// FIXME: Some servers can require the
// parameter in a specific order.
//
// a possible solution is to use the a list of QueryValue
pub fn parse_form(body string) map[string]string { pub fn parse_form(body string) map[string]string {
words := body.split('&') words := body.split('&')
mut form := map[string]string{} mut form := map[string]string{}

View File

@ -835,28 +835,32 @@ fn parse_query_values(mut m Values, query string) ?bool {
} }
// encode encodes the values into ``URL encoded'' form // encode encodes the values into ``URL encoded'' form
// ('bar=baz&foo=quux') sorted by key. // ('bar=baz&foo=quux').
// The syntx of the query string is specified in the
// RFC173 https://datatracker.ietf.org/doc/html/rfc1738
//
// HTTP grammar
//
// httpurl = "http://" hostport [ "/" hpath [ "?" search ]]
// hpath = hsegment *[ "/" hsegment ]
// hsegment = *[ uchar | ";" | ":" | "@" | "&" | "=" ]
// search = *[ uchar | ";" | ":" | "@" | "&" | "=" ]
pub fn (v Values) encode() string { pub fn (v Values) encode() string {
if v.len == 0 { if v.len == 0 {
return '' return ''
} }
mut buf := strings.new_builder(200) mut buf := strings.new_builder(200)
mut keys := []string{} for qvalue in v.data {
for k, _ in v.data { key_kscaped := query_escape(qvalue.key)
keys << k
}
keys.sort()
for k in keys {
vs := v.data[k]
key_kscaped := query_escape(k)
for _, val in vs.data {
if buf.len > 0 { if buf.len > 0 {
buf.write_string('&') buf.write_string('&')
} }
buf.write_string(key_kscaped) buf.write_string(key_kscaped)
buf.write_string('=') if qvalue.value == '' {
buf.write_string(query_escape(val)) continue
} }
buf.write_string('=')
buf.write_string(query_escape(qvalue.value))
} }
return buf.str() return buf.str()
} }

View File

@ -40,8 +40,14 @@ fn test_parse_query() ? {
q2 := urllib.parse_query('format="%l:+%c+%t"') ? q2 := urllib.parse_query('format="%l:+%c+%t"') ?
// dump(q1) // dump(q1)
// dump(q2) // dump(q2)
assert q1.data['format'].data == ['"%l: %c %t"'] assert q1.get('format') == '"%l: %c %t"'
assert q2.data['format'].data == ['"%l: %c %t"'] assert q2.get('format') == '"%l: %c %t"'
}
fn test_parse_query_orders() ? {
query_one := urllib.parse_query('https://someapi.com/endpoint?gamma=zalibaba&tau=1&alpha=alibaba&signature=alibaba123') ?
qvalues := query_one.values()
assert qvalues == ['zalibaba', '1', 'alibaba', 'alibaba123']
} }
fn test_parse_missing_host() ? { fn test_parse_missing_host() ? {
@ -49,3 +55,46 @@ fn test_parse_missing_host() ? {
url := urllib.parse('http:///') ? url := urllib.parse('http:///') ?
assert url.str() == 'http://///' assert url.str() == 'http://///'
} }
// testing the case where the key as a null value
// e.g ?key=
fn test_parse_none_value() ? {
query_one := urllib.parse_query('gamma=zalibaba&tau=1&alpha=alibaba&signature=') ?
qvalues := query_one.values()
qvalues_map := query_one.to_map()
assert qvalues == ['zalibaba', '1', 'alibaba']
assert qvalues_map == {
'gamma': ['zalibaba']
'tau': ['1']
'alpha': ['alibaba']
'signature': ['']
}
}
// testing the case where the query as empity value
// e.g https://www.vlang.dev?alibaba
fn test_parse_empty_query_one() ? {
query_str := 'alibaba'
query_one := urllib.parse_query(query_str) ?
qvalues := query_one.values()
qvalues_map := query_one.to_map()
query_encode := query_one.encode()
assert qvalues == []
assert qvalues_map == {
'alibaba': ['']
}
assert query_str == query_encode
}
// testing the case where the query as empity value
// e.g https://www.vlang.dev?
fn test_parse_empty_query_two() ? {
query_str := ''
query_one := urllib.parse_query(query_str) ?
qvalues := query_one.values()
qvalues_map := query_one.to_map()
query_encode := query_one.encode()
assert qvalues == []
assert qvalues_map == {}
assert query_str == query_encode
}

View File

@ -3,14 +3,15 @@
// that can be found in the LICENSE file. // that can be found in the LICENSE file.
module urllib module urllib
struct Value { struct QueryValue {
pub mut: pub mut:
data []string key string
value string
} }
struct Values { struct Values {
pub mut: pub mut:
data map[string]Value data []QueryValue
len int len int
} }
@ -20,17 +21,10 @@ pub mut:
// values.encode() will return the encoded data // values.encode() will return the encoded data
pub fn new_values() Values { pub fn new_values() Values {
return Values{ return Values{
data: map[string]Value{} data: []QueryValue{}
} }
} }
// Currently you will need to use all()[key].data
// once map[string][]string is implemented
// this will be fixed
pub fn (v &Value) all() []string {
return v.data
}
// get gets the first value associated with the given key. // get gets the first value associated with the given key.
// If there are no values associated with the key, get returns // If there are no values associated with the key, get returns
// a empty string. // a empty string.
@ -38,11 +32,12 @@ pub fn (v &Values) get(key string) string {
if v.data.len == 0 { if v.data.len == 0 {
return '' return ''
} }
vs := v.data[key] for qvalue in v.data {
if vs.data.len == 0 { if qvalue.key == key {
return '' return qvalue.value
} }
return vs.data[0] }
return ''
} }
// get_all gets the all the values associated with the given key. // get_all gets the all the values associated with the given key.
@ -52,36 +47,68 @@ pub fn (v &Values) get_all(key string) []string {
if v.data.len == 0 { if v.data.len == 0 {
return [] return []
} }
vs := v.data[key] mut values := []string{}
if vs.data.len == 0 { for qvalue in v.data {
return [] if qvalue.key == key {
values << qvalue.value
} }
return vs.data }
return values
} }
// set sets the key to value. It replaces any existing // set sets the key to value. It replaces any existing
// values. // values.
pub fn (mut v Values) set(key string, value string) { pub fn (mut v Values) set(key string, value string) {
mut a := v.data[key] // A query string can contains several
a.data = [value] // duplicate, so we need to make sure that we
v.data[key] = a // cover all the edge case.
v.len = v.data.len for mut qvalue in v.data {
qvalue.value = value
}
} }
// add adds the value to key. It appends to any existing // add adds the value to key. It appends to any existing
// values associated with key. // values associated with key.
pub fn (mut v Values) add(key string, value string) { pub fn (mut v Values) add(key string, value string) {
mut a := v.data[key] v.data << QueryValue{
if a.data.len == 0 { key: key
a.data = [] value: value
} }
a.data << value
v.data[key] = a
v.len = v.data.len v.len = v.data.len
} }
// del deletes the values associated with key. // del deletes the values associated with key.
pub fn (mut v Values) del(key string) { pub fn (mut v Values) del(key string) {
v.data.delete(key) for idx, qvalue in v.data {
if qvalue.key == key {
v.data.delete(idx)
}
}
v.len = v.data.len v.len = v.data.len
} }
// return the list of values in the query string
pub fn (v Values) values() []string {
mut values := []string{}
for qvalue in v.data {
if qvalue.value != '' {
values << qvalue.value
}
}
return values
}
// return a map <key []value> of the query string
pub fn (v Values) to_map() map[string][]string {
mut result := map[string][]string{}
for qvalue in v.data {
if qvalue.key in result {
result[qvalue.key] << qvalue.value
} else {
result[qvalue.key] = [qvalue.value]
}
}
return result
}

View File

@ -49,8 +49,8 @@ fn parse_attrs(name string, attrs []string) ?([]http.Method, string) {
fn parse_query_from_url(url urllib.URL) map[string]string { fn parse_query_from_url(url urllib.URL) map[string]string {
mut query := map[string]string{} mut query := map[string]string{}
for k, v in url.query().data { for qvalue in url.query().data {
query[k] = v.data[0] query[qvalue.key] = qvalue.value
} }
return query return query
} }