]> git.saurik.com Git - wxWidgets.git/blob - src/common/fs_inet.cpp
More debug code
[wxWidgets.git] / src / common / fs_inet.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: fs_inet.cpp
3 // Purpose: HTTP and FTP file system
4 // Author: Vaclav Slavik
5 // Copyright: (c) 1999 Vaclav Slavik
6 // Licence: wxWindows Licence
7 /////////////////////////////////////////////////////////////////////////////
8
9 /*
10
11 REMARKS :
12
13 This FS creates local cache (in /tmp directory). The cache is freed
14 on program exit.
15
16 Size of cache is limited to cca 1000 items (due to GetTempFileName
17 limitation)
18
19
20 */
21
22 #ifdef __GNUG__
23 #pragma implementation
24 #endif
25
26 #include "wx/wxprec.h"
27
28 #ifdef __BORDLANDC__
29 #pragma hdrstop
30 #endif
31
32 #if !wxUSE_SOCKETS
33 #undef wxUSE_FS_INET
34 #define wxUSE_FS_INET 0
35 #endif
36
37 #if wxUSE_FILESYSTEM && wxUSE_FS_INET
38
39 #ifndef WXPRECOMP
40 #include "wx/wx.h"
41 #endif
42
43 #include "wx/wfstream.h"
44 #include "wx/url.h"
45 #include "wx/filesys.h"
46 #include "wx/fs_inet.h"
47 #include "wx/module.h"
48
49 class wxInetCacheNode : public wxObject
50 {
51 private:
52 wxString m_Temp;
53 wxString m_Mime;
54
55 public:
56 wxInetCacheNode(const wxString& l, const wxString& m) : wxObject() {m_Temp = l; m_Mime = m;}
57 const wxString& GetTemp() const {return m_Temp;}
58 const wxString& GetMime() const {return m_Mime;}
59 };
60
61
62
63
64
65 //--------------------------------------------------------------------------------
66 // wxInternetFSHandler
67 //--------------------------------------------------------------------------------
68
69
70 static wxString StripProtocolAnchor(const wxString& location)
71 {
72 wxString myloc(location.BeforeLast(wxT('#')));
73 if (myloc.IsEmpty()) myloc = location.AfterFirst(wxT(':'));
74 else myloc = myloc.AfterFirst(wxT(':'));
75
76 // fix malformed url:
77 if (myloc.Left(2) != wxT("//"))
78 {
79 if (myloc.GetChar(0) != wxT('/')) myloc = wxT("//") + myloc;
80 else myloc = wxT("/") + myloc;
81 }
82 if (myloc.Mid(2).Find(wxT('/')) == wxNOT_FOUND) myloc << wxT('/');
83
84 return myloc;
85 }
86
87
88
89 bool wxInternetFSHandler::CanOpen(const wxString& location)
90 {
91 wxString p = GetProtocol(location);
92 if ((p == wxT("http")) || (p == wxT("ftp")))
93 {
94 wxURL url(p + wxT(":") + StripProtocolAnchor(location));
95 return (url.GetError() == wxURL_NOERR);
96 }
97 else
98 return FALSE;
99 }
100
101
102 wxFSFile* wxInternetFSHandler::OpenFile(wxFileSystem& WXUNUSED(fs), const wxString& location)
103 {
104 wxString right = GetProtocol(location) + wxT(":") + StripProtocolAnchor(location);
105 wxInputStream *s;
106 wxString content;
107 wxInetCacheNode *info;
108
109 info = (wxInetCacheNode*) m_Cache.Get(right);
110
111 // Add item into cache:
112 if (info == NULL)
113 {
114 wxURL url(right);
115 if (url.GetError() == wxURL_NOERR)
116 {
117 s = url.GetInputStream();
118 content = url.GetProtocol().GetContentType();
119 if (content == wxEmptyString) content = GetMimeTypeFromExt(location);
120 if (s)
121 {
122 wxChar buf[256];
123
124 wxGetTempFileName( wxT("wxhtml"), buf);
125 info = new wxInetCacheNode(buf, content);
126 m_Cache.Put(right, info);
127
128 { // ok, now copy it:
129 wxFileOutputStream sout((wxString)buf);
130 s -> Read(sout); // copy the stream
131 }
132 delete s;
133 }
134 else
135 return (wxFSFile*) NULL; // we can't open the URL
136 }
137 else
138 return (wxFSFile*) NULL; // incorrect URL
139 }
140
141 // Load item from cache:
142 s = new wxFileInputStream(info->GetTemp());
143 if (s)
144 {
145 return new wxFSFile(s,
146 right,
147 info->GetMime(),
148 GetAnchor(location),
149 wxDateTime::Now());
150 }
151 else return (wxFSFile*) NULL;
152 }
153
154
155
156 wxInternetFSHandler::~wxInternetFSHandler()
157 {
158 wxNode *n;
159 wxInetCacheNode *n2;
160
161 m_Cache.BeginFind();
162 while ((n = m_Cache.Next()) != NULL)
163 {
164 n2 = (wxInetCacheNode*) n->GetData();
165 wxRemoveFile(n2->GetTemp());
166 delete n2;
167 }
168 }
169
170 class wxFileSystemInternetModule : public wxModule
171 {
172 DECLARE_DYNAMIC_CLASS(wxFileSystemInternetModule)
173
174 public:
175 virtual bool OnInit()
176 {
177 wxFileSystem::AddHandler(new wxInternetFSHandler);
178 return TRUE;
179 }
180 virtual void OnExit() {}
181 };
182
183 IMPLEMENT_DYNAMIC_CLASS(wxFileSystemInternetModule, wxModule)
184
185 #endif // wxUSE_FILESYSTEM && wxUSE_FS_INET