package process

import (
	"github.com/Safing/portbase/database"
	"github.com/Safing/portbase/database/query"
	"github.com/Safing/portbase/log"
	"github.com/Safing/portmaster/profile"
)

var (
	profileDB = database.NewInterface(nil)
)

// FindProfiles finds and assigns a profile set to the process.
func (p *Process) FindProfiles() error {

	p.Lock()
	defer p.Unlock()

	// only find profiles if not already done.
	if p.profileSet != nil {
		return nil
	}

	// User Profile
	it, err := profileDB.Query(query.New(profile.MakeProfileKey(profile.UserNamespace, "")).Where(query.Where("LinkedPath", query.SameAs, p.Path)))
	if err != nil {
		return err
	}

	var userProfile *profile.Profile
	for r := range it.Next {
		it.Cancel()
		userProfile, err = profile.EnsureProfile(r)
		if err != nil {
			return err
		}
		break
	}
	if it.Err() != nil {
		return it.Err()
	}

	// create new profile if it does not exist.
	if userProfile == nil {
		// create new profile
		userProfile = profile.New()
		userProfile.Name = p.ExecName
		userProfile.LinkedPath = p.Path
	}

	if userProfile.MarkUsed() {
		userProfile.Save(profile.UserNamespace)
	}

	// Stamp
	// Find/Re-evaluate Stamp profile
	// 1. check linked stamp profile
	// 2. if last check is was more than a week ago, fetch from stamp:
	// 3. send path identifier to stamp
	// 4. evaluate all returned profiles
	// 5. select best
	// 6. link stamp profile to user profile
	// FIXME: implement!

	p.UserProfileKey = userProfile.Key()
	p.profileSet = profile.NewSet(userProfile, nil)
	go p.Save()

	return nil
}

func selectProfile(p *Process, profs []*profile.Profile) (selectedProfile *profile.Profile) {
	var highestScore int
	for _, prof := range profs {
		score := matchProfile(p, prof)
		if score > highestScore {
			selectedProfile = prof
		}
	}
	return
}

func matchProfile(p *Process, prof *profile.Profile) (score int) {
	for _, fp := range prof.Fingerprints {
		score += matchFingerprint(p, fp)
	}
	return
}

func matchFingerprint(p *Process, fp *profile.Fingerprint) (score int) {
	if !fp.MatchesOS() {
		return 0
	}

	switch fp.Type {
	case "full_path":
		if p.Path == fp.Value {
		}
		return profile.GetFingerprintWeight(fp.Type)
	case "partial_path":
		// FIXME: if full_path matches, do not match partial paths
		return profile.GetFingerprintWeight(fp.Type)
	case "md5_sum", "sha1_sum", "sha256_sum":
		// FIXME: one sum is enough, check sums in a grouped form, start with the best
		sum, err := p.GetExecHash(fp.Type)
		if err != nil {
			log.Errorf("process: failed to get hash of executable: %s", err)
		} else if sum == fp.Value {
			return profile.GetFingerprintWeight(fp.Type)
		}
	}

	return 0
}