]> git.saurik.com Git - wxWidgets.git/blobdiff - src/common/url.cpp
wxTextOutputStream::PutChar and text stream test
[wxWidgets.git] / src / common / url.cpp
index cbe5866c76ee9705d5afd6e5feef10ae8f3feb94..2c30bffffcc3964e23b67718fa1ae3b2802bdc86 100644 (file)
@@ -32,7 +32,7 @@
 #include <ctype.h>
 
 IMPLEMENT_CLASS(wxProtoInfo, wxObject)
-IMPLEMENT_CLASS(wxURL, wxObject)
+IMPLEMENT_CLASS(wxURL, wxURI)
 
 // Protocols list
 wxProtoInfo *wxURL::ms_protocols = NULL;
@@ -49,14 +49,28 @@ USE_PROTOCOL(wxFTP)
 #endif
 
 // --------------------------------------------------------------
-// wxURL
+//
+//                          wxURL
+//
 // --------------------------------------------------------------
 
 // --------------------------------------------------------------
-// --------- wxURL CONSTRUCTOR DESTRUCTOR -----------------------
+// Construction
 // --------------------------------------------------------------
 
-wxURL::wxURL(const wxString& url)
+wxURL::wxURL(const wxString& url) : wxURI(url)
+{
+    Init(url);
+    ParseURL();
+}
+
+wxURL::wxURL(const wxURI& url) : wxURI(url)
+{
+    Init(url.BuildURI());
+    ParseURL();
+}
+
+void wxURL::Init(const wxString& url)
 {
     m_protocol = NULL;
     m_error = wxURL_NOERR;
@@ -81,21 +95,43 @@ wxURL::wxURL(const wxString& url)
     m_proxy = ms_proxyDefault;
 #endif // wxUSE_SOCKETS
 
+}
+
+// --------------------------------------------------------------
+// Assignment
+// --------------------------------------------------------------
+
+wxURL& wxURL::operator = (const wxURI& url)
+{
+    wxURI::operator = (url);
+    Init(url.BuildURI());
     ParseURL();
+    return *this;
+}
+wxURL& wxURL::operator = (const wxString& url)
+{
+    wxURI::operator = (url);
+    Init(url);
+    ParseURL();
+    return *this;
 }
 
+// --------------------------------------------------------------
+// ParseURL
+//
+// Builds the URL and takes care of the old protocol stuff
+// --------------------------------------------------------------
+
 bool wxURL::ParseURL()
 {
-  wxString last_url = m_url;
-
   // If the URL was already parsed (m_protocol != NULL), pass this section.
   if (!m_protocol)
   {
     // Clean up
     CleanData();
 
-    // Extract protocol name
-    if (!PrepProto(last_url))
+    // Make sure we have a protocol/scheme
+    if (!HasScheme())
     {
       m_error = wxURL_SNTXERR;
       return false;
@@ -111,22 +147,14 @@ bool wxURL::ParseURL()
     // Do we need a host name ?
     if (m_protoinfo->m_needhost)
     {
-      // Extract it
-      if (!PrepHost(last_url))
+      //  Make sure we have one, then
+      if (!HasServer())
       {
         m_error = wxURL_SNTXERR;
         return false;
       }
     }
-
-    // Extract full path
-    if (!PrepPath(last_url))
-    {
-      m_error = wxURL_NOPATH;
-      return false;
-    }
   }
-  // URL parse finished.
 
 #if wxUSE_SOCKETS
   if (m_useProxy)
@@ -135,11 +163,9 @@ bool wxURL::ParseURL()
     delete m_protocol;
 
     // Third, we rebuild the URL.
-    m_url = m_protoname + wxT(":");
+    m_url = m_scheme + wxT(":");
     if (m_protoinfo->m_needhost)
-      m_url = m_url + wxT("//") + m_hostname;
-
-    m_url += m_path;
+      m_url = m_url + wxT("//") + m_server;
 
     // We initialize specific variables.
     m_protocol = m_proxy; // FIXME: we should clone the protocol
@@ -150,6 +176,10 @@ bool wxURL::ParseURL()
   return true;
 }
 
+// --------------------------------------------------------------
+// Destruction/Cleanup
+// --------------------------------------------------------------
+
 void wxURL::CleanData()
 {
 #if wxUSE_SOCKETS
@@ -171,101 +201,22 @@ wxURL::~wxURL()
 }
 
 // --------------------------------------------------------------
-// --------- wxURL urls decoders --------------------------------
+// FetchProtocol
 // --------------------------------------------------------------
 
-bool wxURL::PrepProto(wxString& url)
-{
-  int pos;
-
-  // Find end
-  pos = url.Find(wxT(':'));
-  if (pos == wxNOT_FOUND)
-    return false;
-
-  m_protoname = url(0, pos);
-
-  url = url(pos+1, url.Length());
-
-  return true;
-}
-
-bool wxURL::PrepHost(wxString& url)
-{
-  wxString temp_url;
-  int pos, pos2;
-
-  if ((url.GetChar(0) != wxT('/')) || (url.GetChar(1) != wxT('/')))
-    return false;
-
-  url = url(2, url.Length());
-
-  pos = url.Find(wxT('/'));
-  if (pos == wxNOT_FOUND)
-    pos = url.Length();
-
-  if (pos == 0)
-    return false;
-
-  temp_url = url(0, pos);
-  url = url(url.Find(wxT('/')), url.Length());
-
-  // Retrieve service number
-  pos2 = temp_url.Find(wxT(':'), true);
-  if (pos2 != wxNOT_FOUND && pos2 < pos)
-  {
-    m_servname = temp_url(pos2+1, pos);
-    if (!m_servname.IsNumber())
-      return false;
-    temp_url = temp_url(0, pos2);
-  }
-
-  // Retrieve user and password.
-  pos2 = temp_url.Find(wxT('@'));
-  // Even if pos2 equals wxNOT_FOUND, this code is right.
-  m_hostname = temp_url(pos2+1, temp_url.Length());
-
-  m_user = wxT("");
-  m_password = wxT("");
-
-  if (pos2 == wxNOT_FOUND)
-    return true;
-
-  temp_url = temp_url(0, pos2);
-  pos2 = temp_url.Find(wxT(':'));
-
-  if (pos2 == wxNOT_FOUND)
-    return false;
-
-  m_user = temp_url(0, pos2);
-  m_password = temp_url(pos2+1, url.Length());
-
-  return true;
-}
-
-bool wxURL::PrepPath(wxString& url)
-{
-  if (url.Length() != 0)
-    m_path = ConvertToValidURI(url);
-  else
-    m_path = wxT("/");
-  return true;
-}
-
 bool wxURL::FetchProtocol()
 {
   wxProtoInfo *info = ms_protocols;
 
   while (info)
   {
-    if (m_protoname == info->m_protoname)
+    if (m_scheme == info->m_protoname)
     {
-      if (m_servname.IsNull())
-        m_servname = info->m_servname;
-
-      m_protoinfo = info;
-      m_protocol = (wxProtocol *)m_protoinfo->m_cinfo->CreateObject();
-      return true;
+        if (m_port.IsNull())
+            m_port = info->m_servname;
+        m_protoinfo = info;
+        m_protocol = (wxProtocol *)m_protoinfo->m_cinfo->CreateObject();
+        return true;
     }
     info = info->next;
   }
@@ -273,7 +224,7 @@ bool wxURL::FetchProtocol()
 }
 
 // --------------------------------------------------------------
-// --------- wxURL get ------------------------------------------
+// GetInputStream
 // --------------------------------------------------------------
 
 wxInputStream *wxURL::GetInputStream()
@@ -285,10 +236,17 @@ wxInputStream *wxURL::GetInputStream()
   }
 
   m_error = wxURL_NOERR;
