// SiYuan - Refactor your thinking
// Copyright (c) 2020-present, b3log.org
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.

package model

import (
	"bytes"
	"os"
	"path/filepath"
	"strings"

	"github.com/88250/gulu"
	"github.com/siyuan-note/logging"
	"github.com/siyuan-note/siyuan/kernel/util"
)

func MoveLocalShorthands(boxID, hPath, parentID, id string) (retID string, err error) {
	shorthandsDir := filepath.Join(util.ShortcutsPath, "shorthands")
	if !gulu.File.IsDir(shorthandsDir) {
		return
	}

	entries, err := os.ReadDir(shorthandsDir)
	if nil != err {
		logging.LogErrorf("read dir [%s] failed: %s", shorthandsDir, err)
		return
	}

	buff := bytes.Buffer{}
	var toRemoves []string
	assetsDir := filepath.Join(util.DataDir, "assets")
	for _, entry := range entries {
		if entry.IsDir() {
			if "assets" == entry.Name() {
				assetsEntries, readErr := os.ReadDir(filepath.Join(shorthandsDir, entry.Name()))
				if nil != readErr {
					logging.LogErrorf("read dir [%s] failed: %s", shorthandsDir, readErr)
					continue
				}
				for _, assetEntry := range assetsEntries {
					if assetEntry.IsDir() {
						continue
					}

					p := filepath.Join(shorthandsDir, entry.Name(), assetEntry.Name())
					assetWritePath := filepath.Join(assetsDir, assetEntry.Name())
					if renameErr := os.Rename(p, assetWritePath); nil != renameErr {
						logging.LogErrorf("rename file [%s] to [%s] failed: %s", p, assetWritePath, renameErr)
						continue
					}
				}
			}
		}

		if filepath.Ext(entry.Name()) != ".md" {
			continue
		}

		p := filepath.Join(shorthandsDir, entry.Name())
		data, readErr := os.ReadFile(p)
		if nil != readErr {
			logging.LogErrorf("read file [%s] failed: %s", p, readErr)
			continue
		}

		buff.Write(bytes.TrimSpace(data))
		buff.WriteString("\n\n")
		toRemoves = append(toRemoves, p)
	}

	clearShorthand := func(toRemoves []string) {
		for _, p := range toRemoves {
			if removeErr := os.Remove(p); nil != removeErr {
				logging.LogErrorf("remove file [%s] failed: %s", p, removeErr)
			}
		}
	}

	content := strings.TrimSpace(buff.String())
	if 1 > len(content) {
		clearShorthand(toRemoves)
		return
	}

	retID, err = CreateWithMarkdown("", boxID, hPath, content, parentID, id, false, "")
	if nil != err {
		logging.LogErrorf("create doc failed: %s", err)
		return
	}

	clearShorthand(toRemoves)
	return
}

func WatchLocalShorthands() {
	shorthandsDir := filepath.Join(util.ShortcutsPath, "shorthands")
	if !gulu.File.IsDir(shorthandsDir) {
		return
	}

	entries, err := os.ReadDir(shorthandsDir)
	if nil != err {
		logging.LogErrorf("read dir [%s] failed: %s", shorthandsDir, err)
		return
	}

	shorthandCount := 0
	for _, entry := range entries {
		if entry.IsDir() {
			continue
		}

		if filepath.Ext(entry.Name()) != ".md" {
			continue
		}

		shorthandCount++
	}

	if 1 > shorthandCount {
		return
	}

	util.PushLocalShorthandCount(shorthandCount)
}