package pslog

import (
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"
	"strconv"
	"strings"
	"sync"
	"sync/atomic"
	"time"

	"gitee.com/xuesongtao/gotool/base"
	plg "gitee.com/xuesongtao/ps-log/log"
)

const (
	saveOffsetDir         = ".pslog" // 保存偏移量的文件目录
	cleanOffsetFileDayDur = 3        // 清理偏移量文件变动多少天之前的文件
)

type FileInfo struct {
	Handler      *Handler // 这里优先 PsLog.handler
	Dir          string   // 文件目录
	Name         string   // 文件名
	offsetChange int32    // 记录 offset 变化次数
	offset       int64    // 当前文件偏移量
	beginOffset  int64    // 记录最开始的偏移量
	mu           sync.Mutex
}

// HandlerIsNil
func (f *FileInfo) HandlerIsNil() bool {
	return f.Handler == nil
}

// FileName 获取文件的全路径名
func (f *FileInfo) FileName() string {
	return filepath.Join(f.Dir, f.Name)
}

// Parse 解析 path
func (f *FileInfo) Parse(path string) {
	f.Dir, f.Name = filepath.Split(path)
}

// IsExpire 文件是否过期
func (f *FileInfo) IsExpire(t ...time.Time) bool {
	defaultTime := time.Now()
	if len(t) > 0 {
		defaultTime = t[0]
	}
	if f.Handler != nil {
		return f.Handler.ExpireAt.Before(defaultTime)
	}
	return false
}

// CleanNameFmt 清理 Name 中的格式, 如: test.log => test
func (f *FileInfo) CleanNameFmt() string {
	if f.Name == "" {
		return ""
	}
	tmpIndex := strings.Index(f.Name, ".")
	return f.Name[:tmpIndex]
}

// PsLogDir 目录
func (f *FileInfo) PsLogDir() string {
	return filepath.Join(f.Dir, saveOffsetDir)
}

// offsetDir 保存偏移量文件的目录
func (f *FileInfo) offsetDir() string {
	return filepath.Join(f.PsLogDir(), "offset")
}

// offsetFilename 获取保存文件偏移量的名称
func (f *FileInfo) offsetFilename() string {
	// 处理为 xxx/.pslog/offset/_xxx.log.txt
	return filepath.Join(f.offsetDir(), "_"+f.Name+".txt")
}

func (f *FileInfo) storeOffset(o int64) {
	atomic.StoreInt64(&f.offset, o)
}

func (f *FileInfo) loadOffset() int64 {
	return atomic.LoadInt64(&f.offset)
}

func (f *FileInfo) loadBeginOffset() int64 {
	return atomic.LoadInt64(&f.beginOffset)
}

// initOffset 初始化文件 offset
func (f *FileInfo) initOffset() {
	// 初次使用需要判断下是否需要清除偏移量
	if f.cleanOffset() {
		return
	}

	// 需要判断下是否已处理过
	if f.offset > 0 {
		return
	}

	// 从文件中读取偏移量
	filename := f.offsetFilename()
	offset, err := f.getContent(filename)
	if err != nil {
		plg.Errorf("f.getContent %q is failed, err: %v", filename, err)
		return
	}
	offsetInt, _ := strconv.Atoi(offset)
	f.offset = int64(offsetInt)
	f.beginOffset = f.offset
}

func (f *FileInfo) cleanOffset() (skip bool) {
	if !f.Handler.CleanOffset {
		return
	}
	f.offset = 0
	f.beginOffset = f.offset
	f.putContent(f.offsetFilename(), "0")
	f.Handler.CleanOffset = false
	skip = true
	return
}

// saveOffset 保存偏移量
// 通过隐藏文件来保存
func (f *FileInfo) saveOffset(mustSaveOffset bool, offset int64) {
	filename := f.offsetFilename()
	// 判断下是否需要持久化
	if mustSaveOffset || f.Handler.Change == -1 {
		if _, err := f.putContent(filename, base.ToString(offset)); err != nil {
			plg.Error("f.putContent is failed, err:", err)
		}
		f.removeOffsetFile()
		return
	}

	f.offsetChange++
	if f.offsetChange > f.Handler.Change {
		if _, err := f.putContent(filename, base.ToString(offset)); err != nil {
			plg.Error("f.putContent is failed, err:", err)
		}
		f.removeOffsetFile()
		f.offsetChange = 0
	}
}

// getContent 查询
func (f *FileInfo) getContent(path string) (string, error) {
	fh, err := filePool.Get(path, os.O_RDWR)
	if err != nil {
		return "", fmt.Errorf("filePool.Get %q is failed, err: %v", path, err)
	}
	defer filePool.Put(fh)

	data, err := fh.GetContent()
	if err != nil {
		return "", fmt.Errorf("getContent %q is failed, err: %v", path, err)
	}
	return data, nil
}

// putContent 覆写
func (f *FileInfo) putContent(path string, content string) (int, error) {
	fh, err := filePool.Get(path, os.O_RDWR|os.O_TRUNC)
	if err != nil {
		return 0, fmt.Errorf("filePool.Get %q is failed, err: %v", path, err)
	}
	defer filePool.Put(fh)
	return fh.PutContent(content)
}

// removeOffsetFile 移除保存文件偏移量的文件
func (f *FileInfo) removeOffsetFile(filename ...string) {
	if len(filename) > 0 && filename[0] != "" {
		if err := os.Remove(filename[0]); err != nil {
			plg.Errorf("os.Remove %q is failed, err: %v", filename[0], err)
		}
		return
	}

	// 移除当前目录下cleanOffsetFileDayDur天前的文件
	offsetFiles, err := ioutil.ReadDir(f.offsetDir())
	if err != nil {
		plg.Errorf("ioutil.ReadDir %q is failed, err: %v", f.offsetDir(), err)
		return
	}
	curTime := time.Now()
	for _, offsetFile := range offsetFiles {
		if curTime.Sub(offsetFile.ModTime())/base.DayDur <= cleanOffsetFileDayDur {
			continue
		}

		delFilename := filepath.Join(f.offsetDir(), offsetFile.Name())
		if err := os.Remove(delFilename); err != nil {
			plg.Warningf("os.Remove %q is failed, err: %v", delFilename, err)
		}
	}
}