Merge pull request #41763 from Thevakumar-Luheerathan/fix-ballerina-l…

…ang-iss-40264 Add local repository support for tools
ballerina-platform · Jan 8, 2024 · b910876 · b910876
2 parents b2ac9a9 + d79e43d
commit b910876
Show file tree

Hide file tree

Showing 77 changed files with 1,308 additions and 122 deletions.
diff --git a/cli/ballerina-cli/src/main/java/io/ballerina/cli/cmd/PushCommand.java b/cli/ballerina-cli/src/main/java/io/ballerina/cli/cmd/PushCommand.java
@@ -17,6 +17,8 @@
 */
 package io.ballerina.cli.cmd;
 
+import com.google.gson.Gson;
+import com.google.gson.JsonObject;
 import io.ballerina.cli.BLauncherCmd;
 import io.ballerina.cli.utils.FileUtils;
 import io.ballerina.projects.DependencyManifest;
@@ -44,14 +46,18 @@
 import org.wso2.ballerinalang.util.RepoUtils;
 import picocli.CommandLine;
 
+import java.io.BufferedReader;
 import java.io.File;
+import java.io.FileWriter;
 import java.io.IOException;
 import java.io.PrintStream;
+import java.nio.charset.StandardCharsets;
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.nio.file.StandardOpenOption;
 import java.util.Arrays;
+import java.util.Optional;
 import java.util.stream.Collectors;
 import java.util.zip.ZipEntry;
 import java.util.zip.ZipInputStream;
@@ -60,6 +66,7 @@
 import static io.ballerina.cli.utils.CentralUtils.authenticate;
 import static io.ballerina.cli.utils.CentralUtils.getBallerinaCentralCliTokenUrl;
 import static io.ballerina.cli.utils.CentralUtils.getCentralPackageURL;
+import static io.ballerina.projects.util.ProjectConstants.LOCAL_TOOLS_JSON;
 import static io.ballerina.projects.util.ProjectConstants.SETTINGS_FILE_NAME;
 import static io.ballerina.projects.util.ProjectUtils.getAccessTokenOfCLI;
 import static io.ballerina.projects.util.ProjectUtils.initializeProxy;
@@ -73,6 +80,11 @@
 @CommandLine.Command(name = PUSH_COMMAND, description = "Publish a package to Ballerina Central")
 public class PushCommand implements BLauncherCmd {
 
+    private static final String TOOL_DIR = "tool";
+    private static final String BAL_TOOL_JSON = "bal-tool.json";
+    private static final String TOOL_ID = "tool_id";
+    private static final String ORG = "org";
+    private static final String PACKAGE_NAME = "name";
     @CommandLine.Parameters (arity = "0..1")
     private Path balaPath;
 
@@ -391,9 +403,11 @@ private void pushBalaToCustomRepo(Path balaFilePath) {
                 ProjectUtils.deleteDirectory(balaCachesPath);
             }
             ProjectUtils.extractBala(balaFilePath, balaDestPath);
+            createLocalToolsJsonIfLocalTool(balaDestPath, org, packageName, repoPath.resolve(
+                    ProjectConstants.BALA_DIR_NAME));
         } catch (IOException e) {
             throw new ProjectException("error while pushing bala file '" + balaFilePath + "' to '"
-                    + ProjectConstants.LOCAL_REPOSITORY_NAME + "' repository. " + e.getMessage());
+                    + ProjectConstants.LOCAL_REPOSITORY_NAME + "' repository: " + e.getMessage());
         }
 
         Path relativePathToBalaFile;
@@ -406,6 +420,51 @@ private void pushBalaToCustomRepo(Path balaFilePath) {
                 + " to '" + repositoryName + "' repository.");
     }
 
+    private void createLocalToolsJsonIfLocalTool(Path balaDestPath, String org, String packageName,
+                                                 Path localRepoBalaPath) {
+        Path balToolJsonPath = balaDestPath.resolve(TOOL_DIR).resolve(BAL_TOOL_JSON);
+        JsonObject balToolJson;
+        JsonObject localToolJson;
+        Gson gson = new Gson();
+        if (!balToolJsonPath.toFile().exists()) {
+            return;
+        }
+        try (BufferedReader bufferedReader = Files.newBufferedReader(balToolJsonPath, StandardCharsets.UTF_8)) {
+            balToolJson = gson.fromJson(bufferedReader, JsonObject.class);
+        } catch (IOException e) {
+            throw new ProjectException("Failed to read bal-tools.json file: " + e.getMessage());
+        }
+        Optional<String> optionalToolId = Optional.ofNullable(balToolJson.get(TOOL_ID).getAsString());
+        if (optionalToolId.isEmpty()) {
+            return;
+        }
+        String toolId = optionalToolId.get();
+        JsonObject packageDesc = new JsonObject();
+        packageDesc.addProperty(ORG, org);
+        packageDesc.addProperty(PACKAGE_NAME, packageName);
+        Path localToolJsonPath = localRepoBalaPath.resolve(LOCAL_TOOLS_JSON);
+        if (localToolJsonPath.toFile().exists()) {
+            try (BufferedReader bufferedReader = Files.newBufferedReader(localToolJsonPath, StandardCharsets.UTF_8)) {
+                localToolJson = gson.fromJson(bufferedReader, JsonObject.class);
+                if (localToolJson.has(toolId)) {
+                    localToolJson.remove(toolId);
+                }
+                localToolJson.add(toolId, packageDesc);
+            } catch (IOException e) {
+                throw new ProjectException("Failed to read local-tools.json file: " + e.getMessage());
+            }
+        } else {
+            localToolJson = new JsonObject();
+            localToolJson.add(toolId, packageDesc);
+        }
+
+        try (FileWriter writer = new FileWriter(localToolJsonPath.toFile(), StandardCharsets.UTF_8)) {
+            writer.write(gson.toJson(localToolJson));
+        } catch (IOException e) {
+            throw new ProjectException("Failed to write local-tools.json file: " + e.getMessage());
+        }
+    }
+
     /**
      * Push a bala file to remote repository.
      *