package serverv2_test

import (
	"context"
	"os"
	"path/filepath"
	"testing"
	"time"

	gogoproto "github.com/cosmos/gogoproto/proto"
	"github.com/spf13/viper"
	"github.com/stretchr/testify/require"

	appmodulev2 "cosmossdk.io/core/appmodule/v2"
	"cosmossdk.io/core/transaction"
	"cosmossdk.io/log"
	serverv2 "cosmossdk.io/server/v2"
	"cosmossdk.io/server/v2/api/grpc"
	"cosmossdk.io/server/v2/store"
	storev2 "cosmossdk.io/store/v2"
)

type mockStore struct {
	storev2.RootStore
}

func (*mockStore) Close() error {
	return nil
}

type mockInterfaceRegistry struct{}

func (*mockInterfaceRegistry) Resolve(typeUrl string) (gogoproto.Message, error) {
	panic("not implemented")
}

func (*mockInterfaceRegistry) ListImplementations(ifaceTypeURL string) []string {
	panic("not implemented")
}
func (*mockInterfaceRegistry) ListAllInterfaces() []string { panic("not implemented") }

func TestServer(t *testing.T) {
	currentDir, err := os.Getwd()
	require.NoError(t, err)
	configPath := filepath.Join(currentDir, "testdata")

	v, err := serverv2.ReadConfig(configPath)
	if err != nil {
		v = viper.New()
	}
	cfg := v.AllSettings()

	logger := log.NewLogger(os.Stdout)
	ctx := serverv2.SetServerContext(context.Background(), v, logger)

	grpcServer, err := grpc.New[transaction.Tx](logger, &mockInterfaceRegistry{}, map[string]appmodulev2.Handler{}, nil, cfg)
	require.NoError(t, err)

	storeServer, err := store.New[transaction.Tx](&mockStore{}, cfg)
	require.NoError(t, err)

	mockServer := &mockServer{name: "mock-server-1", ch: make(chan string, 100)}

	server := serverv2.NewServer[transaction.Tx](
		serverv2.DefaultServerConfig(),
		grpcServer,
		storeServer,
		mockServer,
	)

	serverCfgs := server.Configs()
	require.Equal(t, serverCfgs[grpcServer.Name()].(*grpc.Config).Address, grpc.DefaultConfig().Address)
	require.Equal(t, serverCfgs[mockServer.Name()].(*mockServerConfig).MockFieldOne, MockServerDefaultConfig().MockFieldOne)

	// write config
	err = server.WriteConfig(configPath)
	require.NoError(t, err)

	v, err = serverv2.ReadConfig(configPath)
	require.NoError(t, err)

	require.Equal(t, v.GetString(grpcServer.Name()+".address"), grpc.DefaultConfig().Address)

	// start empty
	ctx, cancelFn := context.WithCancel(ctx)
	go func() {
		// wait 5sec and cancel context
		<-time.After(5 * time.Second)
		cancelFn()

		err = server.Stop(ctx)
		require.NoError(t, err)
	}()

	err = server.Start(ctx)
	require.NoError(t, err)
}