package squashfs

import (
	"errors"
	"io"
	"io/fs"
	"math"
	"time"
)

func (w *Writer) fixFolders() error {
	for folder := range w.structure {
		if folder == "/" || w.Contains(folder) {
			continue
		}
		err := w.AddFolderTo(folder, fs.ModePerm)
		if err != nil {
			return err
		}
	}
	return nil
}

//WriteTo attempts to write the archive to the given io.Writer.
//Folder that aren't present (such as if you add a file at /folder/file, but not the folder /folder)
//are added with full permission (777).
func (w *Writer) WriteTo(write io.Writer) (int64, error) {
	err := w.fixFolders()
	if err != nil {
		return 0, err
	}
	if w.BlockSize > 1048576 {
		w.BlockSize = 1048576
	} else if w.BlockSize < 4096 {
		w.BlockSize = 4096
	}
	w.Flags.RemoveDuplicates = false
	w.Flags.Exportable = false
	w.Flags.NoXattr = true
	w.superblock = superblock{
		Magic:           magic,
		InodeCount:      w.countInodes(),
		CreationTime:    uint32(time.Now().Unix()),
		BlockSize:       w.BlockSize,
		CompressionType: uint16(w.compressionType),
		BlockLog:        uint16(math.Log2(float64(w.BlockSize))),
		Flags:           w.Flags.ToUint(),
		IDCount:         uint16(len(w.uidGUIDTable)),
		MajorVersion:    4,
		MinorVersion:    0,
	}
	_ = super
	return 0, errors.New("I SAID DON'T")
}