From 475c4b84585d8c9595b2194f9c75e7896484b544 Mon Sep 17 00:00:00 2001
From: Raid5594 <52794079+Raid5594@users.noreply.github.com>
Date: Wed, 24 Apr 2024 15:36:17 +0200
Subject: [PATCH] extend utils & update compilation process (#396)

Co-authored-by: Raid Ateir <ra@matterlabs.dev>
---
 system-contracts/scripts/compile-yul.ts       | 44 ++++++++++++++++---
 .../scripts/preprocess-system-contracts.ts    | 14 ++++++
 system-contracts/scripts/utils.ts             |  1 +
 3 files changed, 53 insertions(+), 6 deletions(-)

diff --git a/system-contracts/scripts/compile-yul.ts b/system-contracts/scripts/compile-yul.ts
index 5b972ef92..67b468987 100644
--- a/system-contracts/scripts/compile-yul.ts
+++ b/system-contracts/scripts/compile-yul.ts
@@ -1,11 +1,23 @@
 // hardhat import should be the first import in the file
 import type { CompilerPaths } from "./utils";
-import { spawn, compilerLocation, prepareCompilerPaths, getSolcLocation } from "./utils";
+import {
+  spawn,
+  compilerLocation,
+  prepareCompilerPaths,
+  getSolcLocation,
+  needsRecompilation,
+  setCompilationTime,
+} from "./utils";
 import * as fs from "fs";
 import { Command } from "commander";
+import * as _path from "path";
 
 const COMPILER_VERSION = "1.3.18";
 const IS_COMPILER_PRE_RELEASE = true;
+const CONTRACTS_DIR = "contracts-preprocessed";
+const BOOTLOADER_DIR = "bootloader";
+const TIMESTAMP_FILE_YUL = "last_compilation_yul.timestamp";
+const TIMESTAMP_FILE_BOOTLOADER = "last_compilation_bootloader.timestamp";
 
 export async function compileYul(paths: CompilerPaths, file: string) {
   const solcCompilerPath = await getSolcLocation();
@@ -32,14 +44,34 @@ async function main() {
   program.version("0.1.0").name("compile yul").description("publish preimages for the L2 contracts");
 
   program.command("compile-bootloader").action(async () => {
-    await compileYulFolder("bootloader/build");
-    await compileYulFolder("bootloader/tests");
+    const timestampFilePath = _path.join(process.cwd(), TIMESTAMP_FILE_BOOTLOADER);
+    const folderToCheck = _path.join(process.cwd(), BOOTLOADER_DIR);
+
+    if (needsRecompilation(folderToCheck, timestampFilePath)) {
+      console.log("Compilation needed.");
+      await compileYulFolder("bootloader/build");
+      await compileYulFolder("bootloader/tests");
+      setCompilationTime(timestampFilePath);
+    } else {
+      console.log("Compilation not needed.");
+      return;
+    }
   });
 
   program.command("compile-precompiles").action(async () => {
-    await compileYulFolder("contracts-preprocessed");
-    await compileYulFolder("contracts-preprocessed/precompiles");
-    await compileYulFolder("contracts-preprocessed/precompiles/test-contracts");
+    const timestampFilePath = _path.join(process.cwd(), TIMESTAMP_FILE_YUL);
+    const folderToCheck = _path.join(process.cwd(), CONTRACTS_DIR);
+
+    if (needsRecompilation(folderToCheck, timestampFilePath)) {
+      console.log("Compilation needed.");
+      await compileYulFolder("contracts-preprocessed");
+      await compileYulFolder("contracts-preprocessed/precompiles");
+      await compileYulFolder("contracts-preprocessed/precompiles/test-contracts");
+      setCompilationTime(timestampFilePath);
+    } else {
+      console.log("Compilation not needed.");
+      return;
+    }
   });
 
   await program.parseAsync(process.argv);
diff --git a/system-contracts/scripts/preprocess-system-contracts.ts b/system-contracts/scripts/preprocess-system-contracts.ts
index acecee1ac..0b3690a9e 100644
--- a/system-contracts/scripts/preprocess-system-contracts.ts
+++ b/system-contracts/scripts/preprocess-system-contracts.ts
@@ -3,9 +3,11 @@ import path from "path";
 import { renderFile } from "template-file";
 import { glob } from "fast-glob";
 import { Command } from "commander";
+import { needsRecompilation, deleteDir, setCompilationTime, isFolderEmpty } from "./utils";
 
 const CONTRACTS_DIR = "contracts";
 const OUTPUT_DIR = "contracts-preprocessed";
+const TIMESTAMP_FILE = "last_compilation_preprocessing.timestamp"; // File to store the last compilation time
 
 const params = {
   SYSTEM_CONTRACTS_OFFSET: "0x8000",
@@ -17,6 +19,18 @@ async function preprocess(testMode: boolean) {
     params.SYSTEM_CONTRACTS_OFFSET = "0x9000";
   }
 
+  const timestampFilePath = path.join(process.cwd(), TIMESTAMP_FILE);
+  const folderToCheck = path.join(process.cwd(), CONTRACTS_DIR);
+
+  if ((await isFolderEmpty(OUTPUT_DIR)) || needsRecompilation(folderToCheck, timestampFilePath) || testMode) {
+    console.log("Preprocessing needed.");
+    deleteDir(OUTPUT_DIR);
+    setCompilationTime(timestampFilePath);
+  } else {
+    console.log("Preprocessing not needed.");
+    return;
+  }
+
   const contracts = await glob(
     [`${CONTRACTS_DIR}/**/*.sol`, `${CONTRACTS_DIR}/**/*.yul`, `${CONTRACTS_DIR}/**/*.zasm`],
     { onlyFiles: true }
diff --git a/system-contracts/scripts/utils.ts b/system-contracts/scripts/utils.ts
index 2deba9ce0..318fe84fa 100644
--- a/system-contracts/scripts/utils.ts
+++ b/system-contracts/scripts/utils.ts
@@ -7,6 +7,7 @@ import type { Deployer } from "@matterlabs/hardhat-zksync-deploy";
 import type { BigNumberish, BytesLike } from "ethers";
 import { BigNumber, ethers } from "ethers";
 import * as fs from "fs";
+import * as fsPr from "fs/promises";
 import { hashBytecode } from "zksync-web3/build/src/utils";
 import type { YulContractDescription, ZasmContractDescription } from "./constants";
 import { Language, SYSTEM_CONTRACTS } from "./constants";