// Copyright (c) 2020 Proton Technologies AG
//
// This file is part of ProtonMail Bridge.
//
// ProtonMail Bridge is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// ProtonMail Bridge 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with ProtonMail Bridge.  If not, see <https://www.gnu.org/licenses/>.

package cliie

import (
	"fmt"
	"strconv"
	"strings"

	"github.com/ProtonMail/proton-bridge/internal/frontend/types"
	"github.com/abiosoft/ishell"
)

// completeUsernames is a helper to complete usernames as the user types.
func (f *frontendCLI) completeUsernames(args []string) (usernames []string) {
	if len(args) > 1 {
		return
	}
	arg := ""
	if len(args) == 1 {
		arg = args[0]
	}
	for _, user := range f.ie.GetUsers() {
		if strings.HasPrefix(strings.ToLower(user.Username()), strings.ToLower(arg)) {
			usernames = append(usernames, user.Username())
		}
	}
	return
}

// noAccountWrapper is a decorator for functions which need any account to be properly functional.
func (f *frontendCLI) noAccountWrapper(callback func(*ishell.Context)) func(*ishell.Context) {
	return func(c *ishell.Context) {
		users := f.ie.GetUsers()
		if len(users) == 0 {
			f.Println("No active accounts. Please add account to continue.")
		} else {
			callback(c)
		}
	}
}

func (f *frontendCLI) askUserByIndexOrName(c *ishell.Context) types.User {
	user := f.getUserByIndexOrName("")
	if user != nil {
		return user
	}

	numberOfAccounts := len(f.ie.GetUsers())
	indexRange := fmt.Sprintf("number between 0 and %d", numberOfAccounts-1)
	if len(c.Args) == 0 {
		f.Printf("Please choose %s or username.\n", indexRange)
		return nil
	}
	arg := c.Args[0]
	user = f.getUserByIndexOrName(arg)
	if user == nil {
		f.Printf("Wrong input '%s'. Choose %s or username.\n", bold(arg), indexRange)
		return nil
	}
	return user
}

func (f *frontendCLI) getUserByIndexOrName(arg string) types.User {
	users := f.ie.GetUsers()
	numberOfAccounts := len(users)
	if numberOfAccounts == 0 {
		return nil
	}
	if numberOfAccounts == 1 {
		return users[0]
	}
	if index, err := strconv.Atoi(arg); err == nil {
		if index < 0 || index >= numberOfAccounts {
			return nil
		}
		return users[index]
	}
	for _, user := range users {
		if user.Username() == arg {
			return user
		}
	}
	return nil
}