package postgres

import (
	"context"
	"fmt"

	"cosmossdk.io/schema"
)

// ModuleIndexer manages the tables for a module.
type ModuleIndexer struct {
	moduleName   string
	schema       schema.ModuleSchema
	tables       map[string]*ObjectIndexer
	definedEnums map[string]schema.EnumType
	options      Options
}

// NewModuleIndexer creates a new ModuleIndexer for the given module schema.
func NewModuleIndexer(moduleName string, modSchema schema.ModuleSchema, options Options) *ModuleIndexer {
	return &ModuleIndexer{
		moduleName:   moduleName,
		schema:       modSchema,
		tables:       map[string]*ObjectIndexer{},
		definedEnums: map[string]schema.EnumType{},
		options:      options,
	}
}

// InitializeSchema creates tables for all object types in the module schema and creates enum types.
func (m *ModuleIndexer) InitializeSchema(ctx context.Context, conn DBConn) error {
	// create enum types
	var err error
	m.schema.EnumTypes(func(enumType schema.EnumType) bool {
		err = m.CreateEnumType(ctx, conn, enumType)
		return err == nil
	})
	if err != nil {
		return err
	}

	// create tables for all object types
	m.schema.ObjectTypes(func(typ schema.ObjectType) bool {
		tm := NewObjectIndexer(m.moduleName, typ, m.options)
		m.tables[typ.Name] = tm
		err = tm.CreateTable(ctx, conn)
		if err != nil {
			err = fmt.Errorf("failed to create table for %s in module %s: %v", typ.Name, m.moduleName, err) //nolint:errorlint // using %v for go 1.12 compat
		}
		return err == nil
	})

	return err
}

// ObjectIndexers returns the object indexers for the module.
func (m *ModuleIndexer) ObjectIndexers() map[string]*ObjectIndexer {
	return m.tables
}