package it.cavallium.dbengine.database;

import com.google.common.primitives.Ints;
import com.google.common.primitives.Longs;
import java.nio.ByteBuffer;
import java.nio.charset.StandardCharsets;
import java.util.Arrays;
import java.util.Objects;
import java.util.StringJoiner;
import org.apache.lucene.document.Field;

public class LLItem {

	private final LLType type;
	private final String name;
	private final byte[] data;

	public LLItem(LLType type, String name, byte[] data) {
		this.type = type;
		this.name = name;
		this.data = data;
	}

	private LLItem(LLType type, String name, String data) {
		this.type = type;
		this.name = name;
		this.data = data.getBytes(StandardCharsets.UTF_8);
	}

	private LLItem(LLType type, String name, int data) {
		this.type = type;
		this.name = name;
		this.data = Ints.toByteArray(data);
	}

	private LLItem(LLType type, String name, float data) {
		this.type = type;
		this.name = name;
		this.data = ByteBuffer.allocate(4).putFloat(data).array();
	}

	private LLItem(LLType type, String name, long data) {
		this.type = type;
		this.name = name;
		this.data = Longs.toByteArray(data);
	}

	private LLItem(LLType type, String name, int... data) {
		this.type = type;
		this.name = name;
		var ba = new byte[data.length * Integer.BYTES];
		for (int i = 0; i < data.length; i++) {
			System.arraycopy(Ints.toByteArray(data[i]), 0, ba, i * Integer.BYTES, Integer.BYTES);
		}
		this.data = ba;
	}

	private LLItem(LLType type, String name, long... data) {
		this.type = type;
		this.name = name;
		var ba = new byte[data.length * Long.BYTES];
		for (int i = 0; i < data.length; i++) {
			System.arraycopy(Longs.toByteArray(data[i]), 0, ba, i * Long.BYTES, Long.BYTES);
		}
		this.data = ba;
	}

	public static LLItem newIntPoint(String name, int data) {
		return new LLItem(LLType.IntPoint, name, data);
	}

	public static LLItem newIntPointND(String name, int... data) {
		return new LLItem(LLType.IntPointND, name, data);
	}

	public static LLItem newLongPoint(String name, long data) {
		return new LLItem(LLType.LongPoint, name, data);
	}

	public static LLItem newLongPointND(String name, long... data) {
		return new LLItem(LLType.LongPointND, name, data);
	}

	public static LLItem newLongStoredField(String name, long data) {
		return new LLItem(LLType.LongStoredField, name, data);
	}

	public static LLItem newFloatPoint(String name, float data) {
		return new LLItem(LLType.FloatPoint, name, data);
	}

	public static LLItem newTextField(String name, String data, Field.Store store) {
		if (store == Field.Store.YES) {
			return new LLItem(LLType.TextFieldStored, name, data);
		} else {
			return new LLItem(LLType.TextField, name, data);
		}
	}

	public static LLItem newStringField(String name, String data, Field.Store store) {
		if (store == Field.Store.YES) {
			return new LLItem(LLType.StringFieldStored, name, data);
		} else {
			return new LLItem(LLType.StringField, name, data);
		}
	}

	public static LLItem newSortedNumericDocValuesField(String name, long data) {
		return new LLItem(LLType.SortedNumericDocValuesField, name, data);
	}

	public static LLItem newNumericDocValuesField(String name, long data) {
		return new LLItem(LLType.NumericDocValuesField, name, data);
	}

	public String getName() {
		return name;
	}

	public LLType getType() {
		return type;
	}

	public byte[] getData() {
		return data;
	}

	@Override
	public boolean equals(Object o) {
		if (this == o) {
			return true;
		}
		if (o == null || getClass() != o.getClass()) {
			return false;
		}
		LLItem llItem = (LLItem) o;
		return type == llItem.type &&
				Objects.equals(name, llItem.name) &&
				Arrays.equals(data, llItem.data);
	}

	@Override
	public int hashCode() {
		int result = Objects.hash(type, name);
		result = 31 * result + Arrays.hashCode(data);
		return result;
	}

	@Override
	public String toString() {
		var sj = new StringJoiner(", ", "[", "]")
				.add("type=" + type)
				.add("name='" + name + "'");
		if (data != null && data.length > 0) {
			sj.add("data=" + new String(data));
		}
		return sj.toString();
	}

	public String stringValue() {
		return new String(data, StandardCharsets.UTF_8);
	}
}