package checks

import (
	"context"
	"fmt"
	"time"

	"github.com/prometheus/common/model"

	"github.com/cloudflare/pint/internal/diags"
	"github.com/cloudflare/pint/internal/discovery"
	"github.com/cloudflare/pint/internal/output"
)

const (
	RuleForCheckName = "rule/for"
)

type RuleForKey string

const (
	RuleForFor           RuleForKey = "for"
	RuleForKeepFiringFor RuleForKey = "keep_firing_for"
)

func NewRuleForCheck(key RuleForKey, minFor, maxFor time.Duration, comment string, severity Severity) RuleForCheck {
	return RuleForCheck{
		key:      key,
		minFor:   minFor,
		maxFor:   maxFor,
		comment:  comment,
		severity: severity,
		instance: fmt.Sprintf("%s(%s:%s)", RuleForCheckName, output.HumanizeDuration(minFor), output.HumanizeDuration(maxFor)),
	}
}

type RuleForCheck struct {
	key      RuleForKey
	comment  string
	instance string
	minFor   time.Duration
	maxFor   time.Duration
	severity Severity
}

func (c RuleForCheck) Meta() CheckMeta {
	return CheckMeta{
		States: []discovery.ChangeType{
			discovery.Noop,
			discovery.Added,
			discovery.Modified,
			discovery.Moved,
		},
		Online:        false,
		AlwaysEnabled: false,
	}
}

func (c RuleForCheck) String() string {
	return c.instance
}

func (c RuleForCheck) Reporter() string {
	return RuleForCheckName
}

func (c RuleForCheck) Check(_ context.Context, entry discovery.Entry, _ []discovery.Entry) (problems []Problem) {
	if entry.Rule.AlertingRule == nil {
		return nil
	}

	var forDur model.Duration
	var lines diags.LineRange
	var diag diags.Diagnostic

	switch {
	case c.key == RuleForFor && entry.Rule.AlertingRule.For != nil:
		forDur, _ = model.ParseDuration(entry.Rule.AlertingRule.For.Value)
		lines = entry.Rule.AlertingRule.For.Pos.Lines()
		diag = diags.Diagnostic{
			Message:     "",
			Pos:         entry.Rule.AlertingRule.For.Pos,
			FirstColumn: 1,
			LastColumn:  len(entry.Rule.AlertingRule.For.Value),
			Kind:        diags.Issue,
		}
	case c.key == RuleForKeepFiringFor && entry.Rule.AlertingRule.KeepFiringFor != nil:
		forDur, _ = model.ParseDuration(entry.Rule.AlertingRule.KeepFiringFor.Value)
		lines = entry.Rule.AlertingRule.KeepFiringFor.Pos.Lines()
		diag = diags.Diagnostic{
			Message:     "",
			Pos:         entry.Rule.AlertingRule.KeepFiringFor.Pos,
			FirstColumn: 1,
			LastColumn:  len(entry.Rule.AlertingRule.KeepFiringFor.Value),
			Kind:        diags.Issue,
		}
	default:
		lines = entry.Rule.AlertingRule.Alert.Pos.Lines()
		diag = diags.Diagnostic{
			Message:     "",
			Pos:         entry.Rule.AlertingRule.Alert.Pos,
			FirstColumn: 1,
			LastColumn:  len(entry.Rule.AlertingRule.Alert.Value),
			Kind:        diags.Issue,
		}
	}

	if time.Duration(forDur) < c.minFor {
		problems = append(problems, Problem{
			Anchor:   AnchorAfter,
			Lines:    lines,
			Reporter: c.Reporter(),
			Summary:  "duration required",
			Details:  maybeComment(c.comment),
			Severity: c.severity,
			Diagnostics: []diags.Diagnostic{
				{
					Message:     fmt.Sprintf("This alert rule must have a `%s` field with a minimum duration of %s.", c.key, output.HumanizeDuration(c.minFor)),
					Pos:         diag.Pos,
					FirstColumn: diag.FirstColumn,
					LastColumn:  diag.LastColumn,
					Kind:        diag.Kind,
				},
			},
		})
	}

	if c.maxFor > 0 && time.Duration(forDur) > c.maxFor {
		problems = append(problems, Problem{
			Anchor:   AnchorAfter,
			Lines:    lines,
			Reporter: c.Reporter(),
			Summary:  "duration too long",
			Details:  maybeComment(c.comment),
			Severity: c.severity,
			Diagnostics: []diags.Diagnostic{
				{
					Message:     fmt.Sprintf("This alert rule must have a `%s` field with a maximum duration of %s.", c.key, output.HumanizeDuration(c.maxFor)),
					Pos:         diag.Pos,
					FirstColumn: diag.FirstColumn,
					LastColumn:  diag.LastColumn,
					Kind:        diag.Kind,
				},
			},
		})
	}

	return problems
}
