package eu.siacs.conversations.entities;

import java.io.File;

import eu.siacs.conversations.utils.MimeUtils;

public class DownloadableFile extends File {

	private static final long serialVersionUID = 2247012619505115863L;

	private long expectedSize = 0;
	private byte[] sha1sum;
	private byte[] aeskey;
	private byte[] iv;

	public DownloadableFile(String path) {
		super(path);
	}

	public long getSize() {
		return super.length();
	}

	public long getExpectedSize() {
		return this.expectedSize;
	}

	public String getMimeType() {
		String path = this.getAbsolutePath();
		int start = path.lastIndexOf('.') + 1;
		if (start < path.length()) {
			String mime = MimeUtils.guessMimeTypeFromExtension(path.substring(start));
			return mime == null ? "" : mime;
		} else {
			return "";
		}
	}

	public void setExpectedSize(long size) {
		this.expectedSize = size;
	}

	public byte[] getSha1Sum() {
		return this.sha1sum;
	}

	public void setSha1Sum(byte[] sum) {
		this.sha1sum = sum;
	}

	public void setKeyAndIv(byte[] keyIvCombo) {
		// originally, we used a 16 byte IV, then found for aes-gcm a 12 byte IV is ideal
		// this code supports reading either length, with sending 12 byte IV to be done in future
		if (keyIvCombo.length == 48) {
			this.aeskey = new byte[32];
			this.iv = new byte[16];
			System.arraycopy(keyIvCombo, 0, this.iv, 0, 16);
			System.arraycopy(keyIvCombo, 16, this.aeskey, 0, 32);
		} else if (keyIvCombo.length == 44) {
			this.aeskey = new byte[32];
			this.iv = new byte[12];
			System.arraycopy(keyIvCombo, 0, this.iv, 0, 12);
			System.arraycopy(keyIvCombo, 12, this.aeskey, 0, 32);
		} else if (keyIvCombo.length >= 32) {
			this.aeskey = new byte[32];
			System.arraycopy(keyIvCombo, 0, aeskey, 0, 32);
		}
	}

	public void setKey(byte[] key) {
		this.aeskey = key;
	}

	public void setIv(byte[] iv) {
		this.iv = iv;
	}

	public byte[] getKey() {
		return this.aeskey;
	}

	public byte[] getIv() {
		return this.iv;
	}
}