1/*
2 * Copyright (C) 2010 Google Inc.  All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are
6 * met:
7 *
8 *     * Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 *     * Redistributions in binary form must reproduce the above
11 * copyright notice, this list of conditions and the following disclaimer
12 * in the documentation and/or other materials provided with the
13 * distribution.
14 *     * Neither the name of Google Inc. nor the names of its
15 * contributors may be used to endorse or promote products derived from
16 * this software without specific prior written permission.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 */
30
31#ifndef FileWriter_h
32#define FileWriter_h
33
34#if ENABLE(FILE_SYSTEM)
35
36#include "ActiveDOMObject.h"
37#include "EventTarget.h"
38#include "FileWriterBase.h"
39#include <wtf/PassRefPtr.h>
40#include <wtf/RefPtr.h>
41
42namespace WebCore {
43
44class ScriptExecutionContext;
45
46class FileWriter : public FileWriterBase, public ActiveDOMObject, public EventTarget, public AsyncFileWriterClient {
47public:
48    static PassRefPtr<FileWriter> create(ScriptExecutionContext* context)
49    {
50        return adoptRef(new FileWriter(context));
51    }
52
53    enum ReadyState {
54        INIT = 0,
55        WRITING = 1,
56        DONE = 2
57    };
58
59    void write(Blob*, ExceptionCode&);
60    void seek(long long position, ExceptionCode&);
61    void truncate(long long length, ExceptionCode&);
62    void abort(ExceptionCode&);
63    ReadyState readyState() const { return m_readyState; }
64    FileError* error() const { return m_error.get(); }
65
66    // AsyncFileWriterClient
67    void didWrite(long long bytes, bool complete);
68    void didTruncate();
69    void didFail(FileError::ErrorCode);
70
71    // ActiveDOMObject
72    virtual bool canSuspend() const;
73    virtual bool hasPendingActivity() const;
74    virtual void stop();
75
76    // EventTarget
77    virtual FileWriter* toFileWriter() { return this; }
78    virtual ScriptExecutionContext* scriptExecutionContext() const { return ActiveDOMObject::scriptExecutionContext(); }
79
80    using RefCounted<FileWriterBase>::ref;
81    using RefCounted<FileWriterBase>::deref;
82
83    DEFINE_ATTRIBUTE_EVENT_LISTENER(writestart);
84    DEFINE_ATTRIBUTE_EVENT_LISTENER(progress);
85    DEFINE_ATTRIBUTE_EVENT_LISTENER(write);
86    DEFINE_ATTRIBUTE_EVENT_LISTENER(abort);
87    DEFINE_ATTRIBUTE_EVENT_LISTENER(error);
88    DEFINE_ATTRIBUTE_EVENT_LISTENER(writeend);
89
90private:
91    FileWriter(ScriptExecutionContext*);
92
93    virtual ~FileWriter();
94
95    // EventTarget
96    virtual void refEventTarget() { ref(); }
97    virtual void derefEventTarget() { deref(); }
98    virtual EventTargetData* eventTargetData() { return &m_eventTargetData; }
99    virtual EventTargetData* ensureEventTargetData() { return &m_eventTargetData; }
100
101    void fireEvent(const AtomicString& type);
102
103    void setError(FileError::ErrorCode, ExceptionCode&);
104
105    RefPtr<FileError> m_error;
106    EventTargetData m_eventTargetData;
107    ReadyState m_readyState;
108    bool m_startedWriting;
109    long long m_bytesWritten;
110    long long m_bytesToWrite;
111    long long m_truncateLength;
112    RefPtr<Blob> m_blobBeingWritten;
113};
114
115} // namespace WebCore
116
117#endif // ENABLE(FILE_SYSTEM)
118
119#endif // FileWriter_h
120