🪟: Zigdows

Author: trnxdev

.gitignore

@@ -1,3 +1,4 @@
 zig-out
 zig-cache
-.zig-cache
+.zig-cache
+DO_NOT_MODIFY_U_STINKY-zigd-install-temp

src/utils.zig

@@ -5,17 +5,19 @@ pub const cpu_arch: []const u8 = switch (builtin.cpu.arch) {
     .x86, .powerpc64le, .x86_64, .aarch64, .riscv64 => |e| @tagName(e),
     else => @compileError("Unsupported CPU Architecture"),
 };
-pub const os: []const u8 = switch (builtin.os.tag) {
-    .linux, .macos => |e| @tagName(e),
-    else => @compileError("Unsupported OS"), // Windows too
+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."),
 };
+pub const os_tag = builtin.os.tag;
 pub const url_platform = os ++ "-" ++ cpu_arch;
-pub const archive_ext = if (builtin.os.tag == .windows) "zip" else "tar.xz"; // Maybe Windows support in future?
+pub const archive_ext = if (os_tag == .windows) "zip" else "tar.xz"; // Maybe Windows support in future?
 pub const index_url: []const u8 = "https://ziglang.org/download/index.json";
 pub const download_base_url: []const u8 = "https://ziglang.org/download";
 pub const download_base_master_url: []const u8 = "https://ziglang.org/builds"; // Master builds have another url for some unknown reason
 pub const zigd_version = @embedFile("zigd.version");
 pub const custom_env_path_key_for_zigd = "ZIGD_DIRECTORY";
+pub const binary_ext = if (os_tag == .windows) ".exe" else "";
 pub const InDebug = builtin.mode == .Debug;
 
 // == File System Stuff

src/zigdcli.zig

@@ -2,7 +2,6 @@ const std = @import("std");
 const zigdcore = @import("zigdcore.zig");
 const utils = @import("utils.zig");
 
