package types

import "bytes"

var (
	// Slot constants (Ethereum-style storage layout)
	CLIENT_STATE [32]byte = [32]byte{
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
	}
	CONSENSUS_STATE [32]byte = [32]byte{
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01,
	}
	CONNECTIONS [32]byte = [32]byte{
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02,
	}
	CHANNELS [32]byte = [32]byte{
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x03,
	}
	PACKETS [32]byte = [32]byte{
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04,
	}
	PACKET_ACKS [32]byte = [32]byte{
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x05,
	}
	MEMBERSHIP_PROOF [32]byte = [32]byte{
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x06,
	}
	NON_MEMBERSHIP_PROOF [32]byte = [32]byte{
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x07,
	}
	// Sentinel values stored for commitments
	COMMITMENT_MAGIC H256 = [32]byte{
		0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
	}
	COMMITMENT_MAGIC_ACK H256 = [32]byte{
		0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
	}
	NON_MEMBERSHIP_COMMITMENT_VALUE H256 = [32]byte{
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01,
	}
)

func ClientStatePath(clientId ClientId) H256 {
	return Keccak(bytes.Join([][]byte{
		CLIENT_STATE[:],
		u32ToH256(uint32(clientId)),
	}, nil))
}

func ConsensusStatePath(clientId ClientId, height Height) H256 {
	return Keccak(bytes.Join([][]byte{
		CONSENSUS_STATE[:],
		u32ToH256(uint32(clientId)),
		u64ToH256(height.RevisionHeight),
	}, nil))
}

func ConnectionPath(connectionId ConnectionId) H256 {
	return Keccak(bytes.Join([][]byte{
		CONNECTIONS[:],
		u32ToH256(uint32(connectionId)),
	}, nil))
}

func ChannelPath(channelId ChannelId) H256 {
	return Keccak(bytes.Join([][]byte{
		CHANNELS[:],
		u32ToH256(uint32(channelId)),
	}, nil))
}

func mergeAck(ack H256) H256 {
	ack[0] = COMMITMENT_MAGIC[0]
	return ack
}

func PacketCommitmentPath(packet Packet) (H256, error) {
	hash, err := CommitPacket(packet)
	if err != nil {
		return H256{}, err
	}

	return BatchPacketsPath(hash), nil
}

func PacketAcknowledgementPath(packet Packet) (H256, error) {
	hash, err := CommitPacket(packet)
	if err != nil {
		return H256{}, err
	}

	return BatchReceiptsPath(hash), nil
}

func BatchPacketsPath(batchHash H256) H256 {
	return Keccak(bytes.Join([][]byte{
		PACKETS[:],
		batchHash[:],
	}, nil))
}

func BatchReceiptsPath(batchHash H256) H256 {
	return Keccak(bytes.Join([][]byte{
		PACKET_ACKS[:],
		batchHash[:],
	}, nil))
}

func MembershipProofPath(clientId ClientId, proofHeight uint64, path []byte) H256 {
	return Keccak(bytes.Join([][]byte{
		MEMBERSHIP_PROOF[:],
		u32ToH256(uint32(clientId)),
		u64ToH256(proofHeight),
		path,
	}, nil))
}

func NonMembershipProofPath(clientId ClientId, proofHeight uint64, path []byte) H256 {
	return Keccak(bytes.Join([][]byte{
		NON_MEMBERSHIP_PROOF[:],
		u32ToH256(uint32(clientId)),
		u64ToH256(proofHeight),
		path,
	}, nil))
}