7bbb08758f9a9840638a995769aec34c37ecf034
[WebKit-https.git] / WebCore / platform / qt / ResourceLoaderQt.cpp
1 /*
2  * Copyright (C) 2006 Nikolas Zimmermann <zimmermann@kde.org>
3  *
4  * All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
16  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
18  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
19  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
20  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
21  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
22  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
23  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
25  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
26  */
27
28 #include "config.h"
29
30 #include <kio/job.h>
31
32 #include "DocLoader.h"
33 #include "ResourceLoader.h"
34 #include "DeprecatedString.h"
35 #include "ResourceLoaderManager.h"
36 #include "ResourceLoaderInternal.h"
37
38 namespace WebCore {
39
40 ResourceLoaderInternal::~ResourceLoaderInternal()
41 {
42 }
43
44 ResourceLoader::~ResourceLoader()
45 {
46     cancel();
47 }
48
49 bool ResourceLoader::start(DocLoader*)
50 {
51     ref();
52     ResourceLoaderManager::self()->add(this);
53     return true;
54 }
55
56 void ResourceLoader::cancel()
57 {
58     ResourceLoaderManager::self()->cancel(this);
59 }
60
61 void ResourceLoader::assembleResponseHeaders() const
62 {
63     if (!d->assembledResponseHeaders) {
64         d->responseHeaders = d->m_response;
65         d->assembledResponseHeaders = true;
66         d->m_response = QString(); // Reset
67     }
68 }
69
70 void ResourceLoader::retrieveCharset() const
71 {
72     if (!d->retrievedCharset) {
73         d->retrievedCharset = true;
74         d->metaData.set("charset", extractCharsetFromHeaders(d->responseHeaders));
75     }
76 }
77
78 void ResourceLoader::receivedResponse(QString response)
79 {
80     Q_ASSERT(method() == "POST");
81
82     d->assembledResponseHeaders = false;
83     d->retrievedCharset = false;
84     d->m_response = response;
85
86     if (d->client)
87         d->client->receivedResponse(const_cast<ResourceLoader*>(this), response);
88 }
89
90 QString ResourceLoader::extractCharsetFromHeaders(QString headers) const
91 {
92     int pos = headers.find("content-type:", 0, false);
93
94     if (pos > -1) {
95         pos += 13;
96
97         int index = headers.find('\n', pos);
98         QString type = headers.mid(pos, index - pos);
99         index = type.indexOf(';');
100
101         if (index > -1)
102             return type.mid(index + 1).remove(QRegExp("charset[ ]*=[ ]*", Qt::CaseInsensitive)).trimmed();
103     }
104
105     return QString();
106 }
107
108 } // namespace WebCore