Merge pull request #110 from KotlinIsland/fix-intellij

intellij: fix narrowing

Author: KotlinIsland

.gitattributes

@@ -0,0 +1 @@
+intellij/src/test/java/** linguist-vendored

.github/workflows/check.yaml

@@ -2,9 +2,6 @@ name: check
 
 on: push
 
-env:
-  POETRY_VERSION: "1.7.1"
-
 jobs:
   check:
     runs-on: ubuntu-latest
@@ -27,11 +24,7 @@ jobs:
       - uses: actions/setup-python@v5
         with:
           python-version: ${{ matrix.python-version }}
-      - name: Run image
-        uses: abatilo/[email protected]
-        with:
-          poetry-version: ${{ env.POETRY_VERSION }}
-      - run: poetry config virtualenvs.in-project true
+      - run: ./pw poetry config virtualenvs.in-project true
       - name: Set up cache
         uses: actions/cache@v3
         id: cache
@@ -41,9 +34,9 @@ jobs:
       - name: Ensure cache is healthy
         if: steps.cache.outputs.cache-hit == 'true'
         run: timeout 10s poetry run pip --version || rm -rf .venv
-      - run: poetry install
-      - run: poetry run mypy -p basedtyping -p tests --python-version ${{ matrix.usable-python-version }}
-      - run: poetry run pytest tests/
+      - run: ./pw poetry install
+      - run: ./pw poetry run mypy -p basedtyping -p tests --python-version ${{ matrix.usable-python-version }}
+      - run: ./pw poetry run pytest tests/
 
   lint:
     runs-on: ubuntu-latest
@@ -52,11 +45,7 @@ jobs:
       - uses: actions/setup-python@v5
         with:
           python-version: "3.8"
-      - name: Run image
-        uses: abatilo/[email protected]
-        with:
-          poetry-version: ${{ env.POETRY_VERSION }}
-      - run: poetry config virtualenvs.in-project true
+      - run: ./pw poetry config virtualenvs.in-project true
       - name: Set up cache
         uses: actions/cache@v3
         id: cache
@@ -65,7 +54,7 @@ jobs:
           key: venv-${{ runner.os }}-${{ steps.full-python-version.outputs.version }}-${{ hashFiles('**/poetry.lock') }}
       - name: Ensure cache is healthy
         if: steps.cache.outputs.cache-hit == 'true'
-        run: timeout 10s poetry run pip --version || rm -rf .venv
-      - run: poetry install
-      - run: poetry run ruff format --check --diff
-      - run: poetry run ruff check
+        run: timeout 10s ./pw poetry run pip --version || rm -rf .venv
+      - run: ./pw poetry install
+      - run: ./pw poetry run ruff format --check --diff
+      - run: ./pw poetry run ruff check

.gitignore

@@ -1,4 +1,5 @@
 **/__pycache__
 /.venv/
 /venv/
-/dist/
+/dist/
+/.pyprojectx

.idea/basedtyping.iml

@@ -4,8 +4,12 @@
 
 ## [Unreleased]
 
+## [0.1.3]
+
 ### Added
 - Callable literal types `() -> int`
+### Fixed
+- Narrowed types were broken
 
 ## [0.1.2] - 2023-01-12
 

.idea/modules.xml

