package logging

import (
    "fmt"
    "os"
)

// File writes its logs to file.
type File struct {
	fo, fe *os.File
	*LogWriter
}

// NewFile creates a new Logger that writes all logs to the file
// specified by path. If overwrite is specified, the log file will be
// truncated before writing. Otherwise, the log file will be appended
// to.
func NewFile(path string, overwrite bool) (*File, error) {
	fl := new(File)

	var err error

	if overwrite {
		fl.fo, err = os.Create(path)
	} else {
		fl.fo, err = os.OpenFile(path, os.O_WRONLY|os.O_APPEND, 0600) // #nosec G302
	}

	if err != nil {
		return nil, err
	}

	fl.LogWriter = NewLogWriter(fl.fo, fl.fo)
	return fl, nil
}

// NewSplitFile creates a new Logger that writes debug and information
// messages to the output file, and warning and higher messages to the
// error file. If overwrite is specified, the log files will be
// truncated before writing.
func NewSplitFile(outpath, errpath string, overwrite bool) (*File, error) {
	fl := new(File)

	var err error

	if overwrite {
		fl.fo, err = os.Create(outpath)
	} else {
		fl.fo, err = os.OpenFile(outpath, os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0600)
	}

	if err != nil {
		return nil, err
	}

	if overwrite {
		fl.fe, err = os.Create(errpath)
	} else {
		fl.fe, err = os.OpenFile(errpath, os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0600)
	}

 if err != nil {
        if closeErr := fl.Close(); closeErr != nil {
            return nil, fmt.Errorf("failed to open error log: cleanup close failed: %v: %w", closeErr, err)
        }
        return nil, err
    }

	fl.LogWriter = NewLogWriter(fl.fo, fl.fe)
	return fl, nil
}

// Close calls close on the underlying log files.
func (fl *File) Close() error {
	if fl.fo != nil {
		if err := fl.fo.Close(); err != nil {
			return err
		}
		fl.fo = nil
	}

	if fl.fe != nil {
		return fl.fe.Close()
	}

	return nil
}

func (fl *File) Flush() error {
	if err := fl.fo.Sync(); err != nil {
		return err
	}
	return fl.fe.Sync()
}

func (fl *File) Chmod(mode os.FileMode) error {
    if err := fl.fo.Chmod(mode); err != nil {
        return fmt.Errorf("failed to chmod output log: %w", err)
    }

    if err := fl.fe.Chmod(mode); err != nil {
        return fmt.Errorf("failed to chmod error log: %w", err)
    }

    return nil
}