FileSponge/src/main/java/org/warp/filesponge/DiskMetadata.java

/*
 *     FileSponge
 *     Copyright (C) 2021 Andrea Cavalli
 *
 *     This program is free software: you can redistribute it and/or modify
 *     it under the terms of the GNU General Public License as published by
 *     the Free Software Foundation, either version 3 of the License, or
 *     (at your option) any later version.
 *
 *     This program is distributed in the hope that it will be useful,
 *     but WITHOUT ANY WARRANTY; without even the implied warranty of
 *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *     GNU General Public License for more details.
 *
 *     You should have received a copy of the GNU General Public License
 *     along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

package org.warp.filesponge;

import io.net5.buffer.api.Buffer;
import io.net5.buffer.api.BufferAllocator;
import io.net5.buffer.api.Send;
import it.cavallium.dbengine.database.serialization.BufferDataInputOwned;
import it.cavallium.dbengine.database.serialization.BufferDataInputShared;
import it.cavallium.dbengine.database.serialization.BufferDataOutput;
import it.cavallium.dbengine.database.serialization.SerializationException;
import it.cavallium.dbengine.database.serialization.Serializer;
import it.unimi.dsi.fastutil.booleans.BooleanArrayList;
import org.jetbrains.annotations.NotNull;

/**
 * size -1 = unknown size
 */
public record DiskMetadata(int size, BooleanArrayList downloadedBlocks) {

	public boolean isDownloadedFully() {
		boolean downloadedFullyVal;
		// Ensure blocks count is valid by calling getBlocksCount()
		getBlocksCount();
		// It's fully downloaded if every block is true
		downloadedFullyVal = !this.downloadedBlocks.contains(false);
		return downloadedFullyVal;
	}

	@SuppressWarnings("UnusedReturnValue")
	private int getBlocksCount() {
		if (size == -1) {
			return downloadedBlocks.size();
		}
		var expectedBlocksCount = getBlocksCount(size, FileSponge.BLOCK_SIZE);
		if (this.downloadedBlocks.size() != expectedBlocksCount) {
			throw new IllegalStateException(
					"Blocks array length (" + this.downloadedBlocks.size() + ") != expected blocks count ("
							+ expectedBlocksCount + ")");
		}
		return expectedBlocksCount;
	}

	public static int getBlocksCount(int size, int blockSize) {
		if (size == -1) {
			return 0;
		}
		return (size + (blockSize - size % blockSize)) / blockSize;
	}

	public Metadata asMetadata() {
		return new Metadata(size);
	}

	public boolean isDownloadedBlock(int id) {
		if (size == -1 && downloadedBlocks.size() <= id) {
			return false;
		} else {
			return downloadedBlocks.getBoolean(id);
		}
	}

	public static class DiskMetadataSerializer implements Serializer<DiskMetadata> {

		@Override
		public @NotNull DiskMetadata deserialize(@NotNull Buffer serialized) throws SerializationException {
			var dis = new BufferDataInputShared(serialized);
			int size = dis.readInt();
			int blocksCount;
			if (size == -1) {
				blocksCount = dis.readShort();
			} else {
				blocksCount = getBlocksCount(size, FileSponge.BLOCK_SIZE);
			}
			var downloadedBlocks = new BooleanArrayList(blocksCount);
			for (int i = 0; i < blocksCount; i++) {
				downloadedBlocks.add(dis.readBoolean());
			}
			return new DiskMetadata(size, downloadedBlocks);
		}

		@Override
		public void serialize(@NotNull DiskMetadata deserialized, Buffer output) throws SerializationException {
			var dos = new BufferDataOutput(output);
			dos.writeInt(deserialized.size);
			var blocksCount = deserialized.getBlocksCount();
			if (deserialized.size == -1) {
				dos.writeShort(blocksCount);
			}
			for (boolean downloadedBlock : deserialized.downloadedBlocks) {
				dos.writeBoolean(downloadedBlock);
			}
		}

		@Override
		public int getSerializedSizeHint() {
			return Integer.BYTES;
		}
	}
}