runtime code

Author: zemse

src/HuffConfig.sol

@@ -216,6 +216,81 @@ contract HuffConfig {
         // set `msg.sender` for upcoming create context
         vm.prank(deployer);
 
+        vm.ffi(cleanup);
+    }
+
+    function runtime_code(string memory file) public payable returns (bytes memory bytecode) {
+        binary_check();
+
+        // Split the file into its parts
+        strings.slice memory s = file.toSlice();
+        strings.slice memory delim = "/".toSlice();
+        string[] memory parts = new string[](s.count(delim) + 1);
+        for (uint256 i = 0; i < parts.length; i++) {
+            parts[i] = s.split(delim).toString();
+        }
+
+        // Get the system time with our script
+        string[] memory time = new string[](1);
+        time[0] = "./lib/foundry-huff/scripts/rand_bytes.sh";
+        bytes memory retData = vm.ffi(time);
+        string memory rand_bytes = bytes32ToString(keccak256(abi.encode(bytes32(retData))));
+
+        // Re-concatenate the file with a "__TEMP__" prefix
+        string memory tempFile = parts[0];
+        if (parts.length <= 1) {
+            tempFile = string.concat("__TEMP__", rand_bytes, tempFile);
+        } else {
+            for (uint256 i = 1; i < parts.length - 1; i++) {
+                tempFile = string.concat(tempFile, "/", parts[i]);
+            }
+            tempFile = string.concat(tempFile, "/", "__TEMP__", rand_bytes, parts[parts.length - 1]);
+        }
+
+        // Paste the code in a new temp file
+        string[] memory create_cmds = new string[](3);
+        // TODO: create_cmds[0] = "$(find . -name \"file_writer.sh\")";
+        create_cmds[0] = "./lib/foundry-huff/scripts/file_writer.sh";
+        create_cmds[1] = string.concat("src/", tempFile, ".huff");
+        create_cmds[2] = string.concat(code, "\n");
+        vm.ffi(create_cmds);
+
+        // Append the real code to the temp file
+        string[] memory append_cmds = new string[](3);
+        append_cmds[0] = "./lib/foundry-huff/scripts/read_and_append.sh";
+        append_cmds[1] = string.concat("src/", tempFile, ".huff");
+        append_cmds[2] = string.concat("src/", file, ".huff");
+        vm.ffi(append_cmds);
+
+        /// Create a list of strings with the commands necessary to compile Huff contracts
+        string[] memory cmds = new string[](5);
+        if (const_overrides.length > 0) {
+            cmds = new string[](6 + const_overrides.length);
+            cmds[5] = "-c";
+
+            Constant memory cur_const;
+            for (uint256 i; i < const_overrides.length; i++) {
+                cur_const = const_overrides[i];
+                cmds[6 + i] = string.concat(cur_const.key, "=", cur_const.value);
+            }
+        }
+
+        cmds[0] = "huffc";
+        cmds[1] = string(string.concat("src/", tempFile, ".huff"));
+        cmds[2] = "-r";
+        cmds[3] = "-e";
+        cmds[4] = get_evm_version();
+
+        /// @notice compile the Huff contract and return the bytecode
+        bytecode = vm.ffi(cmds);
+
+        // Clean up temp files
+        string[] memory cleanup = new string[](2);
+        cleanup[0] = "rm";
+        cleanup[1] = string.concat("src/", tempFile, ".huff");
+
+        // set `msg.sender` for upcoming create context
+        vm.prank(deployer);
 
         vm.ffi(cleanup);
     }