src/third_party/WebKit/Source/core/workers/WorkerScriptLoader.h

   1 /*
   2  * Copyright (C) 2009 Apple Inc. All Rights Reserved.
   3  * Copyright (C) 2009, 2011 Google Inc. All Rights Reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
  15  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  17  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
  18  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  19  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  20  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  21  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  22  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  24  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25  *
  26  */
  27
  28 #ifndef WorkerScriptLoader_h
  29 #define WorkerScriptLoader_h
  30
  31 #include "core/loader/ThreadableLoader.h"
  32 #include "core/loader/ThreadableLoaderClient.h"
  33 #include "platform/network/ResourceRequest.h"
  34 #include "platform/weborigin/KURL.h"
  35 #include "wtf/FastAllocBase.h"
  36 #include "wtf/PassRefPtr.h"
  37 #include "wtf/RefCounted.h"
  38 #include "wtf/text/StringBuilder.h"
  39
  40 namespace WebCore {
  41
  42     class ResourceRequest;
  43     class ResourceResponse;
  44     class ExecutionContext;
  45     class TextResourceDecoder;
  46     class WorkerScriptLoaderClient;
  47
  48     class WorkerScriptLoader FINAL : public RefCounted<WorkerScriptLoader>, public ThreadableLoaderClient {
  49         WTF_MAKE_FAST_ALLOCATED;
  50     public:
  51         static PassRefPtr<WorkerScriptLoader> create()
  52         {
  53             return adoptRef(new WorkerScriptLoader());
  54         }
  55
  56         void loadSynchronously(ExecutionContext&, const KURL&, CrossOriginRequestPolicy);
  57         void loadAsynchronously(ExecutionContext&, const KURL&, CrossOriginRequestPolicy, WorkerScriptLoaderClient*);
  58
  59         void notifyError();
  60
  61         // This will immediately lead to notifyFinished() if loadAsynchronously
  62         // is in progress.
  63         void cancel();
  64
  65         void setClient(WorkerScriptLoaderClient* client) { m_client = client; }
  66
  67         String script();
  68         const KURL& url() const { return m_url; }
  69         const KURL& responseURL() const;
  70         bool failed() const { return m_failed; }
  71         unsigned long identifier() const { return m_identifier; }
  72
  73         virtual void didReceiveResponse(unsigned long /*identifier*/, const ResourceResponse&) OVERRIDE;
  74         virtual void didReceiveData(const char* data, int dataLength) OVERRIDE;
  75         virtual void didFinishLoading(unsigned long identifier, double) OVERRIDE;
  76         virtual void didFail(const ResourceError&) OVERRIDE;
  77         virtual void didFailRedirectCheck() OVERRIDE;
  78
  79         void setTargetType(ResourceRequest::TargetType targetType) { m_targetType = targetType; }
  80
  81     private:
  82         friend class WTF::RefCounted<WorkerScriptLoader>;
  83
  84         WorkerScriptLoader();
  85         virtual ~WorkerScriptLoader();
  86
  87         PassOwnPtr<ResourceRequest> createResourceRequest();
  88         void notifyFinished();
  89
  90         WorkerScriptLoaderClient* m_client;
  91         RefPtr<ThreadableLoader> m_threadableLoader;
  92         String m_responseEncoding;
  93         OwnPtr<TextResourceDecoder> m_decoder;
  94         StringBuilder m_script;
  95         KURL m_url;
  96         KURL m_responseURL;
  97         bool m_failed;
  98         unsigned long m_identifier;
  99         bool m_finishing;
 100         ResourceRequest::TargetType m_targetType;
 101     };
 102
 103 } // namespace WebCore
 104
 105 #endif // WorkerScriptLoader_h