-  if (m_user != wxT(""))
+  if (HasUser())
   {
-    m_protocol->SetUser(m_user);
-    m_protocol->SetPassword(m_password);
+      size_t dwPasswordPos = m_user.find(':');
+
+      if (dwPasswordPos == wxString::npos)
+          m_protocol->SetUser(m_user);
+      else
+      {
+          m_protocol->SetUser(m_user(0, dwPasswordPos));
+          m_protocol->SetPassword(m_user(dwPasswordPos+1, m_user.length() + 1));
+      }
   }
 
 #if wxUSE_URL_NATIVE
@@ -310,13 +268,13 @@ wxInputStream *wxURL::GetInputStream()
   // m_protoinfo is NULL when we use a proxy
   if (!m_useProxy && m_protoinfo->m_needhost)
   {
-    if (!addr.Hostname(m_hostname))
+    if (!addr.Hostname(m_server))
     {
       m_error = wxURL_NOHOST;
       return NULL;
     }
 
-    addr.Service(m_servname);
+    addr.Service(m_port);
 
     if (!m_protocol->Connect(addr, true)) // Watcom needs the 2nd arg for some reason
     {
@@ -327,9 +285,24 @@ wxInputStream *wxURL::GetInputStream()
 #endif
 
   // When we use a proxy, we have to pass the whole URL to it.
-  wxInputStream *the_i_stream =
-       (m_useProxy) ? m_protocol->GetInputStream(m_url) :
-                      m_protocol->GetInputStream(m_path);
+  wxInputStream *the_i_stream;
+  
+  if (!m_useProxy)
+  {
+      the_i_stream = m_protocol->GetInputStream(m_url);
+  }
+  else
+  {
+      wxString fullPath = m_path;
+
+      if (HasQuery())
+          fullPath += wxT("?") + m_query;
+      
+      if (HasFragment())
+          fullPath += wxT("#") + m_fragment;
+      
+      the_i_stream = m_protocol->GetInputStream(fullPath);
+  }
 
   if (!the_i_stream)
   {
@@ -423,6 +396,8 @@ void wxURL::SetProxy(const wxString& url_proxy)
 #endif // wxUSE_SOCKETS
 
 // ----------------------------------------------------------------------
+// wxURLModule
+//
 // A module which deletes the default proxy if we created it
 // ----------------------------------------------------------------------