/* Copyright (C) 2011 The Android Open Source Project.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.android.exchange.adapter;

import com.android.exchange.eas.EasLoadAttachment.ProgressCallback;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

/**
 * Parse the result of an ItemOperations command; we use this to load attachments in EAS 14.0
 */
public class ItemOperationsParser extends Parser {
    private static final int CHUNK_SIZE = 16*1024;

    private int mStatusCode = 0;
    private final OutputStream mAttachmentOutputStream;
    private final long mAttachmentSize;
    private final ProgressCallback mCallback;

    public ItemOperationsParser(final InputStream in, final OutputStream out, final long size,
            final ProgressCallback callback) throws IOException {
        super(in);
        mAttachmentOutputStream = out;
        mAttachmentSize = size;
        mCallback = callback;
    }

    public int getStatusCode() {
        return mStatusCode;
    }

    private void parseProperties() throws IOException {
        while (nextTag(Tags.ITEMS_PROPERTIES) != END) {
            if (tag == Tags.ITEMS_DATA) {
                // Wrap the input stream in our custom base64 input stream
                Base64InputStream bis = new Base64InputStream(getInput());
                // Read the attachment
                readChunked(bis, mAttachmentOutputStream, mAttachmentSize, mCallback);
            } else {
                skipTag();
            }
        }
    }

    private void parseFetch() throws IOException {
        while (nextTag(Tags.ITEMS_FETCH) != END) {
            if (tag == Tags.ITEMS_PROPERTIES) {
                parseProperties();
            } else {
                skipTag();
            }
        }
    }

    private void parseResponse() throws IOException {
        while (nextTag(Tags.ITEMS_RESPONSE) != END) {
            if (tag == Tags.ITEMS_FETCH) {
                parseFetch();
            } else {
                skipTag();
            }
        }
    }

    @Override
    public boolean parse() throws IOException {
        boolean res = false;
        if (nextTag(START_DOCUMENT) != Tags.ITEMS_ITEMS) {
            throw new IOException();
        }
        while (nextTag(START_DOCUMENT) != END_DOCUMENT) {
            if (tag == Tags.ITEMS_STATUS) {
                // Save the status code
                mStatusCode = getValueInt();
            } else if (tag == Tags.ITEMS_RESPONSE) {
                parseResponse();
            } else {
                skipTag();
            }
        }
        return res;
    }

    /**
     * Read the attachment data in chunks and write the data back out to our attachment file
     * @param inputStream the InputStream we're reading the attachment from
     * @param outputStream the OutputStream the attachment will be written to
     * @param length the number of expected bytes we're going to read
     * @param callback A {@link ProgressCallback} to use to send progress updates to the UI.
     * @throws IOException
     */
    public static void readChunked(final InputStream inputStream, final OutputStream outputStream,
            final long length, final ProgressCallback callback) throws IOException {
        final byte[] bytes = new byte[CHUNK_SIZE];
        // Loop terminates 1) when EOF is reached or 2) IOException occurs
        // One of these is guaranteed to occur
        int totalRead = 0;
        long lastCallbackPct = -1;
        int lastCallbackTotalRead = 0;
        while (true) {
            final int read = inputStream.read(bytes, 0, CHUNK_SIZE);
            if (read < 0) {
                // -1 means EOF
                break;
            }

            // Keep track of how much we've read for progress callback
            totalRead += read;
            // Write these bytes out
            outputStream.write(bytes, 0, read);

            // We can't report percentage if data is chunked; the length of incoming data is unknown
            if (length > 0) {
                final int pct = (int)((totalRead * 100) / length);
                // Callback only if we've read at least 1% more and have read more than CHUNK_SIZE
                // We don't want to spam the Email app
                if ((pct > lastCallbackPct) && (totalRead > (lastCallbackTotalRead + CHUNK_SIZE))) {
                    // Report progress back to the UI
                    callback.doCallback(pct);

                    // TODO: Fix this.
                    //doProgressCallback(pct);
                    lastCallbackTotalRead = totalRead;
                    lastCallbackPct = pct;
                }
            }
        }
    }
}