package files

import (
	"compress/gzip"
	"io"
	"os"
	"path/filepath"

	"go.alanpearce.eu/homestead/internal/buffer"
	"go.alanpearce.eu/homestead/internal/content"
	"go.alanpearce.eu/homestead/internal/multifile"
	"go.alanpearce.eu/homestead/internal/storage"
	"go.alanpearce.eu/x/log"

	"github.com/andybalholm/brotli"
	"github.com/klauspost/compress/zstd"
	"gitlab.com/tozd/go/errors"
)

const (
	gzipLevel   = 6
	brotliLevel = 9
)

type Files struct {
	outputDirectory string
	options         *Options
	log             *log.Logger
}

type Options struct {
	Compress bool
}

func NewWriter(outputDirectory string, logger *log.Logger, opts *Options) (*Files, error) {
	return &Files{
		outputDirectory: outputDirectory,
		options:         opts,
		log:             logger,
	}, nil
}

func (f *Files) OpenRead(filename string) (io.ReadCloser, error) {
	return os.Open(f.join(filename))
}

func (f *Files) OpenWrite(filename string) (io.WriteCloser, error) {
	return openFileWrite(f.join(filename))
}

func (f *Files) WritePost(post *content.Post, content *buffer.Buffer) error {
	fd, err := f.write(post.URL, content)
	if err != nil {
		return err
	}
	err = fd.Close()
	if err != nil {
		return err
	}

	if mf, isMultifile := fd.(*multifile.MultiFile); isMultifile {
		err = mf.Chtimes(post.Date)
	} else {
		err = os.Chtimes(fd.Name(), post.Date, post.Date)
	}
	if err != nil {
		return errors.WithMessage(err, "could not set file times")
	}

	return nil
}

func (f *Files) Write(pathname string, content *buffer.Buffer) error {
	fd, err := f.write(pathname, content)
	if err != nil {
		return err
	}
	fd.Close()

	return nil
}

func (f *Files) WriteFile(file *storage.File, content *buffer.Buffer) error {
	return f.Write(file.Path, content)
}

func (f *Files) write(pathname string, content *buffer.Buffer) (multifile.FileLike, error) {
	filename := pathNameToFileName(pathname)
	err := f.Mkdirp(filepath.Dir(filename))
	if err != nil {
		return nil, errors.WithMessage(err, "could not create directory")
	}

	fd, err := f.OpenFileAndVariants(filename)
	if err != nil {
		return nil, errors.WithMessagef(err, "could not open output file")
	}

	_, err = fd.Write(content.Bytes())
	if err != nil {
		return nil, errors.WithMessage(err, "could not write output file")
	}

	return fd, nil
}

func openFileWrite(filename string) (*os.File, error) {
	return os.OpenFile(filename, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
}

func openFileGz(filename string) (*multifile.CompressWriter, error) {
	filenameGz := filename + ".gz"
	f, err := openFileWrite(filenameGz)
	if err != nil {
		return nil, err
	}
	w, err := gzip.NewWriterLevel(f, gzipLevel)
	if err != nil {
		return nil, err
	}

	return multifile.NewCompressWriter(f, w), err
}

func openFileBrotli(filename string) (*multifile.CompressWriter, error) {
	filenameBrotli := filename + ".br"
	f, err := openFileWrite(filenameBrotli)
	if err != nil {
		return nil, err
	}

	return multifile.NewCompressWriter(f, brotli.NewWriterLevel(f, brotliLevel)), nil
}

func openFileZstd(filename string) (*multifile.CompressWriter, error) {
	f, err := openFileWrite(filename + ".zstd")
	if err != nil {
		return nil, err
	}

	w, err := zstd.NewWriter(f)
	if err != nil {
		return nil, err
	}

	return multifile.NewCompressWriter(f, w), nil
}

func multiOpenFile(filename string) (*multifile.MultiFile, error) {
	r, err := openFileWrite(filename)
	if err != nil {
		return nil, err
	}
	gz, err := openFileGz(filename)
	if err != nil {
		return nil, err
	}
	br, err := openFileBrotli(filename)
	if err != nil {
		return nil, err
	}
	zst, err := openFileZstd(filename)
	if err != nil {
		return nil, err
	}

	return multifile.NewMultiFile(r, gz, br, zst), nil
}

func (f *Files) OpenFileAndVariants(filename string) (multifile.FileLike, error) {
	if f.options.Compress {
		return multiOpenFile(f.join(filename))
	}

	return openFileWrite(f.join(filename))
}

func (f *Files) Mkdirp(dir string) error {
	err := os.MkdirAll(f.join(dir), 0755)
	if err != nil {
		return errors.WithMessage(err, "could not create directory")
	}

	return nil
}

func (f *Files) join(filename string) string {
	return filepath.Join(f.outputDirectory, pathNameToFileName(filename))
}