/*
 *              bitlet - Simple bittorrent library
 *  Copyright (C) 2008 Alessandro Bahgat Shehata, Daniele Castagna
 *
 *  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 org.bitlet.wetorrent.peer.task;

import java.io.DataOutputStream;
import java.io.IOException;
import java.util.logging.Level;
import org.bitlet.wetorrent.Event;
import org.bitlet.wetorrent.Torrent;
import org.bitlet.wetorrent.peer.TorrentPeer;
import org.bitlet.wetorrent.util.stream.OutputStreamLimiter;
import org.bitlet.wetorrent.util.thread.ThreadTask;

public class SendBitfield implements ThreadTask {

    TorrentPeer peer;

    public SendBitfield(TorrentPeer peer) {
        this.peer = peer;
    }

    public boolean execute() throws Exception {
        if (Torrent.verbose) {
            peer.getPeersManager().getTorrent().addEvent(new Event(this, "Sending bitField ", Level.FINER));
        }
        try {
            DataOutputStream os = new DataOutputStream(new OutputStreamLimiter(peer.getSocket().getOutputStream(), peer.getPeersManager().getTorrent().getUploadBandwidthLimiter()));
            byte[] bitField = peer.getPeersManager().getTorrent().getTorrentDisk().getBitfieldCopy();

            os.writeInt(1 + bitField.length);
            os.writeByte(5);
            os.write(bitField);

        } catch (Exception e) {
            if (Torrent.verbose) {
                peer.getPeersManager().getTorrent().addEvent(new Event(this, "Problem sending bitfield", Level.WARNING));
            }
            throw e;
        }
        return false;
    }

    public void interrupt() {
        try {
            peer.getSocket().close();
        } catch (IOException ex) {
            ex.printStackTrace();
        }
    }

    public void exceptionCought(Exception e) {
        e.printStackTrace();
        peer.interrupt();
    }
}