@zzzen/pyright-internal 1.2.0-dev.20260816 → 1.2.0-dev.20260823

package/dist/analyzer/analyzerNodeInfo.d.ts

@@ -1,4 +1,5 @@
-import { ClassNode, ComprehensionNode, ExecutionScopeNode, FunctionNode, IfNode, LambdaNode, ModuleNode, ParseNode, StringNode } from '../parser/parseNodes';
+import { ClassNode, ComprehensionNode, ExecutionScopeNode, FunctionNode, IfNode, LambdaNode, ModuleNode, ParseNode, ExpressionNode, StringListNode, StringNode } from '../parser/parseNodes';
+import { StringAnnotationInfo } from '../parser/stringAnnotationInfo';
 import { AnalyzerFileInfo } from './analyzerFileInfo';
 import { FlowNode } from './codeFlowTypes';
 import { Declaration } from './declaration';
@@ -15,32 +16,126 @@ export interface AnalyzerNodeInfo {
     declaration?: Declaration;
     flowNode?: FlowNode;
     afterFlowNode?: FlowNode;
-    fileInfo?: AnalyzerFileInfo;
     codeFlowExpressions?: Set<string>;
     codeFlowComplexity?: number;
     staticConditionValue?: boolean;
     dunderAllInfo?: DunderAllInfo | undefined;
+    stringAnnotations?: WeakMap<AnalyzerFileInfo, WeakMap<StringListNode, ExpressionNode>>;
+}
+export interface AnalyzerNodeInfoReader {
+    get(node: ParseNode): AnalyzerNodeInfo | undefined;
+    getFileInfo(node: ParseNode): AnalyzerFileInfo | undefined;
+}
+interface AnalyzerNodeInfoReaderPropertyProvider {
+    readonly analyzerNodeInfoReader: AnalyzerNodeInfoReader;
+}
+interface AnalyzerNodeInfoReaderMethodProvider {
+    getAnalyzerNodeInfoReader(): AnalyzerNodeInfoReader;
+}
+export declare function getInfoReader(provider: AnalyzerNodeInfoReaderPropertyProvider): AnalyzerNodeInfoReader;
+export declare function getInfoReader(provider: AnalyzerNodeInfoReaderMethodProvider): AnalyzerNodeInfoReader;
+export interface AnalyzerNodeInfoWriter extends AnalyzerNodeInfoReader {
+    getOrCreate(node: ParseNode): AnalyzerNodeInfo;
+    setFileInfo(root: ModuleNode, fileInfo: AnalyzerFileInfo): void;
+}
+export declare class AnalyzerNodeInfoStore implements AnalyzerNodeInfoWriter {
+    private readonly _infoByTree;
+    private readonly _fileInfoByTree;
+    get(node: ParseNode): AnalyzerNodeInfo | undefined;
+    getFileInfo(node: ParseNode): AnalyzerFileInfo | undefined;
+    getOrCreate(node: ParseNode): AnalyzerNodeInfo;
+    setFileInfo(root: ModuleNode, fileInfo: AnalyzerFileInfo): void;
+}
+export interface AnalyzerNodeInfoBindingSession extends AnalyzerNodeInfoWriter {
+    readonly root: ModuleNode;
+}
+export interface AnalyzerNodeInfoContext extends AnalyzerNodeInfoReader {
+    getCurrentLayerReader(): AnalyzerNodeInfoReader;
+    beginWrite(root: ModuleNode): AnalyzerNodeInfoBindingSession;
+    publish(session: AnalyzerNodeInfoBindingSession): AnalyzerNodeInfoStore;
+    registerStore(root: ModuleNode, store: AnalyzerNodeInfoStore): void;
+    remove(root: ModuleNode): void;
+    enterOverlay(): void;
+    promoteToPreviousLayer(root: ModuleNode): void;
+    discardOverlay(): void;
+    dispose(): void;
+}
+export declare class AnalyzerNodeInfoContextImpl implements AnalyzerNodeInfoContext {
+    private _layers;
+    private _disposed;
+    private readonly _currentLayerReader;
+    get(node: ParseNode): AnalyzerNodeInfo | undefined;
+    getFileInfo(node: ParseNode): AnalyzerFileInfo | undefined;
+    getCurrentLayerReader(): AnalyzerNodeInfoReader;
+    beginWrite(root: ModuleNode): AnalyzerNodeInfoBindingSession;
+    publish(session: AnalyzerNodeInfoBindingSession): AnalyzerNodeInfoStore;
+    registerStore(root: ModuleNode, store: AnalyzerNodeInfoStore): void;
+    remove(root: ModuleNode): void;
+    enterOverlay(): void;
+    promoteToPreviousLayer(root: ModuleNode): void;
+    discardOverlay(): void;
+    dispose(): void;
+    private _createLayer;
+    private _getFromCurrentLayer;
+    private _getFileInfoFromCurrentLayer;
+    private _throwIfDisposed;
 }
 export type ScopedNode = ModuleNode | ClassNode | FunctionNode | LambdaNode | ComprehensionNode;
