// 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 util

import (
	"os"
	"sort"
	"strings"
	"sync"
	"time"

	"github.com/88250/gulu"
	"github.com/ConradIrwin/font/sfnt"
	"github.com/flopp/go-findfont"
	"github.com/siyuan-note/logging"
	ttc "golang.org/x/image/font/sfnt"
	textUnicode "golang.org/x/text/encoding/unicode"
	"golang.org/x/text/transform"
)

var (
	sysFonts     []string
	sysFontsLock = sync.Mutex{}
)

func LoadSysFonts() (ret []string) {
	sysFontsLock.Lock()
	defer sysFontsLock.Unlock()

	if 0 < len(sysFonts) {
		return sysFonts
	}

	start := time.Now()
	fonts := loadFonts()
	ret = []string{}
	for _, font := range fonts {
		ret = append(ret, font.Family)
	}
	ret = gulu.Str.RemoveDuplicatedElem(ret)
	ret = removeUnusedFonts(ret)
	sort.Strings(ret)
	sysFonts = ret
	logging.LogInfof("loaded system fonts [%d] in [%dms]", len(sysFonts), time.Since(start).Milliseconds())
	return
}

func removeUnusedFonts(fonts []string) (ret []string) {
	ret = []string{}
	for _, font := range fonts {
		if strings.HasPrefix(font, "Noto Sans") {
			continue
		}
		ret = append(ret, font)
	}
	return
}

type Font struct {
	Path   string
	Family string
}

func loadFonts() (ret []*Font) {
	ret = []*Font{}
	for _, fontPath := range findfont.List() {
		if strings.HasSuffix(strings.ToLower(fontPath), ".ttc") {
			data, err := os.ReadFile(fontPath)
			if err != nil {
				logging.LogErrorf("read font file [%s] failed: %s", fontPath, err)
				continue
			}
			collection, err := ttc.ParseCollection(data)
			if err != nil {
				//LogErrorf("parse font collection [%s] failed: %s", fontPath, err)
				continue
			}

			for i := 0; i < collection.NumFonts(); i++ {
				font, err := collection.Font(i)
				if err != nil {
					//LogErrorf("get font [%s] failed: %s", fontPath, err)
					continue
				}
				if family := parseFontFamily(font); "" != family {
					ret = append(ret, &Font{fontPath, family})
					//LogInfof("[%s] [%s]", fontPath, family)
				}
			}
		} else if strings.HasSuffix(strings.ToLower(fontPath), ".otf") || strings.HasSuffix(strings.ToLower(fontPath), ".ttf") {
			fontFile, err := os.Open(fontPath)
			if err != nil {
				//LogErrorf("open font file [%s] failed: %s", fontPath, err)
				continue
			}
			font, err := sfnt.Parse(fontFile)
			if err != nil {
				//LogErrorf("parse font [%s] failed: %s", fontPath, err)
				continue
			}

			t, err := font.NameTable()
			if err != nil {
				//LogErrorf("parse font name table [%s] failed: %s", fontPath, err)
				return
			}
			fontFile.Close()
			var family string
			for _, e := range t.List() {
				if sfnt.NameFontFamily != e.NameID && sfnt.NamePreferredFamily != e.NameID {
					continue
				}

				v, _, err := transform.Bytes(textUnicode.UTF16(textUnicode.BigEndian, textUnicode.IgnoreBOM).NewDecoder(), e.Value)
				if err != nil {
					//LogErrorf("decode font family [%s] failed: %s", fontPath, err)
					continue
				}
				val := string(v)
				if sfnt.NameFontFamily == e.NameID && "" != val {
					family = val
				}
				if sfnt.NamePreferredFamily == e.NameID && "" != val {
					family = val
				}
			}
			if "" != family && !strings.HasPrefix(family, ".") {
				ret = append(ret, &Font{fontPath, family})
				//logging.LogInfof("[%s] [%s]", fontPath, family)
			}
		}
	}
	return
}

func parseFontFamily(font *ttc.Font) string {
	family, _ := font.Name(nil, ttc.NameIDTypographicFamily)
	if "" == family {
		family, _ = font.Name(nil, ttc.NameIDFamily)
	}
	if strings.HasPrefix(family, ".") {
		return ""
	}
	return family
}
