/*
   GoToSocial
   Copyright (C) 2021-2022 GoToSocial Authors admin@gotosocial.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 <http://www.gnu.org/licenses/>.
*/

package visibility

import (
	"context"
	"errors"
	"fmt"

	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
)

// relevantAccounts denotes accounts that are replied to, boosted by, or mentioned in a status.
type relevantAccounts struct {
	// Who wrote the status
	Account *gtsmodel.Account
	// Who is the status replying to
	InReplyToAccount *gtsmodel.Account
	// Which accounts are mentioned (tagged) in the status
	MentionedAccounts []*gtsmodel.Account
	// Who authed the boosted status
	BoostedAccount *gtsmodel.Account
	// If the boosted status replies to another account, who does it reply to?
	BoostedInReplyToAccount *gtsmodel.Account
	// Who is mentioned (tagged) in the boosted status
	BoostedMentionedAccounts []*gtsmodel.Account
}

func (f *filter) relevantAccounts(ctx context.Context, status *gtsmodel.Status, getBoosted bool) (*relevantAccounts, error) {
	relAccts := &relevantAccounts{
		MentionedAccounts:        []*gtsmodel.Account{},
		BoostedMentionedAccounts: []*gtsmodel.Account{},
	}

	/*
		Here's what we need to try and extract from the status:

			// 1. Who wrote the status
		    Account *gtsmodel.Account

		    // 2. Who is the status replying to
		    InReplyToAccount *gtsmodel.Account

		    // 3. Which accounts are mentioned (tagged) in the status
		    MentionedAccounts []*gtsmodel.Account

			if getBoosted:
				// 4. Who wrote the boosted status
				BoostedAccount *gtsmodel.Account

				// 5. If the boosted status replies to another account, who does it reply to?
				BoostedInReplyToAccount *gtsmodel.Account

				// 6. Who is mentioned (tagged) in the boosted status
				BoostedMentionedAccounts []*gtsmodel.Account
	*/

	// 1. Account.
	// Account might be set on the status already
	if status.Account != nil {
		// it was set
		relAccts.Account = status.Account
	} else {
		// it wasn't set, so get it from the db
		account, err := f.db.GetAccountByID(ctx, status.AccountID)
		if err != nil {
			return nil, fmt.Errorf("relevantAccounts: error getting account with id %s: %s", status.AccountID, err)
		}
		// set it on the status in case we need it further along
		status.Account = account
		// set it on relevant accounts
		relAccts.Account = account
	}

	// 2. InReplyToAccount
	// only get this if InReplyToAccountID is set
	if status.InReplyToAccountID != "" {
		// InReplyToAccount might be set on the status already
		if status.InReplyToAccount != nil {
			// it was set
			relAccts.InReplyToAccount = status.InReplyToAccount
		} else {
			// it wasn't set, so get it from the db
			inReplyToAccount, err := f.db.GetAccountByID(ctx, status.InReplyToAccountID)
			if err != nil {
				return nil, fmt.Errorf("relevantAccounts: error getting inReplyToAccount with id %s: %s", status.InReplyToAccountID, err)
			}
			// set it on the status in case we need it further along
			status.InReplyToAccount = inReplyToAccount
			// set it on relevant accounts
			relAccts.InReplyToAccount = inReplyToAccount
		}
	}

	// 3. MentionedAccounts
	// First check if status.Mentions is populated with all mentions that correspond to status.MentionIDs
	for _, mID := range status.MentionIDs {
		if mID == "" {
			continue
		}
		if !idIn(mID, status.Mentions) {
			// mention with ID isn't in status.Mentions
			mention, err := f.db.GetMention(ctx, mID)
			if err != nil {
				return nil, fmt.Errorf("relevantAccounts: error getting mention with id %s: %s", mID, err)
			}
			if mention == nil {
				return nil, fmt.Errorf("relevantAccounts: mention with id %s was nil", mID)
			}
			status.Mentions = append(status.Mentions, mention)
		}
	}
	// now filter mentions to make sure we only have mentions with a corresponding ID
	nm := []*gtsmodel.Mention{}
	for _, m := range status.Mentions {
		if m == nil {
			continue
		}
		if mentionIn(m, status.MentionIDs) {
			nm = append(nm, m)
			relAccts.MentionedAccounts = append(relAccts.MentionedAccounts, m.TargetAccount)
		}
	}
	status.Mentions = nm

	if len(status.Mentions) != len(status.MentionIDs) {
		return nil, errors.New("relevantAccounts: mentions length did not correspond with mentionIDs length")
	}

	// if getBoosted is set, we should check the same properties on the boosted account as well
	if getBoosted {
		// 4, 5, 6. Boosted status items
		// get the boosted status if it's not set on the status already
		if status.BoostOfID != "" && status.BoostOf == nil {
			boostedStatus, err := f.db.GetStatusByID(ctx, status.BoostOfID)
			if err != nil {
				return nil, fmt.Errorf("relevantAccounts: error getting boosted status with id %s: %s", status.BoostOfID, err)
			}
			status.BoostOf = boostedStatus
		}

		if status.BoostOf != nil {
			// return relevant accounts for the boosted status
			boostedRelAccts, err := f.relevantAccounts(ctx, status.BoostOf, false) // false because we don't want to recurse
			if err != nil {
				return nil, fmt.Errorf("relevantAccounts: error getting relevant accounts of boosted status %s: %s", status.BoostOf.ID, err)
			}
			relAccts.BoostedAccount = boostedRelAccts.Account
			relAccts.BoostedInReplyToAccount = boostedRelAccts.InReplyToAccount
			relAccts.BoostedMentionedAccounts = boostedRelAccts.MentionedAccounts
		}
	}

	return relAccts, nil
}

// domainBlockedRelevant checks through all relevant accounts attached to a status
// to make sure none of them are domain blocked by this instance.
func (f *filter) domainBlockedRelevant(ctx context.Context, r *relevantAccounts) (bool, error) {
	domains := []string{}

	if r.Account != nil {
		domains = append(domains, r.Account.Domain)
	}

	if r.InReplyToAccount != nil {
		domains = append(domains, r.InReplyToAccount.Domain)
	}

	for _, a := range r.MentionedAccounts {
		if a != nil {
			domains = append(domains, a.Domain)
		}
	}

	if r.BoostedAccount != nil {
		domains = append(domains, r.BoostedAccount.Domain)
	}

	if r.BoostedInReplyToAccount != nil {
		domains = append(domains, r.BoostedInReplyToAccount.Domain)
	}

	for _, a := range r.BoostedMentionedAccounts {
		if a != nil {
			domains = append(domains, a.Domain)
		}
	}

	return f.db.AreDomainsBlocked(ctx, domains)
}

func idIn(id string, mentions []*gtsmodel.Mention) bool {
	for _, m := range mentions {
		if m == nil {
			continue
		}
		if m.ID == id {
			return true
		}
	}
	return false
}

func mentionIn(mention *gtsmodel.Mention, ids []string) bool {
	if mention == nil {
		return false
	}
	for _, i := range ids {
		if mention.ID == i {
			return true
		}
	}
	return false
}