Snippet tools

2026-10-09 21:08:31 +02:00 · 2025-11-26 06:23:25 +00:00
parent 5517016d3b
commit c551dac79f
43 changed files with 2197 additions and 0 deletions
@@ -0,0 +1,125 @@
+import re
+import shutil
+import subprocess
+import textwrap
+from pathlib import Path
+
+from .base import CompileResult, Language, generic_shorten, template, trim_commonpath
+
+
+class LanguageCsharp(Language):
+    NAME = "csharp"
+    SNIPPET_FILENAME = "csharp.cs"
+
+    @classmethod
+    def compile(cls, tmpdir: Path, fnames: list[Path]) -> CompileResult:
+        shutil.copytree("templates/csharp", tmpdir)
+
+        csproj_path = tmpdir / "csharp.csproj"
+        csproj_path.write_text(
+            csproj_path.read_text().replace(
+                "../../clients/csharp/",
+                f"{Path('clients/csharp').absolute()}/",
+            )
+        )
+
+        trimmed_fnames = trim_commonpath(fnames)
+
+        src_cases = []
+
+        result = CompileResult()
+
+        for no, (snippet_fname, trimmed_fname) in enumerate(trimmed_fnames.items()):
+            content = (
+                snippet_fname.read_text()
+                .replace("public class Snippet", f"public class Snippet{no}")
+                .replace("{collection_name}", "collection_name")
+            )
+            dest_path = tmpdir / "s" / trimmed_fname.parent / f"Snippet{no}.cs"
+            dest_path.parent.mkdir(parents=True, exist_ok=True)
+            dest_path.write_text(content)
+            src_cases.append(
+                f'case "{snippet_fname}": await Snippet{no}.Run(); break;\n'
+            )
+            result.run_args[snippet_fname] = [
+                "dotnet",
+                f"{tmpdir}/bin/snippets-amalgamation.dll",
+                snippet_fname,
+            ]
+
+        template(
+            original_fname=Path("templates/csharp/Program.cs"),
+            target_fname=tmpdir / "Program.cs",
+            replacements={"// %cases%": "".join(src_cases)},
+        )
+
+        subprocess.run(["dotnet", "build", "-o", "bin"], cwd=tmpdir, check=True)
+
+        return result
+
+    RE_CODE = re.compile(
+        r"""
+        (?P<imports> .*? )
+        \s*public\ class\ Snippet\s*\{\s*
+        public\ static\ async\ Task\ Run\(\)\s*\{\n
+        (?P<body> .*? )
+        \s*\}\s*}\s*
+        $
+        """,
+        re.DOTALL | re.VERBOSE,
+    )
+
+    EXAMPLE_CODE = textwrap.dedent(
+        """\
+        using Qdrant.Client;
+        using Qdrant.Client.Grpc;
+
+        public class Snippet
+        {
+            public static async Task Run()
+            {
+                // Your code here
+            }
+        }
+        """
+    )
+    assert RE_CODE.match(EXAMPLE_CODE) is not None
+
+    @classmethod
+    def shorten(cls, contents: str) -> str:
+        if (m := LanguageCsharp.RE_CODE.match(contents)) is None:
+            msg = "Invalid snippet format"
+            raise ValueError(msg)
+        return generic_shorten(
+            m["imports"].strip() + "\n\n" + textwrap.dedent(m["body"]).strip()
+        )
+
+    RE_RENDERED = re.compile(
+        r"""
+        (?P<imports> (?:using\s[^;]+;\n|\n)* )
+        (?P<body> .* )
+        $
+        """,
+        re.DOTALL | re.VERBOSE,
+    )
+
+    @classmethod
+    def unshorten(cls, contents: str) -> str:
+        if m := LanguageCsharp.RE_RENDERED.match(contents):
+            return textwrap.dedent(
+                """\
+                {imports}
+
+                public class Snippet
+                {{
+                \tpublic static async Task Run()
+                \t{{
+                {body}
+                \t}}
+                }}
+                """
+            ).format(
+                imports=m["imports"].strip(),
+                body=textwrap.indent(m["body"].strip(), "\t" * 2),
+            )
+        return contents