// Copyright (c) 2021 Proton Technologies AG
//
// This file is part of ProtonMail Bridge.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 fakeapi

import (
	"context"
	"fmt"

	"github.com/ProtonMail/proton-bridge/pkg/pmapi"
)

func (api *FakePMAPI) isLabelFolder(labelID string) bool {
	for _, label := range api.labels {
		if label.ID == labelID {
			return bool(label.Exclusive)
		}
	}
	switch labelID {
	case pmapi.InboxLabel,
		pmapi.TrashLabel,
		pmapi.SpamLabel,
		pmapi.ArchiveLabel,
		pmapi.SentLabel,
		pmapi.DraftLabel:
		return true
	}
	return false
}

func (api *FakePMAPI) ListLabels(context.Context) ([]*pmapi.Label, error) {
	if err := api.checkAndRecordCall(GET, "/labels/1", nil); err != nil {
		return nil, err
	}
	return api.labels, nil
}

func (api *FakePMAPI) CreateLabel(_ context.Context, label *pmapi.Label) (*pmapi.Label, error) {
	if err := api.checkAndRecordCall(POST, "/labels", &pmapi.LabelReq{Label: label}); err != nil {
		return nil, err
	}
	for _, existingLabel := range api.labels {
		if existingLabel.Name == label.Name {
			return nil, fmt.Errorf("folder or label %s already exists", label.Name)
		}
	}
	prefix := "label"
	if label.Exclusive {
		prefix = "folder"
	}
	label.ID = api.controller.labelIDGenerator.next(prefix)
	if label.Path == "" {
		label.Path = label.Name
	}
	api.labels = append(api.labels, label)
	api.addEventLabel(pmapi.EventCreate, label)
	return label, nil
}

func (api *FakePMAPI) UpdateLabel(_ context.Context, label *pmapi.Label) (*pmapi.Label, error) {
	if err := api.checkAndRecordCall(PUT, "/labels", &pmapi.LabelReq{Label: label}); err != nil {
		return nil, err
	}
	for idx, existingLabel := range api.labels {
		if existingLabel.ID == label.ID {
			// Request doesn't have to include all properties and these have to stay the same.
			label.Type = existingLabel.Type
			label.Exclusive = existingLabel.Exclusive
			if label.Path == "" {
				label.Path = label.Name
			}
			api.labels[idx] = label
			api.addEventLabel(pmapi.EventUpdate, label)
			return label, nil
		}
	}
	return nil, fmt.Errorf("label %s does not exist", label.ID)
}

func (api *FakePMAPI) DeleteLabel(_ context.Context, labelID string) error {
	if err := api.checkAndRecordCall(DELETE, "/labels/"+labelID, nil); err != nil {
		return err
	}
	for idx, existingLabel := range api.labels {
		if existingLabel.ID == labelID {
			api.labels = append(api.labels[:idx], api.labels[idx+1:]...)
			api.addEventLabel(pmapi.EventDelete, existingLabel)
			return nil
		}
	}
	return fmt.Errorf("label %s does not exist", labelID)
}