package metadata

import (
	"errors"
	"git.lumeweb.com/LumeWeb/libs5-go/serialize"
	"git.lumeweb.com/LumeWeb/libs5-go/types"
	"github.com/vmihailenco/msgpack/v5"
)

type directoryReferenceMap map[string]DirectoryReference
type fileReferenceMap map[string]FileReference

type DirectoryMetadata struct {
	Details       DirectoryMetadataDetails `json:"details"`
	Directories   directoryReferenceMap    `json:"directories"`
	Files         fileReferenceMap         `json:"files"`
	ExtraMetadata ExtraMetadata            `json:"extraMetadata"`
	BaseMetadata
}

var _ SerializableMetadata = (*DirectoryMetadata)(nil)
var _ SerializableMetadata = (*directoryReferenceMap)(nil)

func NewDirectoryMetadata(details DirectoryMetadataDetails, directories map[string]DirectoryReference, files map[string]FileReference, extraMetadata ExtraMetadata) *DirectoryMetadata {
	dirMetadata := &DirectoryMetadata{
		Details:       details,
		Directories:   directories,
		Files:         files,
		ExtraMetadata: extraMetadata,
	}

	dirMetadata.Type = "directory"
	return dirMetadata
}
func (dm *DirectoryMetadata) EncodeMsgpack(enc *msgpack.Encoder) error {
	err := serialize.InitMarshaller(types.MetadataTypeDirectory, enc)
	if err != nil {
		return err
	}

	items := make([]interface{}, 4)

	items[0] = dm.Details
	items[1] = dm.Directories
	items[2] = dm.Files
	items[3] = dm.ExtraMetadata.Data

	return enc.Encode(items)
}

func (dm *DirectoryMetadata) DecodeMsgpack(dec *msgpack.Decoder) error {
	err := serialize.InitUnmarshaller(types.MetadataTypeDirectory, dec)
	if err != nil {
		return err
	}
	val, err := dec.DecodeArrayLen()

	if err != nil {
		return err
	}

	if val != 4 {
		return errors.New(" Corrupted metadata")
	}

	for i := 0; i < val; i++ {
		switch i {
		case 0:
			err = dec.Decode(&dm.Details)
			if err != nil {
				return err
			}
		case 1:
			err = dec.Decode(&dm.Directories)
			if err != nil {
				return err
			}

		case 2:
			err = dec.Decode(&dm.Files)
			if err != nil {
				return err
			}
		case 3:
			intMap, err := decodeIntMap(dec)
			if err != nil {
				return err
			}
			dm.ExtraMetadata.Data = intMap
		}
	}

	dm.Type = "directory"

	return nil
}
func (drm directoryReferenceMap) EncodeMsgpack(enc *msgpack.Encoder) error {
	// First, encode the length of the map
	if err := enc.EncodeMapLen(len(drm)); err != nil {
		return err
	}

	// Then, encode each key-value pair
	for k, v := range drm {
		if err := enc.EncodeString(k); err != nil {
			return err
		}
		if err := enc.Encode(&v); err != nil { // Assuming DirectoryReference can be encoded by msgpack
			return err
		}
	}

	return nil
}

func (drm *directoryReferenceMap) DecodeMsgpack(dec *msgpack.Decoder) error {
	// Read the map length
	l, err := dec.DecodeMapLen()
	if err != nil {
		return err
	}

	// Initialize the map if it's nil
	if *drm == nil {
		*drm = make(directoryReferenceMap, l)
	}

	// Decode each key-value pair
	for i := 0; i < l; i++ {
		key, err := dec.DecodeString()
		if err != nil {
			return err
		}
		var value DirectoryReference
		if err := dec.Decode(&value); err != nil {
			return err
		}
		(*drm)[key] = value
	}

	return nil
}

func (frm fileReferenceMap) EncodeMsgpack(enc *msgpack.Encoder) error {
	// First, encode the length of the map
	if err := enc.EncodeMapLen(len(frm)); err != nil {
		return err
	}

	// Then, encode each key-value pair
	for k, v := range frm {
		if err := enc.EncodeString(k); err != nil {
			return err
		}
		if err := enc.Encode(&v); err != nil { // Assuming DirectoryReference can be encoded by msgpack
			return err
		}
	}

	return nil
}

func (drm *fileReferenceMap) DecodeMsgpack(dec *msgpack.Decoder) error {
	// Read the map length
	l, err := dec.DecodeMapLen()
	if err != nil {
		return err
	}

	// Initialize the map if it's nil
	if *drm == nil {
		*drm = make(fileReferenceMap, l)
	}

	// Decode each key-value pair
	for i := 0; i < l; i++ {
		key, err := dec.DecodeString()
		if err != nil {
			return err
		}
		var value FileReference
		if err := dec.Decode(&value); err != nil {
			return err
		}
		(*drm)[key] = value
	}

	return nil
}