/*
-REMARKS :
+REMARKS :
This FS creates local cache (in /tmp directory). The cache is freed
on program exit.
*/
#ifdef __GNUG__
-#pragma implementation
+#pragma implementation "fs_inet.h"
#endif
-#include <wx/wxprec.h>
+#include "wx/wxprec.h"
#ifdef __BORDLANDC__
#pragma hdrstop
#endif
+#if !wxUSE_SOCKETS
+ #undef wxUSE_FS_INET
+ #define wxUSE_FS_INET 0
+#endif
+
+#if wxUSE_FILESYSTEM && wxUSE_FS_INET
+
#ifndef WXPRECOMP
-#include <wx/wx.h>
#endif
#include "wx/wfstream.h"
#include "wx/url.h"
#include "wx/filesys.h"
#include "wx/fs_inet.h"
+#include "wx/module.h"
class wxInetCacheNode : public wxObject
{
//--------------------------------------------------------------------------------
-bool wxInternetFSHandler::CanOpen(const wxString& location)
+static wxString StripProtocolAnchor(const wxString& location)
{
- wxString p = GetProtocol(location);
- return (p == "http") || (p == "ftp");
+ wxString myloc(location.BeforeLast(wxT('#')));
+ if (myloc.IsEmpty()) myloc = location.AfterFirst(wxT(':'));
+ else myloc = myloc.AfterFirst(wxT(':'));
+
+ // fix malformed url:
+ if (myloc.Left(2) != wxT("//"))
+ {
+ if (myloc.GetChar(0) != wxT('/')) myloc = wxT("//") + myloc;
+ else myloc = wxT("/") + myloc;
+ }
+ if (myloc.Mid(2).Find(wxT('/')) == wxNOT_FOUND) myloc << wxT('/');
+
+ return myloc;
}
+bool wxInternetFSHandler::CanOpen(const wxString& location)
+{
+ wxString p = GetProtocol(location);
+ if ((p == wxT("http")) || (p == wxT("ftp")))
+ {
+ wxURL url(p + wxT(":") + StripProtocolAnchor(location));
+ return (url.GetError() == wxURL_NOERR);
+ }
+
+ return FALSE;
+}
+
wxFSFile* wxInternetFSHandler::OpenFile(wxFileSystem& WXUNUSED(fs), const wxString& location)
{
- wxString right = GetProtocol(location) + ":" + GetRightLocation(location);
+ wxString right = GetProtocol(location) + wxT(":") + StripProtocolAnchor(location);
wxInputStream *s;
wxString content;
wxInetCacheNode *info;
info = (wxInetCacheNode*) m_Cache.Get(right);
// Add item into cache:
- if (info == NULL) {
+ if (info == NULL)
+ {
wxURL url(right);
- s = url.GetInputStream();
- content = url.GetProtocol().GetContentType();
- if (content == wxEmptyString) content = GetMimeTypeFromExt(location);
- if (s) {
- char buf[256];
-
- wxGetTempFileName("wxhtml", buf);
- info = new wxInetCacheNode(buf, content);
- m_Cache.Put(right, info);
-
- { // ok, now copy it:
- wxFileOutputStream sout(buf);
- s -> Read(sout); // copy the stream
+ if (url.GetError() == wxURL_NOERR)
+ {
+ s = url.GetInputStream();
+ content = url.GetProtocol().GetContentType();
+ if (content == wxEmptyString) content = GetMimeTypeFromExt(location);
+ if (s)
+ {
+ wxChar buf[256];
+
+ wxGetTempFileName( wxT("wxhtml"), buf);
+ info = new wxInetCacheNode(buf, content);
+ m_Cache.Put(right, info);
+
+ { // ok, now copy it:
+ wxFileOutputStream sout((wxString)buf);
+ s -> Read(sout); // copy the stream
+ }
+ delete s;
}
- delete s;
+ else
+ return (wxFSFile*) NULL; // we can't open the URL
}
- else return NULL; //we can't open the URL
+ else
+ return (wxFSFile*) NULL; // incorrect URL
}
// Load item from cache:
- s = new wxFileInputStream(info -> GetTemp());
- if (s) {
- return new wxFSFile(s,
- right,
- info -> GetMime(),
- GetAnchor(location));
- }
- else return NULL;
+ s = new wxFileInputStream(info->GetTemp());
+ if (!s)
+ return (wxFSFile*) NULL;
+
+ return new wxFSFile(s,
+ right,
+ info->GetMime(),
+ GetAnchor(location),
+ wxDateTime::Now());
}
wxInetCacheNode *n2;
m_Cache.BeginFind();
- while ((n = m_Cache.Next()) != NULL) {
- n2 = (wxInetCacheNode*) n -> GetData();
- wxRemoveFile(n2 -> GetTemp());
+ while ((n = m_Cache.Next()) != NULL)
+ {
+ n2 = (wxInetCacheNode*) n->GetData();
+ wxRemoveFile(n2->GetTemp());
delete n2;
}
}
+class wxFileSystemInternetModule : public wxModule
+{
+ DECLARE_DYNAMIC_CLASS(wxFileSystemInternetModule)
+ public:
+ virtual bool OnInit()
+ {
+ wxFileSystem::AddHandler(new wxInternetFSHandler);
+ return TRUE;
+ }
+ virtual void OnExit() {}
+};
+IMPLEMENT_DYNAMIC_CLASS(wxFileSystemInternetModule, wxModule)
+#endif // wxUSE_FILESYSTEM && wxUSE_FS_INET