// Copyright (C) 2022-2023  Luke Shumaker <lukeshu@lukeshu.com>
//
// SPDX-License-Identifier: GPL-2.0-or-later

package btrfsitem

import (
	"fmt"

	"git.lukeshu.com/btrfs-progs-ng/lib/binstruct"
	"git.lukeshu.com/btrfs-progs-ng/lib/binstruct/binutil"
	"git.lukeshu.com/btrfs-progs-ng/lib/containers"
)

// key.objectid = inode number of the file
// key.offset = inode number of the parent directory
//
// Might have multiple entries if the same file has multiple hardlinks
// in the same directory.
type InodeRefs struct { // complex INODE_REF=12
	Refs []InodeRef
}

var inodeRefPool containers.SlicePool[InodeRef]

func (o *InodeRefs) Free() {
	for i := range o.Refs {
		bytePool.Put(o.Refs[i].Name)
		o.Refs[i] = InodeRef{}
	}
	inodeRefPool.Put(o.Refs)
	*o = InodeRefs{}
	inodeRefsPool.Put(o)
}

func (o InodeRefs) Clone() InodeRefs {
	var ret InodeRefs
	ret.Refs = inodeRefPool.Get(len(o.Refs))
	copy(ret.Refs, o.Refs)
	for i := range ret.Refs {
		ret.Refs[i].Name = cloneBytes(o.Refs[i].Name)
	}
	return ret
}

func (o *InodeRefs) UnmarshalBinary(dat []byte) (int, error) {
	o.Refs = nil
	if len(dat) > 0 {
		o.Refs = inodeRefPool.Get(1)[:0]
	}
	n := 0
	for n < len(dat) {
		var ref InodeRef
		_n, err := binstruct.Unmarshal(dat[n:], &ref)
		n += _n
		if err != nil {
			return n, err
		}
		o.Refs = append(o.Refs, ref)
	}
	return n, nil
}

func (o InodeRefs) MarshalBinary() ([]byte, error) {
	var dat []byte
	for _, ref := range o.Refs {
		_dat, err := binstruct.Marshal(ref)
		dat = append(dat, _dat...)
		if err != nil {
			return dat, err
		}
	}
	return dat, nil
}

type InodeRef struct {
	Index         int64  `bin:"off=0x0, siz=0x8"`
	NameLen       uint16 `bin:"off=0x8, siz=0x2"` // [ignored-when-writing]
	binstruct.End `bin:"off=0xa"`
	Name          []byte `bin:"-"`
}

func (o *InodeRef) UnmarshalBinary(dat []byte) (int, error) {
	if err := binutil.NeedNBytes(dat, 0xA); err != nil {
		return 0, err
	}
	n, err := binstruct.UnmarshalWithoutInterface(dat, o)
	if err != nil {
		return n, err
	}
	if o.NameLen > MaxNameLen {
		return 0, fmt.Errorf("maximum name len is %v, but .NameLen=%v",
			MaxNameLen, o.NameLen)
	}
	if err := binutil.NeedNBytes(dat, 0xA+int(o.NameLen)); err != nil {
		return 0, err
	}
	dat = dat[n:]
	o.Name = cloneBytes(dat[:o.NameLen])
	n += int(o.NameLen)
	return n, nil
}

func (o InodeRef) MarshalBinary() ([]byte, error) {
	o.NameLen = uint16(len(o.Name))
	dat, err := binstruct.MarshalWithoutInterface(o)
	if err != nil {
		return dat, err
	}
	dat = append(dat, o.Name...)
	return dat, nil
}