package iso9660

import (
	"fmt"
	"io"
)

/* The following types of core SUSP records are being handled in some way:
 * - [x] CE (SUSP 5.1: continuation)
 * - [ ] PD (SUSP 5.2: padding)
 * - [x] SP (SUSP 5.3: offset)
 * - [ ] ST (SUSP 5.4)
 * - [x] ER (SUSP 5.5: extension record)
 * - [ ] ES (SUSP 5.6)
 */

// SUSP-112 4.1
type SystemUseEntry []byte

func (e SystemUseEntry) Length() int {
	return int(e[2])
}

func (e SystemUseEntry) Data() []byte {
	return e[4:]
}

func (e SystemUseEntry) Type() string {
	return string(e[:2])
}

type ExtensionRecord struct {
	Version    int
	Identifier string
	Descriptor string
	Source     string
}

// See SUSP-112 5.5
func ExtensionRecordDecode(e SystemUseEntry) (*ExtensionRecord, error) {
	if e.Type() != "ER" {
		return nil, fmt.Errorf("wrong type of record, expected ER")
	}
	if e.Length() < 8 {
		return nil, io.ErrUnexpectedEOF
	}

	identifierLen := int(e[4])
	if e.Length() < 8+identifierLen {
		return nil, io.ErrUnexpectedEOF
	}

	descriptorLen := int(e[5])
	if e.Length() < 8+identifierLen+descriptorLen {
		return nil, io.ErrUnexpectedEOF
	}

	sourceLen := int(e[6])
	if e.Length() < 8+identifierLen+descriptorLen+sourceLen {
		return nil, io.ErrUnexpectedEOF
	}

	return &ExtensionRecord{
		Version:    int(e[7]),
		Identifier: string(e[8 : 8+identifierLen]),
		Descriptor: string(e[8+identifierLen : 8+identifierLen+descriptorLen]),
		Source:     string(e[8+identifierLen+descriptorLen : 8+identifierLen+descriptorLen+sourceLen]),
	}, nil
}

// See SUSP-112 5.3
func SPRecordDecode(e SystemUseEntry) (*SPRecord, error) {
	if e.Type() != "SP" {
		return nil, fmt.Errorf("wrong type of record, expected SP")
	}
	if e.Length() < 7 {
		return nil, io.ErrUnexpectedEOF
	}

	if beByte := e[4]; beByte != 0xBE {
		return nil, fmt.Errorf("invalid control byte, %x != 0xBE", beByte)
	}
	if efByte := e[5]; efByte != 0xEF {
		return nil, fmt.Errorf("invalid control byte, %x != 0xEF", efByte)
	}

	return &SPRecord{
		BytesSkipped: e[6],
	}, nil
}

type SPRecord struct {
	BytesSkipped uint8
}

type SystemUseEntrySlice []SystemUseEntry

func (s SystemUseEntrySlice) GetExtensionRecords() ([]*ExtensionRecord, error) {
	results := make([]*ExtensionRecord, 0)
	for _, entry := range s {
		if entry.Type() == "ER" {
			er, err := ExtensionRecordDecode(entry)
			if err != nil {
				return nil, err
			}
			results = append(results, er)
		}
	}

	return results, nil
}

// SUSP-112 5.1
type ContinuationEntry struct {
	blockLocation uint32
	offset        uint32
	lengthOfArea  uint32
}

func umarshalContinuationEntry(e SystemUseEntry) (*ContinuationEntry, error) {
	if e.Length() != 28 {
		return nil, fmt.Errorf("invalid ContinuationArea record with length %d instead of 28", e.Length())
	}

	location, err := UnmarshalUint32LSBMSB(e.Data()[0:8])
	if err != nil {
		return nil, fmt.Errorf("block location: %w", err)
	}
	offset, err := UnmarshalUint32LSBMSB(e.Data()[8:16])
	if err != nil {
		return nil, fmt.Errorf("offset: %w", err)
	}
	length, err := UnmarshalUint32LSBMSB(e.Data()[16:24])
	if err != nil {
		return nil, fmt.Errorf("length: %w", err)
	}

	return &ContinuationEntry{
		blockLocation: location,
		offset:        offset,
		lengthOfArea:  length,
	}, nil
}

const (
	SUEType_ContinuationArea          = "CE"
	SUEType_PaddingField              = "PD"
	SUEType_SharingProtocolIndicator  = "SP"
	SUEType_SharingProtocolTerminator = "ST"
	SUEType_ExtensionsReference       = "ER"
	SUEType_ExtensionSelector         = "ES"
)

func splitSystemUseEntries(data []byte, ra io.ReaderAt) ([]SystemUseEntry, error) {
	output := make([]SystemUseEntry, 0)

	for len(data) > 0 {
		if len(data) < 4 {
			// SUSP-112 4
			// If the remaining allocated space /.../ is less than four bytes long /.../ shall be ignored.
			break
		}

		entryLen := int(data[2])
		if len(data) < entryLen {
			return nil, fmt.Errorf("splitting System Use entries: %w, expected %d bytes but have only %d", io.ErrUnexpectedEOF, entryLen, len(data))
		}

		entry := SystemUseEntry(data[:entryLen])

		if entry.Type() == SUEType_ContinuationArea {
			ce, err := umarshalContinuationEntry(entry)
			if err != nil {
				return output, fmt.Errorf("unmarshaling ContinuationEntry: %w", err)
			}
			continuation := make([]byte, ce.lengthOfArea)
			finalOffset := (ce.blockLocation * sectorSize) + ce.offset
			if _, err := ra.ReadAt(continuation, int64(finalOffset)); err != nil {
				return output, fmt.Errorf("reading Continuation Area: %w", err)
			}

			continuedEntries, err := splitSystemUseEntries(continuation, ra)
			if err != nil {
				return output, fmt.Errorf("splitting Continuation Area: %w", err)
			}
			output = append(output, continuedEntries...)
		} else {
			output = append(output, entry)
		}

		data = data[entryLen:]
	}

	return output, nil
}

type SUSPMetadata struct {
	Offset       uint8
	HasRockRidge bool
}

func (sm *SUSPMetadata) Clone() *SUSPMetadata {
	if sm == nil {
		return nil
	}

	return &SUSPMetadata{
		Offset:       sm.Offset,
		HasRockRidge: sm.HasRockRidge,
	}
}