package godog

import (
	"fmt"
	"regexp"

	"github.com/DATA-DOG/godog/gherkin"
)

type suiteFeature struct {
	suite
	// for hook tests
	befScenarioHook *gherkin.Scenario
}

func (s *suiteFeature) BeforeScenario(scenario *gherkin.Scenario) {
	// reset feature paths
	cfg.paths = []string{}
	// reset hook test references
	s.befScenarioHook = nil
	// reset formatter, which collects all details
	s.fmt = &testFormatter{}
}

func (s *suiteFeature) iHaveBeforeScenarioHook(args ...*Arg) error {
	s.suite.BeforeScenario(BeforeScenarioHandlerFunc(func(scenario *gherkin.Scenario) {
		s.befScenarioHook = scenario
	}))
	return nil
}

func (s *suiteFeature) featurePath(args ...*Arg) error {
	cfg.paths = append(cfg.paths, args[0].String())
	return nil
}

func (s *suiteFeature) parseFeatures(args ...*Arg) (err error) {
	s.features, err = cfg.features()
	return
}

func (s *suiteFeature) iShouldHaveNumFeatureFiles(args ...*Arg) error {
	if len(s.features) != args[0].Int() {
		return fmt.Errorf("expected %d features to be parsed, but have %d", args[0].Int(), len(s.features))
	}
	expected := args[1].PyString().Lines
	var actual []string
	for _, ft := range s.features {
		actual = append(actual, ft.Path)
	}
	if len(expected) != len(actual) {
		return fmt.Errorf("expected %d feature paths to be parsed, but have %d", len(expected), len(actual))
	}
	for i := 0; i < len(expected); i++ {
		if expected[i] != actual[i] {
			return fmt.Errorf(`expected feature path "%s" at position: %d, does not match actual "%s"`, expected[i], i, actual[i])
		}
	}
	return nil
}

func (s *suiteFeature) iRunFeatures(args ...*Arg) error {
	for _, f := range s.features {
		s.runFeature(f)
	}
	return nil
}

func (s *suiteFeature) numScenariosRegistered(args ...*Arg) (err error) {
	var num int
	for _, ft := range s.features {
		num += len(ft.Scenarios)
	}
	if num != args[0].Int() {
		err = fmt.Errorf("expected %d scenarios to be registered, but got %d", args[0].Int(), num)
	}
	return
}

func (s *suiteFeature) iShouldHaveScenarioRecordedInHook(args ...*Arg) (err error) {
	if s.befScenarioHook == nil {
		return fmt.Errorf("there was no scenario executed in before hook")
	}
	if s.befScenarioHook.Title != args[0].String() {
		err = fmt.Errorf(`expected "%s" scenario to be run in hook, but got "%s"`, args[0].String(), s.befScenarioHook.Title)
	}
	return
}

func SuiteContext(g Suite) {
	s := &suiteFeature{
		suite: suite{},
	}

	g.BeforeScenario(s)

	g.Step(
		regexp.MustCompile(`^a feature path "([^"]*)"$`),
		StepHandlerFunc(s.featurePath))
	g.Step(
		regexp.MustCompile(`^I parse features$`),
		StepHandlerFunc(s.parseFeatures))
	g.Step(
		regexp.MustCompile(`^I should have ([\d]+) features? files?:$`),
		StepHandlerFunc(s.iShouldHaveNumFeatureFiles))
	g.Step(
		regexp.MustCompile(`^I should have ([\d]+) scenarios? registered$`),
		StepHandlerFunc(s.numScenariosRegistered))
	g.Step(
		regexp.MustCompile(`^I have a before scenario hook$`),
		StepHandlerFunc(s.iHaveBeforeScenarioHook))
	g.Step(
		regexp.MustCompile(`^I run features$`),
		StepHandlerFunc(s.iRunFeatures))
	g.Step(
		regexp.MustCompile(`^I should have a scenario "([^"]*)" recorded in the hook$`),
		StepHandlerFunc(s.iShouldHaveScenarioRecordedInHook))
}