Merge pull request #410 from vil02/add_to_kotlin

feat: add `to_kotlin`

Author: vil02

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "string-to-code"
-version = "0.4.1"
+version = "0.5.0"
 description = "Generates a piece of messy code in a given language displaying a given string "
 authors = ["piotr.idzik <[email protected]>"]
 readme = "./string_to_code/README.md"

string_to_code/to_kotlin.py

@@ -0,0 +1,71 @@
+"""
+provides string_to_kotlin utilities
+"""
+
+from . import core
+from . import utils
+
+
+_get_function_name = utils.get_function_name_fun("fun")
+
+
+def _atom_to_code(in_atom: core.Atom) -> str:
+    if in_atom.atom_char == "\n":
+        return "println()"
+    special_chars = {
+        '"': '\\"',
+        "'": "\\'",
+        "\\": "\\\\",
+        "\t": "\\t",
+        "{": "{",
+        "}": "}",
+    }
+    res_char = special_chars.get(in_atom.atom_char, in_atom.atom_char)
+    return f"print('{res_char}')"
+
+
+_function_call_str = utils.get_function_call_str_fun(_get_function_name, "", "()")
+
+
+_call_function_or_atom = utils.get_call_function_or_atom(
+    _atom_to_code, _function_call_str
+)
+
+_EMPTY_BODY = "    // No code needed"
+
+_body_to_str = utils.get_body_to_str(
+    "\n", "    ", _call_function_or_atom, "", _EMPTY_BODY
+)
+
+
+def _merge_to_full_function(in_function_name, in_function_body):
+    body_str = "\n" + in_function_body + "\n"
+    return f"fun {in_function_name}() {{{body_str}}}\n"
+
+
+_function_to_code = utils.get_function_to_code(
+    _get_function_name, _body_to_str, _merge_to_full_function
+)
+
+
+def _main_call_to_code(in_initial_call, **kwargs):
+    call_in_main = (
+        "\n" + _EMPTY_BODY
+        if in_initial_call is None
+        else "\n    " + _call_function_or_atom(in_initial_call, **kwargs)
+    )
+    return "fun main() {" + call_in_main + "\n}\n"
+
+
+def _join_to_final(main_call, function_definitions, **_kwargs):
+    function_definitions_str = (
+        "\n".join(function_definitions) if function_definitions else ""
+    )
+    if function_definitions_str:
+        function_definitions_str = function_definitions_str + "\n"
+    return function_definitions_str + main_call
+
+
+proc_printer_program, proc = utils.get_all_proc_functions(
+    _main_call_to_code, _function_to_code, _join_to_final
+)

system_setup_scripts/setup_for_kotlin.sh

@@ -0,0 +1,5 @@
+#!/usr/bin/env bash
+
+set -euo pipefail
+
+apt-get install -y --no-install-recommends default-jdk kotlin

…ample_data/kotlin/function_used_twice.kt …ample_data/kotlin/function_used_twice.kttests/unused_example_data/kotlin/function_used_twice.kt renamed to tests/example_data/kotlin/function_used_twice.kt tests/unused_example_data/kotlin/function_used_twice.kt renamed to tests/example_data/kotlin/function_used_twice.kt

@@ -0,0 +1,56 @@
+"""
+setup for the tests of the module string_to_kotlin
+"""
+
+import general_utilities as gu
+
+from string_to_code import to_kotlin
+
+_KOTLINC = "kotlinc"
+
+_FILE_EXTENSION = "kt"
+_JAVA = "java"
+
+
+def _compile_code(in_code: str, tmp_folder):
+    source_filename = gu.get_unique_filename(tmp_folder, _FILE_EXTENSION)
+    gu.save_str_to_file(tmp_folder / source_filename, in_code)
+    jar_filename = gu.get_unique_filename(tmp_folder, "jar")
+
+    gu.subprocess_run_with_check(
+        [
+            _KOTLINC,
+            "-Werror",
+            "-progressive",
+            source_filename,
+            "-d",
+            jar_filename,
+        ],
+        cwd=str(tmp_folder),
+    )
+    return jar_filename
+
+
+def _run_jar(jar_filename: str, tmp_folder):
+    return gu.subprocess_run_with_check(
+        [_JAVA, "-jar", jar_filename],
+        cwd=str(tmp_folder),
+        capture_output=True,
+        text=True,
+    )
+
+
+def _run_code(in_code: str, tmp_folder):
+    return _run_jar(_compile_code(in_code, tmp_folder), tmp_folder)
+
+
+def get_test_data():
+    """returns test data for the module string_to_kotlin"""
+    return gu.Language(
+        tool_names=[_KOTLINC, _JAVA],
+        string_to_code=to_kotlin.proc,
+        printer_program_to_code=to_kotlin.proc_printer_program,
+        run_code=_run_code,
+        id="kotlin",
+        source_code_file_extension=_FILE_EXTENSION,
+    )