Remove detect_vyper_version_from_source

Author: DanielSchiavini

tests/test_detect_vyper_version.py

@@ -0,0 +1,26 @@
+import pytest
+
+from vvm import detect_vyper_version_from_source
+
+
+def test_detect_vyper_version_from_source(foo_source, all_versions):
+    assert detect_vyper_version_from_source(foo_source) == str(all_versions)
+
+
+@pytest.mark.parametrize(
+    "version_str,decorator",
+    [
+        ("0.1.0b17", "public"),
+        ("0.3.0-beta17", "external"),
+        ("0.4.0rc6", "external"),
+    ],
+)
+def test_detect_vyper_version_beta(version_str, decorator):
+    source = f"""
+# @version {version_str}
+
+@{decorator}
+def foo() -> int128:
+    return 42
+    """
+    assert detect_vyper_version_from_source(source) == version_str

vvm/__init__.py

@@ -5,4 +5,10 @@
     install_vyper,
     set_vyper_version,
 )
-from vvm.main import compile_files, compile_source, compile_standard, get_vyper_version
+from vvm.main import (
+    compile_files,
+    compile_source,
+    compile_standard,
+    detect_vyper_version_from_source,
+    get_vyper_version,
+)

vvm/main.py

@@ -23,6 +23,30 @@ def get_vyper_version() -> Version:
     return wrapper._get_vyper_version(vyper_binary)
 
 
+def detect_vyper_version_from_source(source_code: str) -> Optional[str]:
+    """
+    Detect the version given by the pragma version in the source code.
+    TODO: when the user has a range, we should compare to the installed versions
+
+    Arguments
+    ---------
+    source_code : str
+        Source code to detect the version from.
+
+    Returns
+    -------
+    str
+        vyper version, or None if no version could be detected.
+    """
+    try:
+        finditer = VERSION_RE.finditer(source_code)
+        version_str = next(finditer).group(1)
+        Version(version_str)  # validate the version
+        return version_str
+    except (StopIteration, InvalidVersion):
+        return None
+
+
 def compile_source(
     source: str,
     base_path: Union[Path, str] = None,
@@ -57,6 +81,9 @@ def compile_source(
     Dict
         Compiler output. The source file name is given as `<stdin>`.
     """
+    if vyper_version is None:
+        vyper_version = detect_vyper_version_from_source(source)
+
     source_path = tempfile.mkstemp(suffix=".vy", prefix="vyper-", text=True)[1]
     with open(source_path, "w") as fp:
         fp.write(source)