-const stdout = std.io.getStdOut().writer();
 const Command = enum {
     help,
     install,
@@ -37,7 +36,7 @@ pub fn main() !void {
 }
 
 fn help_menu() !void {
-    try stdout.print(
+    try std.io.getStdOut().writer().print(
         \\> zigd ({s}) cli: Manage zigd stuff
         \\ 
         \\help - Outputs this help Menu
@@ -50,7 +49,7 @@ fn help_menu() !void {
 }
 
 fn version() !void {
-    try stdout.print("{s}\n", .{utils.zigd_version});
+    try std.io.getStdOut().writer().print("{s}\n", .{utils.zigd_version});
     return;
 }
 
@@ -59,7 +58,7 @@ var user_arg: zigdcore.ZigVersion.Source = .UserArg;
 fn install(allocator: std.mem.Allocator, args: []const []const u8) !void {
     if (args.len <= 2) {
         std.log.err("Wrong Usage!\n", .{});
-        try stdout.print(
+        try std.io.getStdOut().writer().print(
             \\Usage: zigd install [version]
             \\For more help use zigd help
             \\
@@ -70,7 +69,7 @@ fn install(allocator: std.mem.Allocator, args: []const []const u8) !void {
     var zig_version = try zigdcore.ZigVersion.parse(allocator, args[2], &user_arg, false);
     defer zig_version.deinitIfMasterOrZigver(allocator);
 
-    try stdout.print("Installing zig version {s}\n", .{zig_version});
+    try std.io.getStdOut().writer().print("Installing zig version {s}\n", .{zig_version});
 
     const download_url = try zigdcore.downloadUrlFromVersion(allocator, zig_version.as_string, zig_version.source == .Master);
     defer allocator.free(download_url);
@@ -84,7 +83,7 @@ fn install(allocator: std.mem.Allocator, args: []const []const u8) !void {
 fn exists(allocator: std.mem.Allocator, args: []const []const u8) !void {
     if (args.len <= 2) {
         std.log.err("Wrong Usage!\n", .{});
-        try stdout.print(
+        try std.io.getStdOut().writer().print(
             \\Usage: zigd exists [version]
             \\For more help use zigd help
             \\
@@ -102,8 +101,8 @@ fn exists(allocator: std.mem.Allocator, args: []const []const u8) !void {
     defer allocator.free(version_path);
 
     if (try utils.isDirectory(version_path)) {
-        try stdout.writeAll("Yes!\n");
+        try std.io.getStdOut().writer().writeAll("Yes!\n");
     } else {
-        try stdout.writeAll("No!\n");
+        try std.io.getStdOut().writer().writeAll("No!\n");
     }
 }

src/zigdcore.zig

@@ -28,13 +28,42 @@ pub fn install_zig(allocator: std.mem.Allocator, download_url: []const u8, insta
 
     try utils.createDirectoryIfNotExist(final_destination);
 
-    var final_dest = try std.fs.openDirAbsolute(final_destination, .{});
-    defer final_dest.close();
+    if (utils.os_tag == .windows) {
+        // std.zip has no strip components :( so we have to do this mess...
+        // TODO: It's kinda slow but it's std.zip's fault... yea, maybe we can do something to speed it up.
+        const temp_name = "DO_NOT_MODIFY_U_STINKY-zigd-install-temp";
 
-    var xz_decompressor = try std.compress.xz.decompress(allocator, req.reader());
-    defer xz_decompressor.deinit();
+        var temporary_storage = try std.fs.cwd().makeOpenPath(temp_name, .{
+            .iterate = true,
+        });
 
-    try std.tar.pipeToFileSystem(final_dest, xz_decompressor.reader(), .{ .strip_components = 1 });
+        const buf = try req.reader().readAllAlloc(allocator, std.math.maxInt(usize));
+        defer allocator.free(buf);
+
+        var fbs = std.io.fixedBufferStream(buf);
+        try std.zip.extract(temporary_storage, fbs.seekableStream(), .{});
+
+        var tempstrg = temporary_storage.iterate();
+        const w_path = try tempstrg.next() orelse return error.ZigInstallationWasLost__Oops;
+
+        const w_path_duped = try allocator.dupe(u8, w_path.name);
+        defer allocator.free(w_path_duped);
+
+        if ((try tempstrg.next()) != null)    
+            return error.InstallationWasSabotaged;
+
+        try temporary_storage.rename(w_path_duped, final_destination);
+        temporary_storage.close();
+        try std.fs.cwd().deleteDir(temp_name);
+    } else {    
+        var final_dest_dir = try std.fs.openDirAbsolute(final_destination, .{});
+        defer final_dest_dir.close();
+
+        var xz_decompressor = try std.compress.xz.decompress(allocator, req.reader());
+        defer xz_decompressor.deinit();
+    
+        try std.tar.pipeToFileSystem(final_dest_dir, xz_decompressor.reader(), .{ .strip_components = 1 });
+    }
     return;
 }
 
@@ -150,8 +179,15 @@ pub fn getZigdPath(allocator: std.mem.Allocator) ![]u8 {
 
     var custom_zigd: bool = true;
     const zigd_directory = env_map.get(utils.custom_env_path_key_for_zigd) orelse v: {
+        custom_zigd = false;
+
+        if (utils.os_tag == .windows) {
+            if (env_map.get("USERPROFILE")) |userprofile| {
+                break :v try std.fs.path.join(allocator, &.{ userprofile, ".zigd" });
+            }
+        }
+        
         if (env_map.get("HOME")) |home| {
-            custom_zigd = false;
             break :v try std.fs.path.join(allocator, &.{ home, ".zigd" });
         }
 

src/zigdemu.zig

@@ -56,7 +56,7 @@ pub fn main() !void {
     zig_version = try zigdcore.ZigVersion.parse(allocator, zig_version.as_string, &zig_version.source, true);
     defer zig_version.deinitIfMasterOrZigver(allocator);
 
-    const zig_binary_path = try std.fs.path.join(allocator, &.{ zigd_path, "versions", zig_version.as_string, "zig" });
+    const zig_binary_path = try std.fs.path.join(allocator, &.{ zigd_path, "versions", zig_version.as_string, "zig" ++ utils.binary_ext });
     defer allocator.free(zig_binary_path);
 
     if (!(try utils.isFile(zig_binary_path))) o: {