Function findLayerModuleName

internal/layergraph/extract.go:364–412

findLayerModuleName resolves the imported identifier name for the "Layer" export from the "effect" package. Falls back to "Layer" if not found.

(sf *ast.SourceFile)

// findLayerModuleName resolves the imported identifier name for the "Layer" export
// from the "effect" package. Falls back to "Layer" if not found.
func findLayerModuleName(sf *ast.SourceFile) string {
	if sf == nil {
		return "Layer"
	}
	for _, stmt := range sf.Statements.Nodes {
		if stmt.Kind != ast.KindImportDeclaration {
			continue
		}
		importDecl := stmt.AsImportDeclaration()
		if importDecl.ModuleSpecifier == nil {
			continue
		}
		moduleName := scanner.GetTextOfNode(importDecl.ModuleSpecifier)
		// Strip quotes from module specifier.
		if len(moduleName) >= 2 && (moduleName[0] == '"' || moduleName[0] == '\'') {
			moduleName = moduleName[1 : len(moduleName)-1]
		}
		if moduleName != "effect" {
			continue
		}
		// Look for named imports.
		if importDecl.ImportClause == nil {
			continue
		}
		clause := importDecl.ImportClause.AsImportClause()
		if clause.NamedBindings == nil || clause.NamedBindings.Kind != ast.KindNamedImports {
			continue
		}
		namedImports := clause.NamedBindings.AsNamedImports()
		if namedImports.Elements == nil {
			continue
		}
		for _, elem := range namedImports.Elements.Nodes {
			spec := elem.AsImportSpecifier()
			// The "real" name is PropertyName if set (import { Layer as X }),
			// otherwise Name.
			importedName := ""
			if spec.PropertyName != nil {
				importedName = scanner.GetTextOfNode(spec.PropertyName)
			} else {
				importedName = scanner.GetTextOfNode(spec.Name())
			}
			if importedName == "Layer" {
				return scanner.GetTextOfNode(spec.Name())
			}
		}
	}
	return "Layer"
}

// collectChildIndices gathers graph node indices for the given AST nodes,
// filtering out nodes that aren't in the graph.