package firefox

import (
	"os"

	"github.com/tidwall/gjson"

	"github.com/moond4rk/hackbrowserdata/types"
)

func extractExtensions(path string) ([]types.ExtensionEntry, error) {
	data, err := os.ReadFile(path)
	if err != nil {
		return nil, err
	}

	var extensions []types.ExtensionEntry
	for _, v := range gjson.GetBytes(data, "addons").Array() {
		// Only include user-installed extensions
		// https://searchfox.org/mozilla-central/source/toolkit/mozapps/extensions/internal/XPIDatabase.jsm#157
		if v.Get("location").String() != "app-profile" {
			continue
		}

		extensions = append(extensions, types.ExtensionEntry{
			Name:        v.Get("defaultLocale.name").String(),
			ID:          v.Get("id").String(),
			Description: v.Get("defaultLocale.description").String(),
			Version:     v.Get("version").String(),
			HomepageURL: v.Get("defaultLocale.homepageURL").String(),
			Enabled:     v.Get("active").Bool(),
		})
	}

	return extensions, nil
}

func countExtensions(path string) (int, error) {
	data, err := os.ReadFile(path)
	if err != nil {
		return 0, err
	}

	var count int
	for _, v := range gjson.GetBytes(data, "addons").Array() {
		if v.Get("location").String() == "app-profile" {
			count++
		}
	}
	return count, nil
}