]>
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 | |
65571936 | 6 | // Licence: wxWindows licence |
5526e819 VS |
7 | ///////////////////////////////////////////////////////////////////////////// |
8 | ||
14f355c2 | 9 | #if defined(__GNUG__) && !defined(NO_GCC_PRAGMA) |
c3f4609e | 10 | #pragma implementation "fs_inet.h" |
5526e819 VS |
11 | #endif |
12 | ||
3096bd2f | 13 | #include "wx/wxprec.h" |
5526e819 | 14 | |
2b5f62a0 | 15 | #ifdef __BORLANDC__ |
5526e819 VS |
16 | #pragma hdrstop |
17 | #endif | |
18 | ||
31528cd3 VZ |
19 | #if !wxUSE_SOCKETS |
20 | #undef wxUSE_FS_INET | |
21 | #define wxUSE_FS_INET 0 | |
22 | #endif | |
23 | ||
24528b0c | 24 | #if wxUSE_FILESYSTEM && wxUSE_FS_INET |
e3e717ec | 25 | |
5526e819 | 26 | #ifndef WXPRECOMP |
5526e819 VS |
27 | #endif |
28 | ||
29 | #include "wx/wfstream.h" | |
30 | #include "wx/url.h" | |
31 | #include "wx/filesys.h" | |
32 | #include "wx/fs_inet.h" | |
cf1f0870 | 33 | #include "wx/module.h" |
5526e819 | 34 | |
3caa34b7 VS |
35 | // ---------------------------------------------------------------------------- |
36 | // Helper classes | |
37 | // ---------------------------------------------------------------------------- | |
38 | ||
39 | // This stream deletes the file when destroyed | |
40 | class wxTemporaryFileInputStream : public wxFileInputStream | |
5526e819 | 41 | { |
3caa34b7 VS |
42 | public: |
43 | wxTemporaryFileInputStream(const wxString& filename) : | |
44 | wxFileInputStream(filename), m_filename(filename) {} | |
a62848fd | 45 | |
3caa34b7 VS |
46 | ~wxTemporaryFileInputStream() |
47 | { | |
48 | // NB: copied from wxFileInputStream dtor, we need to do it before | |
49 | // wxRemoveFile | |
50 | if (m_file_destroy) | |
51 | { | |
52 | delete m_file; | |
53 | m_file_destroy = false; | |
54 | } | |
a62848fd | 55 | wxRemoveFile(m_filename); |
3caa34b7 | 56 | } |
5526e819 | 57 | |
3caa34b7 VS |
58 | protected: |
59 | wxString m_filename; | |
5526e819 VS |
60 | }; |
61 | ||
62 | ||
3caa34b7 | 63 | // ---------------------------------------------------------------------------- |
5526e819 | 64 | // wxInternetFSHandler |
3caa34b7 | 65 | // ---------------------------------------------------------------------------- |
5526e819 | 66 | |
c3a1c7d2 VS |
67 | static wxString StripProtocolAnchor(const wxString& location) |
68 | { | |
69 | wxString myloc(location.BeforeLast(wxT('#'))); | |
70 | if (myloc.IsEmpty()) myloc = location.AfterFirst(wxT(':')); | |
71 | else myloc = myloc.AfterFirst(wxT(':')); | |
72 | ||
73 | // fix malformed url: | |
04dbb646 | 74 | if (myloc.Left(2) != wxT("//")) |
c3a1c7d2 VS |
75 | { |
76 | if (myloc.GetChar(0) != wxT('/')) myloc = wxT("//") + myloc; | |
77 | else myloc = wxT("/") + myloc; | |
78 | } | |
79 | if (myloc.Mid(2).Find(wxT('/')) == wxNOT_FOUND) myloc << wxT('/'); | |
80 | ||
81 | return myloc; | |
82 | } | |
83 | ||
84 | ||
5526e819 VS |
85 | bool wxInternetFSHandler::CanOpen(const wxString& location) |
86 | { | |
87 | wxString p = GetProtocol(location); | |
04dbb646 | 88 | if ((p == wxT("http")) || (p == wxT("ftp"))) |
81915974 | 89 | { |
c3a1c7d2 | 90 | wxURL url(p + wxT(":") + StripProtocolAnchor(location)); |
81915974 VS |
91 | return (url.GetError() == wxURL_NOERR); |
92 | } | |
f6bcfd97 | 93 | |
a62848fd | 94 | return false; |
5526e819 VS |
95 | } |
96 | ||
97 | ||
3caa34b7 VS |
98 | wxFSFile* wxInternetFSHandler::OpenFile(wxFileSystem& WXUNUSED(fs), |
99 | const wxString& location) | |
5526e819 | 100 | { |
3caa34b7 VS |
101 | wxString right = |
102 | GetProtocol(location) + wxT(":") + StripProtocolAnchor(location); | |
5526e819 | 103 | |
3caa34b7 VS |
104 | wxURL url(right); |
105 | if (url.GetError() == wxURL_NOERR) | |
324dbfec | 106 | { |
3caa34b7 VS |
107 | wxInputStream *s = url.GetInputStream(); |
108 | wxString content = url.GetProtocol().GetContentType(); | |
109 | if (content == wxEmptyString) content = GetMimeTypeFromExt(location); | |
110 | if (s) | |
07d4b94c | 111 | { |
3caa34b7 VS |
112 | wxString tmpfile = |
113 | wxFileName::CreateTempFileName(wxT("wxhtml")); | |
5526e819 | 114 | |
3caa34b7 VS |
115 | { // now copy streams content to temporary file: |
116 | wxFileOutputStream sout(tmpfile); | |
117 | s->Read(sout); | |
118 | } | |
119 | delete s; | |
a62848fd | 120 | |
3caa34b7 VS |
121 | return new wxFSFile(new wxTemporaryFileInputStream(tmpfile), |
122 | right, | |
123 | content, | |
124 | GetAnchor(location) | |
e2b87f38 | 125 | #if wxUSE_DATETIME |
3caa34b7 | 126 | , wxDateTime::Now() |
e2b87f38 VZ |
127 | #endif // wxUSE_DATETIME |
128 | ); | |
3caa34b7 | 129 | } |
5526e819 | 130 | } |
3caa34b7 VS |
131 | |
132 | return (wxFSFile*) NULL; // incorrect URL | |
5526e819 VS |
133 | } |
134 | ||
3caa34b7 | 135 | |
f61815af GL |
136 | class wxFileSystemInternetModule : public wxModule |
137 | { | |
138 | DECLARE_DYNAMIC_CLASS(wxFileSystemInternetModule) | |
139 | ||
140 | public: | |
141 | virtual bool OnInit() | |
142 | { | |
143 | wxFileSystem::AddHandler(new wxInternetFSHandler); | |
a62848fd | 144 | return true; |
f61815af GL |
145 | } |
146 | virtual void OnExit() {} | |
147 | }; | |
148 | ||
149 | IMPLEMENT_DYNAMIC_CLASS(wxFileSystemInternetModule, wxModule) | |
150 | ||
24528b0c | 151 | #endif // wxUSE_FILESYSTEM && wxUSE_FS_INET |