// Copyright 2023 Northern.tech AS
//
//	Licensed under the Apache License, Version 2.0 (the "License");
//	you may not use this file except in compliance with the License.
//	You may obtain a copy of the License at
//
//	    http://www.apache.org/licenses/LICENSE-2.0
//
//	Unless required by applicable law or agreed to in writing, software
//	distributed under the License is distributed on an "AS IS" BASIS,
//	WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//	See the License for the specific language governing permissions and
//	limitations under the License.
package model

import (
	"github.com/pkg/errors"

	"github.com/mendersoftware/mender-server/pkg/mongo/v2/oid"
)

const (
	LimitMaxDevicesCount       = "max_devices"
	LimitUnlimited       int64 = -1
)

var (
	ValidLimits = []string{LimitMaxDevicesCount}
)

type Limit struct {
	Id       oid.ObjectID `json:"-" bson:"_id,omitempty"`
	Name     string       `json:"name" bson:"name"`
	Value    int64        `json:"value" bson:"value"`
	TenantID string       `json:"-" bson:"tenant_id"`
}

func (r *Limit) Validate() error {
	if !IsValidLimit(r.Name) {
		return errors.Errorf("unsupported limit %v", r.Name)
	}

	isNormalOrUnlimited := r.Value >= 0 || r.IsUnlimited()
	if !isNormalOrUnlimited {
		return errors.Errorf("unsupported limit value %d", r.Value)
	}

	return nil
}

func (l Limit) IsUnlimited() bool {
	return l.Value == LimitUnlimited
}

func (l Limit) IsLess(what int64) bool {
	return what < l.Value
}

func IsValidLimit(name string) bool {
	for _, n := range ValidLimits {
		if name == n {
			return true
		}
	}
	return false
}
