import.go

package ast

import (
	"fmt"

	sitter "github.com/smacker/go-tree-sitter"
)

// ImportNode represents an import statement in a source file
// This is a language agnostic representation of an import statement.
// Not all attributes may be present in all languages
type ImportNode struct {
	Node

	// The package or module being imported
	moduleNameNode *sitter.Node

	// The object being imported
	moduleItemNode *sitter.Node

	// The alias of the import in the current scope
	moduleAliasNode *sitter.Node

	// Whether the import is a wildcard import
	isWildcardImport bool
}

// NewImportNode creates a new ImportNode instance
// using the import statement node from the tree-sitter parser
func NewImportNode(content Content) *ImportNode {
	return &ImportNode{
		Node:             Node{content},
		isWildcardImport: false,
	}
}

func (i *ImportNode) ModuleName() string {
	return i.contentForNode(i.moduleNameNode)
}

func (i *ImportNode) ModuleItem() string {
	return i.contentForNode(i.moduleItemNode)
}

func (i *ImportNode) ModuleAlias() string {
	return i.contentForNode(i.moduleAliasNode)
}

func (i *ImportNode) IsWildcardImport() bool {
	return i.isWildcardImport
}

func (i *ImportNode) GetModuleNameNode() *sitter.Node {
	return i.moduleNameNode
}

func (i *ImportNode) SetModuleNameNode(node *sitter.Node) {
	i.moduleNameNode = node
}

func (i *ImportNode) SetModuleItemNode(node *sitter.Node) {
	i.moduleItemNode = node
}

func (i *ImportNode) SetModuleAliasNode(node *sitter.Node) {
	i.moduleAliasNode = node
}

func (i *ImportNode) SetIsWildcardImport(isWildcardImport bool) {
	i.isWildcardImport = isWildcardImport
}

func (i *ImportNode) String() string {
	return fmt.Sprintf("ImportNode{ModuleName: %s, ModuleItem: %s, ModuleAlias: %s, WildcardImport: %t}",
		i.ModuleName(), i.ModuleItem(), i.ModuleAlias(), i.IsWildcardImport())
}