@zzzen/pyright-internal 1.2.0-dev.20220703 → 1.2.0-dev.20220724

package/README.md

@@ -1 +1,82 @@
-# pyright-packager
+# @zzzen/pyright-internal
+
+Repackaged [pyright-internal](https://github.com/microsoft/pyright/tree/main/packages/pyright-internal) with type declaration enabled.
+
+# Installation
+
+```sh
+npm i @zzzen/pyright-internal
+```
+
+# Demo
+
+print ast of code
+
+```ts
+import { ImportResolver } from "@zzzen/pyright-internal/dist/analyzer/importResolver";
+import { printParseNodeType } from "@zzzen/pyright-internal/dist/analyzer/parseTreeUtils";
+import { ParseTreeWalker } from "@zzzen/pyright-internal/dist/analyzer/parseTreeWalker";
+import { Program } from "@zzzen/pyright-internal/dist/analyzer/program";
+import { ConfigOptions } from "@zzzen/pyright-internal/dist/common/configOptions";
+import {
+  lib,
+  sitePackages,
+} from "@zzzen/pyright-internal/dist/common/pathConsts";
+import {
+  combinePaths,
+  normalizeSlashes,
+} from "@zzzen/pyright-internal/dist/common/pathUtils";
+import { ParseNode } from "@zzzen/pyright-internal/dist/parser/parseNodes";
+import { PyrightFileSystem } from "@zzzen/pyright-internal/dist/pyrightFileSystem";
+import { TestAccessHost } from "@zzzen/pyright-internal/dist/tests/harness/testAccessHost";
+import { TestFileSystem } from "@zzzen/pyright-internal/dist/tests/harness/vfs/filesystem";
+
+const FILE_PATH = "/t.py";
+
+const defaultCode = `
+def f(a = 1):
+  return a
+f(1)
+`;
+
+// console.log(process.execArgv.join())
+const libraryRoot = combinePaths(normalizeSlashes("/"), lib, sitePackages);
+
+const tfs = new TestFileSystem(false, {
+  files: {
+    [FILE_PATH]: defaultCode,
+  },
+});
+
+const fs = new PyrightFileSystem(tfs);
+
+const configOptions = new ConfigOptions(normalizeSlashes("/"));
+configOptions.typeshedPath = normalizeSlashes(
+  "/node_modules/pyright/dist/typeshed-fallback"
+);
+const importResolver = new ImportResolver(
+  fs,
+  configOptions,
+  new TestAccessHost(fs.getModulePath(), [libraryRoot])
+);
+const program = new Program(importResolver, configOptions);
+program.setTrackedFiles([FILE_PATH]);
+
+while (program.analyze()) {
+  // Continue to call analyze until it completes. Since we're not
+  // specifying a timeout, it should complete the first time.
+}
+
+const sourceFile = program.getSourceFile(FILE_PATH)!;
+
+class PrintWalker extends ParseTreeWalker {
+  override visitNode(node: ParseNode) {
+    console.log(printParseNodeType(node.nodeType));
+    return super.visitNode(node);
+  }
+}
+
+new PrintWalker().walk(sourceFile.getParseResults()!.parseTree);
+```
+
+For more details, checkout [pyright-ast-viewer](https://github.com/Zzzen/pyright-ast-viewer).

package/dist/analyzer/analysis.js

@@ -30,7 +30,7 @@ exports.analyzeProgram = exports.nullCallback = void 0;
 const cancellationUtils_1 = require("../common/cancellationUtils");
 const debug = __importStar(require("../common/debug"));
 const timing_1 = require("../common/timing");
-const nullCallback = (_) => {
+const nullCallback = () => {
     /* empty */
 };
 exports.nullCallback = nullCallback;

package/dist/analyzer/analysis.js.map

@@ -1 +1 @@
-{"version":3,"file":"analysis.js","sourceRoot":"","sources":["../../../src/analyzer/analysis.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;;;;;;;;;;;;;;;;;;;;AAIH,mEAAuG;AAGvG,uDAAyC;AAEzC,6CAA4C;AAGrC,MAAM,YAAY,GAA6B,CAAC,CAAC,EAAE,EAAE;IACxD,WAAW;AACf,CAAC,CAAC;AAFW,QAAA,YAAY,gBAEvB;AAeF,SAAgB,cAAc,CAC1B,OAAgB,EAChB,OAAoC,EACpC,aAA4B,EAC5B,QAA8C,EAC9C,OAAyB,EACzB,KAAwB;IAExB,IAAI,aAAa,GAAG,KAAK,CAAC;IAE1B,QAAQ,GAAG,QAAQ,aAAR,QAAQ,cAAR,QAAQ,GAAI,oBAAY,CAAC;IAEpC,IAAI;QACA,IAAA,gDAA4B,EAAC,KAAK,CAAC,CAAC;QAEpC,MAAM,QAAQ,GAAG,IAAI,iBAAQ,EAAE,CAAC;QAChC,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAEhD,MAAM,kBAAkB,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC;QAC5D,MAAM,WAAW,GAAG,OAAO,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;QAC1D,MAAM,mBAAmB,GAAG,WAAW,CAAC,MAAM,CAAC;QAC/C,MAAM,WAAW,GAAG,QAAQ,CAAC,oBAAoB,EAAE,CAAC;QAEpD,wCAAwC;QACxC,IAAI,mBAAmB,GAAG,CAAC,IAAI,CAAC,aAAa,EAAE;YAC3C,QAAQ,CAAC;gBACL,WAAW;gBACX,cAAc,EAAE,OAAO,CAAC,YAAY,EAAE;gBACtC,sBAAsB,EAAE,kBAAkB;gBAC1C,qBAAqB,EAAE,OAAO,CAAC,uBAAuB,EAAE;gBACxD,kBAAkB,EAAE,KAAK;gBACzB,wBAAwB,EAAE,KAAK;gBAC/B,WAAW;aACd,CAAC,CAAC;SACN;KACJ;IAAC,OAAO,CAAM,EAAE;QACb,IAAI,8CAA0B,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE;YAClC,OAAO,KAAK,CAAC;SAChB;QAED,MAAM,OAAO,GAAG,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;QACxC,OAAO,CAAC,KAAK,CAAC,6BAA6B,GAAG,OAAO,CAAC,CAAC;QAEvD,QAAQ,CAAC;YACL,WAAW,EAAE,EAAE;YACf,cAAc,EAAE,CAAC;YACjB,sBAAsB,EAAE,CAAC;YACzB,qBAAqB,EAAE,IAAI;YAC3B,kBAAkB,EAAE,IAAI;YACxB,wBAAwB,EAAE,KAAK;YAC/B,WAAW,EAAE,CAAC;YACd,KAAK,EAAE,KAAK,CAAC,oBAAoB,CAAC,CAAC,CAAC;SACvC,CAAC,CAAC;KACN;IAED,OAAO,aAAa,CAAC;AACzB,CAAC;AAxDD,wCAwDC"}
+{"version":3,"file":"analysis.js","sourceRoot":"","sources":["../../../src/analyzer/analysis.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;;;;;;;;;;;;;;;;;;;;AAIH,mEAAuG;AAGvG,uDAAyC;AAEzC,6CAA4C;AAGrC,MAAM,YAAY,GAA6B,GAAG,EAAE;IACvD,WAAW;AACf,CAAC,CAAC;AAFW,QAAA,YAAY,gBAEvB;AAeF,SAAgB,cAAc,CAC1B,OAAgB,EAChB,OAAoC,EACpC,aAA4B,EAC5B,QAA8C,EAC9C,OAAyB,EACzB,KAAwB;IAExB,IAAI,aAAa,GAAG,KAAK,CAAC;IAE1B,QAAQ,GAAG,QAAQ,aAAR,QAAQ,cAAR,QAAQ,GAAI,oBAAY,CAAC;IAEpC,IAAI;QACA,IAAA,gDAA4B,EAAC,KAAK,CAAC,CAAC;QAEpC,MAAM,QAAQ,GAAG,IAAI,iBAAQ,EAAE,CAAC;QAChC,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAEhD,MAAM,kBAAkB,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC;QAC5D,MAAM,WAAW,GAAG,OAAO,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;QAC1D,MAAM,mBAAmB,GAAG,WAAW,CAAC,MAAM,CAAC;QAC/C,MAAM,WAAW,GAAG,QAAQ,CAAC,oBAAoB,EAAE,CAAC;QAEpD,wCAAwC;QACxC,IAAI,mBAAmB,GAAG,CAAC,IAAI,CAAC,aAAa,EAAE;YAC3C,QAAQ,CAAC;gBACL,WAAW;gBACX,cAAc,EAAE,OAAO,CAAC,YAAY,EAAE;gBACtC,sBAAsB,EAAE,kBAAkB;gBAC1C,qBAAqB,EAAE,OAAO,CAAC,uBAAuB,EAAE;gBACxD,kBAAkB,EAAE,KAAK;gBACzB,wBAAwB,EAAE,KAAK;gBAC/B,WAAW;aACd,CAAC,CAAC;SACN;KACJ;IAAC,OAAO,CAAM,EAAE;QACb,IAAI,8CAA0B,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE;YAClC,OAAO,KAAK,CAAC;SAChB;QAED,MAAM,OAAO,GAAG,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;QACxC,OAAO,CAAC,KAAK,CAAC,6BAA6B,GAAG,OAAO,CAAC,CAAC;QAEvD,QAAQ,CAAC;YACL,WAAW,EAAE,EAAE;YACf,cAAc,EAAE,CAAC;YACjB,sBAAsB,EAAE,CAAC;YACzB,qBAAqB,EAAE,IAAI;YAC3B,kBAAkB,EAAE,IAAI;YACxB,wBAAwB,EAAE,KAAK;YAC/B,WAAW,EAAE,CAAC;YACd,KAAK,EAAE,KAAK,CAAC,oBAAoB,CAAC,CAAC,CAAC;SACvC,CAAC,CAAC;KACN;IAED,OAAO,aAAa,CAAC;AACzB,CAAC;AAxDD,wCAwDC"}

package/dist/analyzer/analyzerFileInfo.d.ts

@@ -3,6 +3,7 @@ import { TextRangeDiagnosticSink } from '../common/diagnosticSink';
 import { TextRange } from '../common/textRange';
 import { TextRangeCollection } from '../common/textRangeCollection';
 import { Scope } from './scope';
+import { IPythonMode } from './sourceFile';
 import { SymbolTable } from './symbol';
 export interface AbsoluteModuleDescriptor {
     importingFilePath: string;
@@ -33,7 +34,7 @@ export interface AnalyzerFileInfo {
     isTypingExtensionsStubFile: boolean;
     isBuiltInStubFile: boolean;
     isInPyTypedPackage: boolean;
-    isIPythonMode: boolean;
+    ipythonMode: IPythonMode;
     accessedSymbolSet: Set<number>;
 }
 export declare function isAnnotationEvaluationPostponed(fileInfo: AnalyzerFileInfo): boolean;

package/dist/analyzer/analyzerFileInfo.js.map

@@ -1 +1 @@
-{"version":3,"file":"analyzerFileInfo.js","sourceRoot":"","sources":["../../../src/analyzer/analyzerFileInfo.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG;;;AA6CH,SAAgB,+BAA+B,CAAC,QAA0B;IACtE,OAAO,QAAQ,CAAC,aAAa,CAAC,GAAG,CAAC,aAAa,CAAC,KAAK,SAAS,IAAI,QAAQ,CAAC,UAAU,CAAC;AAC1F,CAAC;AAFD,0EAEC"}
+{"version":3,"file":"analyzerFileInfo.js","sourceRoot":"","sources":["../../../src/analyzer/analyzerFileInfo.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG;;;AA8CH,SAAgB,+BAA+B,CAAC,QAA0B;IACtE,OAAO,QAAQ,CAAC,aAAa,CAAC,GAAG,CAAC,aAAa,CAAC,KAAK,SAAS,IAAI,QAAQ,CAAC,UAAU,CAAC;AAC1F,CAAC;AAFD,0EAEC"}

package/dist/analyzer/binder.js

@@ -550,6 +550,20 @@ class Binder extends parseTreeWalker_1.ParseTreeWalker {
             this._createAssignmentTargetFlowNodes(node.leftExpression, /* walkTargets */ true, /* unbound */ false);
             createdAssignmentTargetFlowNodes = true;
         }
+        // If the assignment target base expression is potentially a
+        // TypedDict, add the base expression to the flow expressions set
+        // to accommodate TypedDict type narrowing.
+        if (node.leftExpression.nodeType === 24 /* Index */) {
+            const target = node.leftExpression;
+            if (target.items.length === 1 &&
+                !target.trailingComma &&
+                target.items[0].valueExpression.nodeType === 48 /* StringList */) {
+                if ((0, codeFlowTypes_1.isCodeFlowSupportedForReference)(target.baseExpression)) {
+                    const baseExprReferenceKey = (0, codeFlowTypes_1.createKeyForReference)(target.baseExpression);
+                    this._currentScopeCodeFlowExpressions.add(baseExprReferenceKey);
+                }
+            }
+        }
         this.walk(node.rightExpression);
         let isPossibleTypeAlias = true;
         if (ParseTreeUtils.getEnclosingFunction(node)) {
@@ -1126,7 +1140,7 @@ class Binder extends parseTreeWalker_1.ParseTreeWalker {
         // Make sure this is within an async lambda or function.
         const enclosingFunction = ParseTreeUtils.getEnclosingFunction(node);
         if (enclosingFunction === undefined || !enclosingFunction.isAsync) {
-            if (this._fileInfo.isIPythonMode && enclosingFunction === undefined) {
+            if (this._fileInfo.ipythonMode && enclosingFunction === undefined) {
                 // Top level await is allowed in ipython mode.
                 return true;
             }