@@ -26,9 +26,9 @@ dependencies {
 
 // Set the JVM language level used to build the project. Use Java 11 for 2020.3+, and Java 17 for 2022.2+.
 kotlin {
-    @Suppress("UnstableApiUsage")
     jvmToolchain {
         languageVersion = JavaLanguageVersion.of(17)
+        @Suppress("UnstableApiUsage")
         vendor = JvmVendorSpec.JETBRAINS
     }
 }

intellij/CHANGELOG.md

@@ -4,22 +4,22 @@ pluginGroup = org.basedsoft.plugins.basedtyping
 pluginName = basedtyping
 pluginRepositoryUrl = https://github.com/KotlinIsland/basedtyping
 # SemVer format -> https://semver.org
-pluginVersion = 0.1.2
+pluginVersion = 0.1.3
 
 # Supported build number ranges and IntelliJ Platform versions -> https://plugins.jetbrains.com/docs/intellij/build-number-ranges.html
 pluginSinceBuild = 233
-pluginUntilBuild = 233.*
+pluginUntilBuild = 241.*
 
 # IntelliJ Platform Properties -> https://plugins.jetbrains.com/docs/intellij/tools-gradle-intellij-plugin.html#configuration-intellij-extension
-platformType = IC
-platformVersion = 2023.3.2
+platformType = PC
+platformVersion = 2023.3.3
 
 # Plugin Dependencies -> https://plugins.jetbrains.com/docs/intellij/plugin-dependencies.html
 # Example: platformPlugins = com.intellij.java, com.jetbrains.php:203.4449.22
-platformPlugins = PythonCore:233.11799.300
+platformPlugins = PythonCore
 
 # Gradle Releases -> https://github.com/gradle/gradle/releases
-gradleVersion = 8.5
+gradleVersion = 8.6
 
 # Opt-out flag for bundling Kotlin standard library -> https://jb.gg/intellij-platform-kotlin-stdlib
 kotlin.stdlib.default.dependency = false

intellij/build.gradle.kts

@@ -3,11 +3,12 @@
 annotations = "24.1.0"
 
 # plugins
-kotlin = "1.9.21"
+kotlin = "1.9.22"
 changelog = "2.2.0"
-gradleIntelliJPlugin = "1.16.1"
+gradleIntelliJPlugin = "1.17.1"
 kover = "0.7.5"
 
+
 [libraries]
 annotations = { group = "org.jetbrains", name = "annotations", version.ref = "annotations" }
 

intellij/gradle.properties

@@ -1,6 +1,6 @@
 distributionBase=GRADLE_USER_HOME
 distributionPath=wrapper/dists
-distributionUrl=https\://services.gradle.org/distributions/gradle-8.5-bin.zip
+distributionUrl=https\://services.gradle.org/distributions/gradle-8.6-bin.zip
 networkTimeout=10000
 validateDistributionUrl=true
 zipStoreBase=GRADLE_USER_HOME

intellij/gradle/libs.versions.toml

@@ -43,11 +43,11 @@ set JAVA_EXE=java.exe
 %JAVA_EXE% -version >NUL 2>&1
 if %ERRORLEVEL% equ 0 goto execute
 
-echo.
-echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
-echo.
-echo Please set the JAVA_HOME variable in your environment to match the
-echo location of your Java installation.
+echo. 1>&2
+echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH. 1>&2
+echo. 1>&2
+echo Please set the JAVA_HOME variable in your environment to match the 1>&2
+echo location of your Java installation. 1>&2
 
 goto fail
 
@@ -57,11 +57,11 @@ set JAVA_EXE=%JAVA_HOME%/bin/java.exe
 
 if exist "%JAVA_EXE%" goto execute
 
-echo.
-echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME%
-echo.
-echo Please set the JAVA_HOME variable in your environment to match the
-echo location of your Java installation.
+echo. 1>&2
+echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME% 1>&2
+echo. 1>&2
+echo Please set the JAVA_HOME variable in your environment to match the 1>&2
+echo location of your Java installation. 1>&2
 
 goto fail
 

intellij/gradle/wrapper/gradle-wrapper.properties

@@ -45,7 +45,7 @@ private class BasedTypingTypeProvider : PyTypeProviderBase() {
         val param = referenceExpression.followAssignmentsChain(PyResolveContext.defaultContext(context)).element
         if (param !is PyNamedParameter) return null
         val annotation = param.annotation?.value ?: return null
-        return getType(annotation, context)
+        return getType(annotation, context, simple=true)
     }
     /**
      * Needed to work around a limitation in PyTypingTypeProvider
@@ -137,6 +137,7 @@ fun getOverload(param: PyNamedParameter, func: PyFunction, context: TypeEvalCont
             .filterNotNull()
             .toSet()
             .takeIf { it.isNotEmpty() }
+            // TODO: keep the correct order
             ?.let { PyUnionType.union(it) }
 
 fun getOverloadReturn(func: PyFunction, context: TypeEvalContext) =

intellij/gradlew.bat

@@ -0,0 +1,128 @@
+// Copyright 2000-2021 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
+package com.jetbrains.python;
+
+import com.intellij.openapi.application.Application;
+import com.intellij.openapi.application.ApplicationManager;
+import com.intellij.openapi.projectRoots.*;
+import com.intellij.openapi.roots.OrderRootType;
+import com.intellij.openapi.util.KeyWithDefaultValue;
+import com.intellij.openapi.vfs.LocalFileSystem;
+import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.util.containers.ContainerUtil;
+import com.jetbrains.python.codeInsight.typing.PyTypeShed;
+import com.jetbrains.python.codeInsight.userSkeletons.PyUserSkeletonsUtil;
+import com.jetbrains.python.psi.LanguageLevel;
+import com.jetbrains.python.sdk.PythonSdkUtil;
+import org.jdom.Element;
+import org.jetbrains.annotations.NonNls;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import java.io.File;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+public final class PythonMockSdk {
+
+    private PythonMockSdk() {
+    }
+
+    public static @NotNull Sdk create() {
+        return create(LanguageLevel.getLatest());
+    }
+
+    public static @NotNull Sdk create(@NotNull String sdkPath) {
+        return create(sdkPath, LanguageLevel.getLatest());
+    }
+
+    public static @NotNull Sdk create(@NotNull LanguageLevel level, VirtualFile @NotNull ... additionalRoots) {
+        return create(PythonHelpersLocator.getPythonCommunityPath() + "/testData" + "/MockSdk", level, additionalRoots);
+    }
+
+    private static @NotNull Sdk create(@NotNull String sdkPath, @NotNull LanguageLevel level, VirtualFile @NotNull ... additionalRoots) {
+        String sdkName = "Mock " + PyNames.PYTHON_SDK_ID_NAME + " " + level.toPythonVersion();
+        return create(sdkName, sdkPath, new PyMockSdkType(level), level, additionalRoots);
+    }
+
+    public static @NotNull Sdk create(@NotNull String sdkName,
+                                      @NotNull String sdkPath,
+                                      @NotNull SdkTypeId sdkType,
+                                      @NotNull LanguageLevel level,
+                                      VirtualFile @NotNull ... additionalRoots) {
+        Sdk sdk = ProjectJdkTable.getInstance().createSdk(sdkName, sdkType);
+        SdkModificator sdkModificator = sdk.getSdkModificator();
+        sdkModificator.setHomePath(sdkPath + "/bin/python");
+        sdkModificator.setVersionString(toVersionString(level));
+
+        createRoots(sdkPath, level).forEach(vFile -> {
+            sdkModificator.addRoot(vFile, OrderRootType.CLASSES);
+        });
+
+        Arrays.asList(additionalRoots).forEach(vFile -> {
+            sdkModificator.addRoot(vFile, OrderRootType.CLASSES);
+        });
+
+        Application application = ApplicationManager.getApplication();
+        Runnable runnable = () -> sdkModificator.commitChanges();
+        if (application.isDispatchThread()) {
+            application.runWriteAction(runnable);
+        } else {
+            application.invokeAndWait(() -> application.runWriteAction(runnable));
+        }
+        sdk.putUserData(KeyWithDefaultValue.create("MOCK_PY_MARKER_KEY", true), true);
+        return sdk;
+
+        // com.jetbrains.python.psi.resolve.PythonSdkPathCache.getInstance() corrupts SDK, so have to clone
+        //return sdk.clone();
+    }
+
+    private static @NotNull List<VirtualFile> createRoots(@NotNull @NonNls String mockSdkPath, @NotNull LanguageLevel level) {
+        final var result = new ArrayList<VirtualFile>();
+
+        final var localFS = LocalFileSystem.getInstance();
+        ContainerUtil.addIfNotNull(result, localFS.refreshAndFindFileByIoFile(new File(mockSdkPath, "Lib")));
+        ContainerUtil.addIfNotNull(result, localFS.refreshAndFindFileByIoFile(new File(mockSdkPath, PythonSdkUtil.SKELETON_DIR_NAME)));
+
+        ContainerUtil.addIfNotNull(result, PyUserSkeletonsUtil.getUserSkeletonsDirectory());
+
+        result.addAll(PyTypeShed.INSTANCE.findRootsForLanguageLevel(level));
+
+        return result;
+    }
+
+    private static @NotNull String toVersionString(@NotNull LanguageLevel level) {
+        return "Python " + level.toPythonVersion();
+    }
+
+    private static final class PyMockSdkType implements SdkTypeId {
+
+        @NotNull
+        private final LanguageLevel myLevel;
+
+        private PyMockSdkType(@NotNull LanguageLevel level) {
+            myLevel = level;
+        }
+
+        @NotNull
+        @Override
+        public String getName() {
+            return PyNames.PYTHON_SDK_ID_NAME;
+        }
+
+        @Override
+        public @NotNull String getVersionString(@NotNull Sdk sdk) {
+            return toVersionString(myLevel);
+        }
+
+        @Override
+        public void saveAdditionalData(@NotNull SdkAdditionalData additionalData, @NotNull Element additional) {
+        }
+
+        @Nullable
+        @Override
+        public SdkAdditionalData loadAdditionalData(@NotNull Sdk currentSdk, @NotNull Element additional) {
+            return null;
+        }
+    }
+}

intellij/src/main/kotlin/org/basedsoft/plugins/basedtyping/BasedTypingTypeProvider.kt

@@ -0,0 +1 @@
+This stuff is copied from JetBrains/intellij-community because they don't ship it