-export declare function cleanNodeAnalysisInfo(node: ParseNode): void;
-export declare function getImportInfo(node: ParseNode): ImportResult | undefined;
-export declare function setImportInfo(node: ParseNode, importInfo: ImportResult): void;
-export declare function getScope(node: ParseNode): Scope | undefined;
-export declare function setScope(node: ParseNode, scope: Scope): void;
-export declare function getDeclaration(node: ParseNode): Declaration | undefined;
-export declare function setDeclaration(node: ParseNode, decl: Declaration): void;
-export declare function getFlowNode(node: ParseNode): FlowNode | undefined;
-export declare function setFlowNode(node: ParseNode, flowNode: FlowNode): void;
-export declare function getAfterFlowNode(node: ParseNode): FlowNode | undefined;
-export declare function setAfterFlowNode(node: ParseNode, flowNode: FlowNode): void;
-export declare function getFileInfo(node: ParseNode): AnalyzerFileInfo;
-export declare function setFileInfo(node: ModuleNode, fileInfo: AnalyzerFileInfo): void;
-export declare function getCodeFlowExpressions(node: ExecutionScopeNode): Set<string> | undefined;
-export declare function setCodeFlowExpressions(node: ExecutionScopeNode, expressions: Set<string>): void;
-export declare function getCodeFlowComplexity(node: ExecutionScopeNode): number;
-export declare function setCodeFlowComplexity(node: ExecutionScopeNode, complexity: number): void;
-export declare function getStaticConditionValue(node: IfNode): boolean | undefined;
-export declare function setStaticConditionValue(node: IfNode, value: boolean | undefined): void;
-export declare function getDunderAllInfo(node: ModuleNode): DunderAllInfo | undefined;
-export declare function setDunderAllInfo(node: ModuleNode, names: DunderAllInfo | undefined): void;
-export declare function isCodeUnreachable(node: ParseNode): boolean;
+export declare class AnalyzerNodeInfoAccessor implements AnalyzerNodeInfoReader {
+    private readonly _reader;
+    private readonly _writer?;
+    constructor(_reader: AnalyzerNodeInfoReader, _writer?: AnalyzerNodeInfoWriter | undefined);
+    get(node: ParseNode): AnalyzerNodeInfo | undefined;
+    getImportInfo(node: ParseNode): ImportResult | undefined;
+    setImportInfo(node: ParseNode, importInfo: ImportResult): void;
+    getScope(node: ParseNode): Scope | undefined;
+    setScope(node: ParseNode, scope: Scope): void;
+    getDeclaration(node: ParseNode): Declaration | undefined;
+    setDeclaration(node: ParseNode, declaration: Declaration): void;
+    getStaticConditionValue(node: IfNode): boolean | undefined;
+    setStaticConditionValue(node: IfNode, value: boolean | undefined): void;
+    getFlowNode(node: ParseNode): FlowNode | undefined;
+    setFlowNode(node: ParseNode, flowNode: FlowNode): void;
+    getAfterFlowNode(node: ParseNode): FlowNode | undefined;
+    setAfterFlowNode(node: ParseNode, flowNode: FlowNode): void;
+    getFileInfoIfAvailable(node: ParseNode): AnalyzerFileInfo | undefined;
+    getFileInfo(node: ParseNode): AnalyzerFileInfo;
+    setFileInfo(node: ModuleNode, fileInfo: AnalyzerFileInfo): void;
+    getCodeFlowExpressions(node: ExecutionScopeNode): Set<string> | undefined;
+    setCodeFlowExpressions(node: ExecutionScopeNode, expressions: Set<string>): void;
+    getCodeFlowComplexity(node: ExecutionScopeNode): number;
+    setCodeFlowComplexity(node: ExecutionScopeNode, complexity: number): void;
+    getDunderAllInfo(node: ModuleNode): DunderAllInfo | undefined;
+    setDunderAllInfo(node: ModuleNode, names: DunderAllInfo | undefined): void;
+    getStringAnnotation(node: StringListNode): ExpressionNode | undefined;
+    setStringAnnotation(node: StringListNode, annotation: ExpressionNode, nestedAnnotations: StringAnnotationInfo): void;
+    isCodeUnreachable(node: ParseNode): boolean;
+    private _getWriter;
+    private _write;
+}
+export declare function createAnalyzerNodeInfoAccessor(reader: AnalyzerNodeInfoReader, writer?: AnalyzerNodeInfoWriter): AnalyzerNodeInfoAccessor;
+export declare function getImportInfo(node: ParseNode, reader: AnalyzerNodeInfoReader): ImportResult | undefined;
+export declare function setImportInfo(node: ParseNode, importInfo: ImportResult, writer: AnalyzerNodeInfoWriter): void;
+export declare function getScope(node: ParseNode, reader: AnalyzerNodeInfoReader): Scope | undefined;
+export declare function setScope(node: ParseNode, scope: Scope, writer: AnalyzerNodeInfoWriter): void;
+export declare function getDeclaration(node: ParseNode, reader: AnalyzerNodeInfoReader): Declaration | undefined;
+export declare function setDeclaration(node: ParseNode, decl: Declaration, writer: AnalyzerNodeInfoWriter): void;
+export declare function getStaticConditionValue(node: IfNode, reader: AnalyzerNodeInfoReader): boolean | undefined;
+export declare function setStaticConditionValue(node: IfNode, value: boolean | undefined, writer: AnalyzerNodeInfoWriter): void;
+export declare function getFlowNode(node: ParseNode, reader: AnalyzerNodeInfoReader): FlowNode | undefined;
+export declare function setFlowNode(node: ParseNode, flowNode: FlowNode, writer: AnalyzerNodeInfoWriter): void;
+export declare function getAfterFlowNode(node: ParseNode, reader: AnalyzerNodeInfoReader): FlowNode | undefined;
+export declare function setAfterFlowNode(node: ParseNode, flowNode: FlowNode, writer: AnalyzerNodeInfoWriter): void;
+export declare function getStringAnnotation(node: StringListNode, reader: AnalyzerNodeInfoReader): ExpressionNode | undefined;
+export declare function setStringAnnotation(node: StringListNode, annotation: ExpressionNode, nestedAnnotations: StringAnnotationInfo, reader: AnalyzerNodeInfoReader): void;
+export declare function getFileInfoIfAvailable(node: ParseNode, reader: AnalyzerNodeInfoReader): AnalyzerFileInfo | undefined;
+export declare function getFileInfo(node: ParseNode, reader: AnalyzerNodeInfoReader): AnalyzerFileInfo;
+export declare function setFileInfo(node: ModuleNode, fileInfo: AnalyzerFileInfo, writer: AnalyzerNodeInfoWriter): void;
+export declare function getCodeFlowExpressions(node: ExecutionScopeNode, reader: AnalyzerNodeInfoReader): Set<string> | undefined;
+export declare function setCodeFlowExpressions(node: ExecutionScopeNode, expressions: Set<string>, writer: AnalyzerNodeInfoWriter): void;
+export declare function getCodeFlowComplexity(node: ExecutionScopeNode, reader: AnalyzerNodeInfoReader): number;
+export declare function setCodeFlowComplexity(node: ExecutionScopeNode, complexity: number, writer: AnalyzerNodeInfoWriter): void;
+export declare function getDunderAllInfo(node: ModuleNode, reader: AnalyzerNodeInfoReader): DunderAllInfo | undefined;
+export declare function setDunderAllInfo(node: ModuleNode, names: DunderAllInfo | undefined, writer: AnalyzerNodeInfoWriter): void;
+export declare function isCodeUnreachable(node: ParseNode, reader: AnalyzerNodeInfoReader): boolean;
+export {};