Home | History | Annotate | Download | only in workers 
      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
    106