url/url_canon_stdstring.h

   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 #ifndef URL_URL_CANON_STDSTRING_H_
   6 #define URL_URL_CANON_STDSTRING_H_
   7
   8 // This header file defines a canonicalizer output method class for STL
   9 // strings. Because the canonicalizer tries not to be dependent on the STL,
  10 // we have segregated it here.
  11
  12 #include <string>
  13
  14 #include "base/compiler_specific.h"
  15 #include "base/strings/string_piece.h"
  16 #include "url/url_canon.h"
  17 #include "url/url_export.h"
  18
  19 namespace url {
  20
  21 // Write into a std::string given in the constructor. This object does not own
  22 // the string itself, and the user must ensure that the string stays alive
  23 // throughout the lifetime of this object.
  24 //
  25 // The given string will be appended to; any existing data in the string will
  26 // be preserved.
  27 //
  28 // Note that when canonicalization is complete, the string will likely have
  29 // unused space at the end because we make the string very big to start out
  30 // with (by |initial_size|). This ends up being important because resize
  31 // operations are slow, and because the base class needs to write directly
  32 // into the buffer.
  33 //
  34 // Therefore, the user should call Complete() before using the string that
  35 // this class wrote into.
  36 class URL_EXPORT StdStringCanonOutput : public CanonOutput {
  37  public:
  38   StdStringCanonOutput(std::string* str);
  39   ~StdStringCanonOutput() override;
  40
  41   // Must be called after writing has completed but before the string is used.
  42   void Complete();
  43
  44   void Resize(int sz) override;
  45
  46  protected:
  47   std::string* str_;
  48 };
  49
  50 // An extension of the Replacements class that allows the setters to use
  51 // StringPieces (implicitly allowing strings or char*s).
  52 //
  53 // The contents of the StringPieces are not copied and must remain valid until
  54 // the StringPieceReplacements object goes out of scope.
  55 template<typename STR>
  56 class StringPieceReplacements : public Replacements<typename STR::value_type> {
  57  public:
  58   void SetSchemeStr(const base::BasicStringPiece<STR>& s) {
  59     this->SetScheme(s.data(), Component(0, static_cast<int>(s.length())));
  60   }
  61   void SetUsernameStr(const base::BasicStringPiece<STR>& s) {
  62     this->SetUsername(s.data(), Component(0, static_cast<int>(s.length())));
  63   }
  64   void SetPasswordStr(const base::BasicStringPiece<STR>& s) {
  65     this->SetPassword(s.data(), Component(0, static_cast<int>(s.length())));
  66   }
  67   void SetHostStr(const base::BasicStringPiece<STR>& s) {
  68     this->SetHost(s.data(), Component(0, static_cast<int>(s.length())));
  69   }
  70   void SetPortStr(const base::BasicStringPiece<STR>& s) {
  71     this->SetPort(s.data(), Component(0, static_cast<int>(s.length())));
  72   }
  73   void SetPathStr(const base::BasicStringPiece<STR>& s) {
  74     this->SetPath(s.data(), Component(0, static_cast<int>(s.length())));
  75   }
  76   void SetQueryStr(const base::BasicStringPiece<STR>& s) {
  77     this->SetQuery(s.data(), Component(0, static_cast<int>(s.length())));
  78   }
  79   void SetRefStr(const base::BasicStringPiece<STR>& s) {
  80     this->SetRef(s.data(), Component(0, static_cast<int>(s.length())));
  81   }
  82 };
  83
  84 }  // namespace url
  85
  86 #endif  // URL_URL_CANON_STDSTRING_H_