Function getAutocompleteInfo

packages/runtime-python/src/autocomplete.ts:223–261

(
  program: StructuredEditorProgram,
  type: Type,
  seen?: Set<string>
)

}

export function getAutocompleteInfo(
  program: StructuredEditorProgram,
  type: Type,
  seen?: Set<string>
): AutocompleteInfo[] {
  if (!seen) {
    seen = new Set()
  }

  if (type.category === TC.Module) {
    return suggestionsForEntries(program, type.fields, seen)
  } else if (type.category === TC.Class) {
    const suggestions: AutocompleteInfo[] = suggestionsForEntries(program, type.details.fields, seen, type.details.name)

    for (const base of type.details.baseClasses) {
      if (base.category !== TC.Class) {
        continue
      }

      suggestions.push(...suggestionsForEntries(program, (base as any).details.fields, seen, type.details.name))
    }

    return suggestions
  } else if (type.category === TC.Union) {
    const suggestions: AutocompleteInfo[] = []

    for (const subtype of type.subtypes) {
      suggestions.push(...getAutocompleteInfo(program, subtype, seen))
    }

    return suggestions
  } else if (type.category === TC.Any || type.category === TC.Unknown) {
    // pass
  } else {
    console.error('unhandled type category', type)
  }

  return []
}