Implement zigd setup

Author: trnxdev

README.md

@@ -9,12 +9,13 @@
 
 ### Commands:
 
-| Command           | What does it do?                                  |
-| :---------------: | :-----------------------------------------------: |
-| help              | Help screen                                       |
-| version           | Outputs zigd version                              |
-| install [version] | Installs a zig version                            |
-| exists  [version] | Check if a zig version is installed on the system |
+| Command           | What does it do?                                             |
+| :---------------: | :----------------------------------------------------------: |
+| install [version] | Installs a zig version                                       |
+| setup [version]   | Installs a zig version and sets it as default in the config. |
+| exists  [version] | Check if a zig version is installed on the system            |
+| help              | Help screen                                                  |
+| version           | Outputs zigd version                                         |
 
 ## zigdemu
 

build.zig.zon

@@ -1,6 +1,6 @@
 .{
     .name = "zigd",
-    .version = "0.0.2",
+    .version = "0.0.3",
     .minimum_zig_version = "0.13.0",
     .paths = .{
         "build.zig",

src/utils.zig

@@ -3,11 +3,11 @@ const builtin = @import("builtin");
 
 pub const cpu_arch: []const u8 = switch (builtin.cpu.arch) {
     .x86, .powerpc64le, .x86_64, .aarch64, .riscv64 => |e| @tagName(e),
-    else => @compileError("Unsupported CPU Architecture"),
+    else => @compileError("Unsupported CPU Architecture, if you think your CPU Architecture is supported file an issue on github."),
 };
 pub const os: []const u8 = switch (os_tag) {
     .windows, .linux, .macos => |e| @tagName(e),
-    else => @compileError("Unsupported OS, if you think your OS supported file an issue on github."),
+    else => @compileError("Unsupported OS, if you think your OS is supported file an issue on github."),
 };
 pub const os_tag = builtin.os.tag;
 pub const url_platform = os ++ "-" ++ cpu_arch;

src/zigd.version

@@ -1 +1 @@
-0.0.2
+0.0.3

src/zigdcli.zig

@@ -3,10 +3,11 @@ const zigdcore = @import("zigdcore.zig");
 const utils = @import("utils.zig");
 
 const Command = enum {
-    help,
     install,
-    version,
+    setup,
     exists,
+    help,
+    version,
 };
 
 pub fn main() !void {
@@ -25,6 +26,7 @@ pub fn main() !void {
     if (std.meta.stringToEnum(Command, args[1])) |command| {
         return try switch (command) {
             .install => install(allocator, args),
+            .setup => setup(allocator, args),
             .exists => exists(allocator, args),
             .help => help_menu(),
             .version => version(),
@@ -39,10 +41,11 @@ fn help_menu() !void {
     try std.io.getStdOut().writer().print(
         \\> zigd ({s}) cli: Manage zigd stuff
         \\ 
-        \\help - Outputs this help Menu
-        \\version - Outputs zigd version
         \\install [version] - Install a zig version
+        \\setup [version] - First time setup (creates a config and installs the version)
         \\exists [version] - Check if a zig version is installed on the system
+        \\help - Outputs this help Menu
+        \\version - Outputs zigd version
         \\
     , .{utils.zigd_version});
     return;
@@ -77,7 +80,78 @@ fn install(allocator: std.mem.Allocator, args: []const []const u8) !void {
     const zigd_path = try zigdcore.getZigdPath(allocator);
     defer allocator.free(zigd_path);
 
-    try zigdcore.install_zig(allocator, download_url, zigd_path, zig_version.as_string);
+    if (!try zigdcore.install_zig(allocator, download_url, zigd_path, zig_version)) {
+        std.log.err("Installation failed!", .{});
+    }
+
+    return;
+}
+
+fn setup(allocator: std.mem.Allocator, args: []const []const u8) !void {
+    if (args.len <= 2) {
+        std.log.err("Wrong Usage!\n", .{});
+        try std.io.getStdOut().writer().print(
+            \\Usage: zigd setup [version]
+            \\For more help use zigd help
+            \\
+        , .{});
+        return;
+    }
+
+    const zigd_path = try zigdcore.getZigdPath(allocator);
+    defer allocator.free(zigd_path);
+
+    const config_path = try std.fs.path.join(allocator, &.{ zigd_path, "config" });
+    defer allocator.free(config_path);
+
+    if (try utils.isFile(config_path)) {
+        o: while (true) {
+            std.log.warn("A config file already exists, overwrite (y/n?)", .{});
+            const byte = try std.io.getStdIn().reader().readByte();
+
+            switch (byte) {
+                'y' => break :o,
+                'n' => return,
+                else => continue :o,
+            }
+        }
+    }
+
+    var zig_version = try zigdcore.ZigVersion.parse(allocator, args[2], &user_arg, false);
+    defer zig_version.deinitIfMasterOrZigverOrZonver(allocator);
+
+    if (zig_version.source == .Master) {
+        o: while (true) {
+            std.log.warn("It is not recommended to setup with master, are you sure you want to set master as default in the config? (y/n)", .{});
+            const byte = try std.io.getStdIn().reader().readByte();
+
+            switch (byte) {
+                'y' => break :o,
+                'n' => return,
+                else => continue :o,
+            }
+        }
+    }
+
+    const download_url = try zigdcore.downloadUrlFromVersion(allocator, zig_version.as_string, zig_version.source == .Master);
+    defer allocator.free(download_url);
+
+    try std.io.getStdOut().writer().print("Installing version {s}...\n", .{zig_version});
+    _ = try zigdcore.install_zig(allocator, download_url, zigd_path, zig_version);
+
+    try std.io.getStdOut().writer().print("Creating a config...\n", .{});
+    const config_file = try std.fs.createFileAbsolute(config_path, .{
+        .truncate = true,
+    });
+    defer config_file.close();
+
+    try config_file.writer().writeAll("# Generated by `zigd setup`, btw :)\n");
+    try config_file.writer().writeAll("default=");
+    switch (zig_version.source) {
+        .Master => try config_file.writeAll("master"),
+        else => try config_file.writeAll(zig_version.as_string),
+    }
+    try config_file.writer().writeByte('\n');
 }
 
 fn exists(allocator: std.mem.Allocator, args: []const []const u8) !void {

src/zigdcore.zig

@@ -2,8 +2,27 @@ const std = @import("std");
 const utils = @import("utils.zig");
 
 /// Version cannot be master!
-/// Returns the path to the zig binary
-pub fn install_zig(allocator: std.mem.Allocator, download_url: []const u8, install_path: []const u8, version: []const u8) !void {
+/// Returns a bool if it installed/reinstalled zig
+pub fn install_zig(allocator: std.mem.Allocator, download_url: []const u8, install_path: []const u8, zig_version: ZigVersion) !bool {
+    const final_destination = try std.fs.path.join(allocator, &.{ install_path, "versions", zig_version.as_string });
+    defer allocator.free(final_destination);
+
+    if (try utils.isDirectory(final_destination)) {
+        o: while (true) {
+            const byte = try std.io.getStdIn().reader().readByte();
+            std.log.warn("Version {} is already installed on your system! Re-install? (y/n)", .{zig_version});
+
+            switch (byte) {
+                'y' => {
+                    try std.fs.deleteTreeAbsolute(final_destination);
+                    break :o;
+                },
+                'n' => return false,
+                else => continue :o,
+            }
+        }
+    }
+
     var client = std.http.Client{ .allocator = allocator };
     defer client.deinit();
 
@@ -23,10 +42,7 @@ pub fn install_zig(allocator: std.mem.Allocator, download_url: []const u8, insta
     if (req.response.status != .ok)
         return error.ResponseWasNotOk;
 
-    const final_destination = try std.fs.path.join(allocator, &.{ install_path, "versions", version });
-    defer allocator.free(final_destination);
-
-    try utils.createDirectoryIfNotExist(final_destination);
+    try utils.createDirectoryIgnoreExist(final_destination);
 
     if (utils.os_tag == .windows) {
         // std.zip has no strip components :( so we have to do this mess...
@@ -64,7 +80,8 @@ pub fn install_zig(allocator: std.mem.Allocator, download_url: []const u8, insta
 
         try std.tar.pipeToFileSystem(final_dest_dir, xz_decompressor.reader(), .{ .strip_components = 1 });
     }
-    return;
+
+    return true;
 }
 
 pub fn masterFromIndex(allocator: std.mem.Allocator) ![]const u8 {

src/zigdemu.zig

@@ -78,7 +78,11 @@ pub fn main() !void {
             const download_url = try zigdcore.downloadUrlFromVersion(allocator, zig_version.as_string, zig_version.source == .Master);
             defer allocator.free(download_url);
 
-            try zigdcore.install_zig(allocator, download_url, zigd_path, zig_version.as_string);
+            if (!try zigdcore.install_zig(allocator, download_url, zigd_path, zig_version)) {
+                std.log.err("Installation failed! Exiting...", .{});
+                return;
+            }
+
             break :o;
         }