// SPDX-FileCopyrightText: Copyright 2015-2025 go-swagger maintainers
// SPDX-License-Identifier: Apache-2.0

package antest

import (
	"encoding/json"
	"flag"
	"os"
	"path/filepath"
	"sync"
	"testing"

	"github.com/go-openapi/spec"
	"github.com/go-openapi/swag/loading"
	"github.com/go-openapi/testify/v2/require"
)

var (
	oncePathLoader  sync.Once //nolint:gochecknoglobals // it's okay to use sync.Once as a global
	enableLongTests bool      //nolint:gochecknoglobals // it's okay to store test flags as a global
)

func init() { //nolint:gochecknoinits // it's okay to call init() to register go test extra flags
	if flag.Lookup("enable-long") == nil {
		flag.BoolVar(&enableLongTests, "enable-long", false, "enable long runnning tests")
	}
}

// LongTestsEnabled returns the CLI flag.
func LongTestsEnabled() bool {
	return enableLongTests
}

func initPathLoader() {
	spec.PathLoader = func(path string) (json.RawMessage, error) {
		ext := filepath.Ext(path)
		if ext == ".yml" || ext == ".yaml" {
			return loading.YAMLDoc(path)
		}

		data, err := loading.LoadFromFileOrHTTP(path)
		if err != nil {
			return nil, err
		}

		return json.RawMessage(data), nil
	}
}

// LoadSpec loads a [json] or a yaml spec.
func LoadSpec(path string) (*spec.Swagger, error) {
	oncePathLoader.Do(initPathLoader)

	data, err := spec.PathLoader(path)
	if err != nil {
		return nil, err
	}

	var sw spec.Swagger
	if err := json.Unmarshal(data, &sw); err != nil {
		return nil, err
	}

	return &sw, nil
}

// LoadOrFail fetches a spec from a relative path or dies if the spec cannot be loaded properly.
func LoadOrFail(t testing.TB, relative string) *spec.Swagger {
	cwd, _ := os.Getwd()
	sp, err := LoadSpec(filepath.Join(cwd, relative))
	require.NoError(t, err)

	return sp
}

// AsJSON unmarshals anything as JSON or dies.
func AsJSON(t testing.TB, in any) string {
	bbb, err := json.MarshalIndent(in, "", " ")
	require.NoError(t, err)

	return string(bbb)
}