]>
Commit | Line | Data |
---|---|---|
5526e819 VS |
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 | ||
e3e717ec | 11 | REMARKS : |
5526e819 VS |
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 | ||
3096bd2f | 26 | #include "wx/wxprec.h" |
5526e819 VS |
27 | |
28 | #ifdef __BORDLANDC__ | |
29 | #pragma hdrstop | |
30 | #endif | |
31 | ||
31528cd3 VZ |
32 | #if !wxUSE_SOCKETS |
33 | #undef wxUSE_FS_INET | |
34 | #define wxUSE_FS_INET 0 | |
35 | #endif | |
36 | ||
e3e717ec VZ |
37 | #if wxUSE_FS_INET |
38 | ||
5526e819 | 39 | #ifndef WXPRECOMP |
3096bd2f | 40 | #include "wx/wx.h" |
5526e819 VS |
41 | #endif |
42 | ||
43 | #include "wx/wfstream.h" | |
44 | #include "wx/url.h" | |
45 | #include "wx/filesys.h" | |
46 | #include "wx/fs_inet.h" | |
cf1f0870 | 47 | #include "wx/module.h" |
5526e819 VS |
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 | bool wxInternetFSHandler::CanOpen(const wxString& location) | |
71 | { | |
72 | wxString p = GetProtocol(location); | |
81915974 VS |
73 | if ((p == wxT("http")) || (p == wxT("ftp"))) |
74 | { | |
75 | wxURL url(GetProtocol(location) + wxT(":") + GetRightLocation(location)); | |
76 | return (url.GetError() == wxURL_NOERR); | |
77 | } | |
78 | else | |
79 | return FALSE; | |
5526e819 VS |
80 | } |
81 | ||
82 | ||
5526e819 VS |
83 | wxFSFile* wxInternetFSHandler::OpenFile(wxFileSystem& WXUNUSED(fs), const wxString& location) |
84 | { | |
223d09f6 | 85 | wxString right = GetProtocol(location) + wxT(":") + GetRightLocation(location); |
5526e819 VS |
86 | wxInputStream *s; |
87 | wxString content; | |
88 | wxInetCacheNode *info; | |
89 | ||
90 | info = (wxInetCacheNode*) m_Cache.Get(right); | |
91 | ||
92 | // Add item into cache: | |
f61815af | 93 | if (info == NULL) |
324dbfec | 94 | { |
5526e819 | 95 | wxURL url(right); |
81915974 | 96 | if (url.GetError() == wxURL_NOERR) |
07d4b94c | 97 | { |
81915974 VS |
98 | s = url.GetInputStream(); |
99 | content = url.GetProtocol().GetContentType(); | |
100 | if (content == wxEmptyString) content = GetMimeTypeFromExt(location); | |
101 | if (s) | |
102 | { | |
103 | wxChar buf[256]; | |
104 | ||
105 | wxGetTempFileName( wxT("wxhtml"), buf); | |
106 | info = new wxInetCacheNode(buf, content); | |
107 | m_Cache.Put(right, info); | |
108 | ||
109 | { // ok, now copy it: | |
110 | wxFileOutputStream sout((wxString)buf); | |
111 | s -> Read(sout); // copy the stream | |
112 | } | |
113 | delete s; | |
5526e819 | 114 | } |
81915974 VS |
115 | else |
116 | return (wxFSFile*) NULL; // we can't open the URL | |
5526e819 | 117 | } |
324dbfec | 118 | else |
81915974 | 119 | return (wxFSFile*) NULL; // incorrect URL |
5526e819 VS |
120 | } |
121 | ||
122 | // Load item from cache: | |
324dbfec | 123 | s = new wxFileInputStream(info->GetTemp()); |
913df6f2 | 124 | if (s) |
324dbfec | 125 | { |
5526e819 VS |
126 | return new wxFSFile(s, |
127 | right, | |
324dbfec | 128 | info->GetMime(), |
5526e819 VS |
129 | GetAnchor(location)); |
130 | } | |
324dbfec | 131 | else return (wxFSFile*) NULL; |
5526e819 VS |
132 | } |
133 | ||
134 | ||
135 | ||
136 | wxInternetFSHandler::~wxInternetFSHandler() | |
137 | { | |
138 | wxNode *n; | |
139 | wxInetCacheNode *n2; | |
140 | ||
141 | m_Cache.BeginFind(); | |
913df6f2 | 142 | while ((n = m_Cache.Next()) != NULL) |
324dbfec RR |
143 | { |
144 | n2 = (wxInetCacheNode*) n->GetData(); | |
145 | wxRemoveFile(n2->GetTemp()); | |
5526e819 VS |
146 | delete n2; |
147 | } | |
148 | } | |
149 | ||
f61815af GL |
150 | class wxFileSystemInternetModule : public wxModule |
151 | { | |
152 | DECLARE_DYNAMIC_CLASS(wxFileSystemInternetModule) | |
153 | ||
154 | public: | |
155 | virtual bool OnInit() | |
156 | { | |
157 | wxFileSystem::AddHandler(new wxInternetFSHandler); | |
158 | return TRUE; | |
159 | } | |
160 | virtual void OnExit() {} | |
161 | }; | |
162 | ||
163 | IMPLEMENT_DYNAMIC_CLASS(wxFileSystemInternetModule, wxModule) | |
164 | ||
e3e717ec | 165 | #endif // wxUSE_FS_INET |