net/base/url_util.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "net/base/url_util.h"
   6
   7 #include <utility>
   8
   9 #include "base/logging.h"
  10 #include "base/strings/string_piece.h"
  11 #include "net/base/escape.h"
  12 #include "url/gurl.h"
  13
  14 namespace net {
  15
  16 GURL AppendQueryParameter(const GURL& url,
  17                           const std::string& name,
  18                           const std::string& value) {
  19   std::string query(url.query());
  20
  21   if (!query.empty())
  22     query += "&";
  23
  24   query += (EscapeQueryParamValue(name, true) + "=" +
  25             EscapeQueryParamValue(value, true));
  26   GURL::Replacements replacements;
  27   replacements.SetQueryStr(query);
  28   return url.ReplaceComponents(replacements);
  29 }
  30
  31 GURL AppendOrReplaceQueryParameter(const GURL& url,
  32                                    const std::string& name,
  33                                    const std::string& value) {
  34   bool replaced = false;
  35   std::string param_name = EscapeQueryParamValue(name, true);
  36   std::string param_value = EscapeQueryParamValue(value, true);
  37
  38   const std::string input = url.query();
  39   url::Component cursor(0, input.size());
  40   std::string output;
  41   url::Component key_range, value_range;
  42   while (url::ExtractQueryKeyValue(input.data(), &cursor, &key_range,
  43                                    &value_range)) {
  44     const base::StringPiece key(
  45         input.data() + key_range.begin, key_range.len);
  46     std::string key_value_pair;
  47     // Check |replaced| as only the first pair should be replaced.
  48     if (!replaced && key == param_name) {
  49       replaced = true;
  50       key_value_pair = (param_name + "=" + param_value);
  51     } else {
  52       key_value_pair.assign(input.data(),
  53                             key_range.begin,
  54                             value_range.end() - key_range.begin);
  55     }
  56     if (!output.empty())
  57       output += "&";
  58
  59     output += key_value_pair;
  60   }
  61   if (!replaced) {
  62     if (!output.empty())
  63       output += "&";
  64
  65     output += (param_name + "=" + param_value);
  66   }
  67   GURL::Replacements replacements;
  68   replacements.SetQueryStr(output);
  69   return url.ReplaceComponents(replacements);
  70 }
  71
  72 QueryIterator::QueryIterator(const GURL& url)
  73     : url_(url),
  74       at_end_(!url.is_valid()) {
  75   if (!at_end_) {
  76     query_ = url.parsed_for_possibly_invalid_spec().query;
  77     Advance();
  78   }
  79 }
  80
  81 QueryIterator::~QueryIterator() {
  82 }
  83
  84 std::string QueryIterator::GetKey() const {
  85   DCHECK(!at_end_);
  86   if (key_.is_nonempty())
  87     return url_.spec().substr(key_.begin, key_.len);
  88   return std::string();
  89 }
  90
  91 std::string QueryIterator::GetValue() const {
  92   DCHECK(!at_end_);
  93   if (value_.is_nonempty())
  94     return url_.spec().substr(value_.begin, value_.len);
  95   return std::string();
  96 }
  97
  98 const std::string& QueryIterator::GetUnescapedValue() {
  99   DCHECK(!at_end_);
 100   if (value_.is_nonempty() && unescaped_value_.empty()) {
 101     unescaped_value_ = UnescapeURLComponent(
 102         GetValue(),
 103         UnescapeRule::SPACES |
 104         UnescapeRule::URL_SPECIAL_CHARS |
 105         UnescapeRule::REPLACE_PLUS_WITH_SPACE);
 106   }
 107   return unescaped_value_;
 108 }
 109
 110 bool QueryIterator::IsAtEnd() const {
 111   return at_end_;
 112 }
 113
 114 void QueryIterator::Advance() {
 115   DCHECK (!at_end_);
 116   key_.reset();
 117   value_.reset();
 118   unescaped_value_.clear();
 119   at_end_ =
 120       !url::ExtractQueryKeyValue(url_.spec().c_str(), &query_, &key_, &value_);
 121 }
 122
 123 bool GetValueForKeyInQuery(const GURL& url,
 124                            const std::string& search_key,
 125                            std::string* out_value) {
 126   for (QueryIterator it(url); !it.IsAtEnd(); it.Advance()) {
 127     if (it.GetKey() == search_key) {
 128       *out_value = it.GetUnescapedValue();
 129       return true;
 130     }
 131   }
 132   return false;
 133 }
 134
 135 }  // namespace net