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 "url/url_canon.h"
16#include "url/url_export.h"
17
18namespace url {
19
20// Write into a std::string given in the constructor. This object does not own
21// the string itself, and the user must ensure that the string stays alive
22// throughout the lifetime of this object.
23//
24// The given string will be appended to; any existing data in the string will
25// be preserved. The caller should reserve() the amount of data in the string
26// they expect to be written. We will resize if necessary, but that's slow.
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.
36class URL_EXPORT StdStringCanonOutput : public CanonOutput {
37 public:
38  StdStringCanonOutput(std::string* str);
39  virtual ~StdStringCanonOutput();
40
41  // Must be called after writing has completed but before the string is used.
42  void Complete();
43
44  virtual 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// standard strings.
52//
53// The strings passed as arguments are not copied and must remain valid until
54// this class goes out of scope.
55template<typename STR>
56class StdStringReplacements : public Replacements<typename STR::value_type> {
57 public:
58  void SetSchemeStr(const STR& s) {
59    this->SetScheme(s.data(), Component(0, static_cast<int>(s.length())));
60  }
61  void SetUsernameStr(const STR& s) {
62    this->SetUsername(s.data(), Component(0, static_cast<int>(s.length())));
63  }
64  void SetPasswordStr(const STR& s) {
65    this->SetPassword(s.data(), Component(0, static_cast<int>(s.length())));
66  }
67  void SetHostStr(const STR& s) {
68    this->SetHost(s.data(), Component(0, static_cast<int>(s.length())));
69  }
70  void SetPortStr(const STR& s) {
71    this->SetPort(s.data(), Component(0, static_cast<int>(s.length())));
72  }
73  void SetPathStr(const STR& s) {
74    this->SetPath(s.data(), Component(0, static_cast<int>(s.length())));
75  }
76  void SetQueryStr(const STR& s) {
77    this->SetQuery(s.data(), Component(0, static_cast<int>(s.length())));
78  }
79  void SetRefStr(const 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_
87