const bun = @import("bun");
const default_allocator = bun.default_allocator;
const Global = bun.Global;
const json_parser = bun.JSON;
const logger = bun.logger;
const Output = bun.Output;
const FileSystem = @import("../fs.zig").FileSystem;
const Install = @import("./install.zig");
const DependencyID = Install.DependencyID;
const PackageManager = Install.PackageManager;
const Integrity = @import("./integrity.zig").Integrity;
const Npm = @import("./npm.zig");
const Resolution = @import("./resolution.zig").Resolution;
const Semver = @import("./semver.zig");
const std = @import("std");
const string = @import("../string_types.zig").string;
const strings = @import("../string_immutable.zig");
const ExtractTarball = @This();

name: strings.StringOrTinyString,
resolution: Resolution,
registry: string,
cache_dir: std.fs.Dir,
temp_dir: std.fs.Dir,
dependency_id: DependencyID,
skip_verify: bool = false,
integrity: Integrity = .{},
url: string = "",
package_manager: *PackageManager,

pub inline fn run(this: ExtractTarball, bytes: []const u8) !Install.ExtractData {
    if (!this.skip_verify and this.integrity.tag.isSupported()) {
        if (!this.integrity.verify(bytes)) {
            this.package_manager.log.addErrorFmt(
                null,
                logger.Loc.Empty,
                this.package_manager.allocator,
                "Integrity check failed<r> for tarball: {s}",
                .{this.name.slice()},
            ) catch unreachable;
            return error.IntegrityCheckFailed;
        }
    }
    return this.extract(bytes);
}

pub fn buildURL(
    registry_: string,
    full_name_: strings.StringOrTinyString,
    version: Semver.Version,
    string_buf: []const u8,
) !string {
    return try buildURLWithPrinter(
        registry_,
        full_name_,
        version,
        string_buf,
        @TypeOf(FileSystem.instance.dirname_store),
        string,
        anyerror,
        FileSystem.instance.dirname_store,
        FileSystem.DirnameStore.print,
    );
}

pub fn buildURLWithWriter(
    comptime Writer: type,
    writer: Writer,
    registry_: string,
    full_name_: strings.StringOrTinyString,
    version: Semver.Version,
    string_buf: []const u8,
) !void {
    const Printer = struct {
        writer: Writer,

        pub fn print(this: @This(), comptime fmt: string, args: anytype) Writer.Error!void {
            return try std.fmt.format(this.writer, fmt, args);
        }
    };

    return try buildURLWithPrinter(
        registry_,
        full_name_,
        version,
        string_buf,
        Printer,
        void,
        Writer.Error,
        Printer{
            .writer = writer,
        },
        Printer.print,
    );
}

pub fn buildURLWithPrinter(
    registry_: string,
    full_name_: strings.StringOrTinyString,
    version: Semver.Version,
    string_buf: []const u8,
    comptime PrinterContext: type,
    comptime ReturnType: type,
    comptime ErrorType: type,
    printer: PrinterContext,
    comptime print: fn (ctx: PrinterContext, comptime str: string, args: anytype) ErrorType!ReturnType,
) ErrorType!ReturnType {
    const registry = std.mem.trimRight(u8, registry_, "/");
    const full_name = full_name_.slice();

    var name = full_name;
    if (name[0] == '@') {
        if (std.mem.indexOfScalar(u8, name, '/')) |i| {
            name = name[i + 1 ..];
        }
    }

    const default_format = "{s}/{s}/-/";

    if (!version.tag.hasPre() and !version.tag.hasBuild()) {
        const args = .{ registry, full_name, name, version.major, version.minor, version.patch };
        return try print(
            printer,
            default_format ++ "{s}-{d}.{d}.{d}.tgz",
            args,
        );
    } else if (version.tag.hasPre() and version.tag.hasBuild()) {
        const args = .{ registry, full_name, name, version.major, version.minor, version.patch, version.tag.pre.slice(string_buf), version.tag.build.slice(string_buf) };
        return try print(
            printer,
            default_format ++ "{s}-{d}.{d}.{d}-{s}+{s}.tgz",
            args,
        );
    } else if (version.tag.hasPre()) {
        const args = .{ registry, full_name, name, version.major, version.minor, version.patch, version.tag.pre.slice(string_buf) };
        return try print(
            printer,
            default_format ++ "{s}-{d}.{d}.{d}-{s}.tgz",
            args,
        );
    } else if (version.tag.hasBuild()) {
        const args = .{ registry, full_name, name, version.major, version.minor, version.patch, version.tag.build.slice(string_buf) };
        return try print(
            printer,
            default_format ++ "{s}-{d}.{d}.{d}+{s}.tgz",
            args,
        );
    } else {
        unreachable;
    }
}

threadlocal var final_path_buf: [bun.MAX_PATH_BYTES]u8 = undefined;
threadlocal var folder_name_buf: [bun.MAX_PATH_BYTES]u8 = undefined;
threadlocal var json_path_buf: [bun.MAX_PATH_BYTES]u8 = undefined;

fn extract(this: *const ExtractTarball, tgz_bytes: []const u8) !Install.ExtractData {
    var tmpdir = this.temp_dir;
    var tmpname_buf: [256]u8 = undefined;
    const name = this.name.slice();

    var basename = this.name.slice();
    if (basename[0] == '@') {
        if (std.mem.indexOfScalar(u8, basename, '/')) |i| {
            basename = basename[i + 1 ..];
        }
    }

    var resolved: string = "";
    var tmpname = try FileSystem.instance.tmpname(basename[0..@min(basename.len, 32)], &tmpname_buf, tgz_bytes.len);
    {
        var extract_destination = tmpdir.makeOpenPathIterable(std.mem.span(tmpname), .{}) catch |err| {
            this.package_manager.log.addErrorFmt(
                null,
                logger.Loc.Empty,
                this.package_manager.allocator,
                "{s} when create temporary directory named \"{s}\" (while extracting \"{s}\")",
                .{ @errorName(err), tmpname, name },
            ) catch unreachable;
            return error.InstallFailed;
        };

        defer extract_destination.close();

        if (PackageManager.verbose_install) {
            Output.prettyErrorln("[{s}] Start extracting {s}<r>", .{ name, tmpname });
            Output.flush();
        }

        const Archive = @import("../libarchive/libarchive.zig").Archive;
        const Zlib = @import("../zlib.zig");
        var zlib_pool = Npm.Registry.BodyPool.get(default_allocator);
        zlib_pool.data.reset();
        defer Npm.Registry.BodyPool.release(zlib_pool);

        var zlib_entry = try Zlib.ZlibReaderArrayList.init(tgz_bytes, &zlib_pool.data.list, default_allocator);
        zlib_entry.readAll() catch |err| {
            this.package_manager.log.addErrorFmt(
                null,
                logger.Loc.Empty,
                this.package_manager.allocator,
                "{s} decompressing \"{s}\"",
                .{ @errorName(err), name },
            ) catch unreachable;
            return error.InstallFailed;
        };
        switch (this.resolution.tag) {
            .github => {
                const DirnameReader = struct {
                    needs_first_dirname: bool = true,
                    outdirname: *[]const u8,
                    pub fn onFirstDirectoryName(dirname_reader: *@This(), first_dirname: []const u8) void {
                        std.debug.assert(dirname_reader.needs_first_dirname);
                        dirname_reader.needs_first_dirname = false;
                        dirname_reader.outdirname.* = FileSystem.DirnameStore.instance.append([]const u8, first_dirname) catch unreachable;
                    }
                };
                var dirname_reader = DirnameReader{ .outdirname = &resolved };

                _ = if (PackageManager.verbose_install)
                    try Archive.extractToDir(
                        zlib_pool.data.list.items,
                        extract_destination,
                        null,
                        *DirnameReader,
                        &dirname_reader,
                        // for GitHub tarballs, the root dir is always <user>-<repo>-<commit_id>
                        1,
                        true,
                        true,
                    )
                else
                    try Archive.extractToDir(
                        zlib_pool.data.list.items,
                        extract_destination,
                        null,
                        *DirnameReader,
                        &dirname_reader,
                        // for GitHub tarballs, the root dir is always <user>-<repo>-<commit_id>
                        1,
                        true,
                        false,
                    );

                // This tag is used to know which version of the package was
                // installed from GitHub. package.json version becomes sort of
                // meaningless in cases like this.
                if (resolved.len > 0) insert_tag: {
                    const gh_tag = extract_destination.dir.createFileZ(".bun-tag", .{ .truncate = true }) catch break :insert_tag;
                    defer gh_tag.close();
                    gh_tag.writeAll(resolved) catch {
                        extract_destination.dir.deleteFileZ(".bun-tag") catch {};
                    };
                }
            },
            else => {
                _ = if (PackageManager.verbose_install)
                    try Archive.extractToDir(
                        zlib_pool.data.list.items,
                        extract_destination,
                        null,
                        void,
                        void{},
                        // for npm packages, the root dir is always "package"
                        1,
                        true,
                        true,
                    )
                else
                    try Archive.extractToDir(
                        zlib_pool.data.list.items,
                        extract_destination,
                        null,
                        void,
                        void{},
                        // for npm packages, the root dir is always "package"
                        1,
                        true,
                        false,
                    );
            },
        }

        if (PackageManager.verbose_install) {
            Output.prettyErrorln("[{s}] Extracted<r>", .{name});
            Output.flush();
        }
    }
    const folder_name = switch (this.resolution.tag) {
        .npm => this.package_manager.cachedNPMPackageFolderNamePrint(&folder_name_buf, name, this.resolution.value.npm.version),
        .github => PackageManager.cachedGitHubFolderNamePrint(&folder_name_buf, resolved),
        else => unreachable,
    };
    if (folder_name.len == 0 or (folder_name.len == 1 and folder_name[0] == '/')) @panic("Tried to delete root and stopped it");
    var cache_dir = this.cache_dir;
    cache_dir.deleteTree(folder_name) catch {};

    // e.g. @next
    // if it's a namespace package, we need to make sure the @name folder exists
    if (basename.len != name.len) {
        cache_dir.makeDir(std.mem.trim(u8, name[0 .. name.len - basename.len], "/")) catch {};
    }

    // Now that we've extracted the archive, we rename.
    std.os.renameatZ(tmpdir.fd, tmpname, cache_dir.fd, folder_name) catch |err| {
        this.package_manager.log.addErrorFmt(
            null,
            logger.Loc.Empty,
            this.package_manager.allocator,
            "moving \"{s}\" to cache dir failed: {s}\n  From: {s}\n    To: {s}",
            .{ name, @errorName(err), tmpname, folder_name },
        ) catch unreachable;
        return error.InstallFailed;
    };

    // We return a resolved absolute absolute file path to the cache dir.
    // To get that directory, we open the directory again.
    var final_dir = cache_dir.openDirZ(folder_name, .{}, true) catch |err| {
        this.package_manager.log.addErrorFmt(
            null,
            logger.Loc.Empty,
            this.package_manager.allocator,
            "failed to verify cache dir for \"{s}\": {s}",
            .{ name, @errorName(err) },
        ) catch unreachable;
        return error.InstallFailed;
    };

    defer final_dir.close();
    // and get the fd path
    var final_path = bun.getFdPath(
        final_dir.fd,
        &final_path_buf,
    ) catch |err| {
        this.package_manager.log.addErrorFmt(
            null,
            logger.Loc.Empty,
            this.package_manager.allocator,
            "failed to resolve cache dir for \"{s}\": {s}",
            .{ name, @errorName(err) },
        ) catch unreachable;
        return error.InstallFailed;
    };

    // create an index storing each version of a package installed
    create_index: {
        var index_dir = cache_dir.makeOpenPathIterable(name, .{}) catch break :create_index;
        defer index_dir.close();
        index_dir.dir.symLink(
            final_path,
            switch (this.resolution.tag) {
                .github => folder_name["@GH@".len..],
                // trim "name@" from the prefix
                .npm => folder_name[name.len + 1 ..],
                else => folder_name,
            },
            .{},
        ) catch break :create_index;
    }

    var json_path: []u8 = "";
    var json_buf: []u8 = "";
    var json_len: usize = 0;
    switch (this.resolution.tag) {
        .github => {
            const json_file = final_dir.openFileZ("package.json", .{ .mode = .read_only }) catch |err| {
                this.package_manager.log.addErrorFmt(
                    null,
                    logger.Loc.Empty,
                    this.package_manager.allocator,
                    "\"package.json\" for \"{s}\" failed to open: {s}",
                    .{ name, @errorName(err) },
                ) catch unreachable;
                return error.InstallFailed;
            };
            defer json_file.close();
            const json_stat = try json_file.stat();
            json_buf = try this.package_manager.allocator.alloc(u8, json_stat.size + 64);
            json_len = try json_file.preadAll(json_buf, 0);

            json_path = bun.getFdPath(
                json_file.handle,
                &json_path_buf,
            ) catch |err| {
                this.package_manager.log.addErrorFmt(
                    null,
                    logger.Loc.Empty,
                    this.package_manager.allocator,
                    "\"package.json\" for \"{s}\" failed to resolve: {s}",
                    .{ name, @errorName(err) },
                ) catch unreachable;
                return error.InstallFailed;
            };
            // TODO remove extracted files not matching any globs under "files"
        },
        else => {},
    }

    const ret_json_path = try FileSystem.instance.dirname_store.append(@TypeOf(json_path), json_path);
    return .{
        .url = this.url,
        .resolved = resolved,
        .json_path = ret_json_path,
        .json_buf = json_buf,
        .json_len = json_len,
    };
}
p;follow=1'>deps: add missing mimalloc dep to jsc bindings header generator</a></td><td><img src='//www.gravatar.com/avatar/5a1fe8229d73768d36ff624423f984dd?s=13&amp;d=retro' width='13' height='13' alt='Gravatar' /> Kenta Iwasaki
</td><td>2</td><td><span class='deletions'>-2</span>/<span class='insertions'>+3</span></td></tr>
<tr class='nohover-highlight'><td/><td colspan='5' class='logmsg'>
</td></tr>
<tr class='logheader'><td><span title='2021-11-24 17:09:54 +0000'>2021-11-24</span></td><td class='logsubject'><a href='/ansg191/bun/commit/build.zig?h=lithdew/picohttp-mimalloc&amp;id=283a01493c6c4b5ba155de15fde52e6637bdbc01&amp;follow=1'>deps: build picohttp and mimalloc using zig</a></td><td><img src='//www.gravatar.com/avatar/5a1fe8229d73768d36ff624423f984dd?s=13&amp;d=retro' width='13' height='13' alt='Gravatar' /> Kenta Iwasaki
</td><td>2</td><td><span class='deletions'>-131</span>/<span class='insertions'>+123</span></td></tr>
<tr class='nohover-highlight'><td/><td colspan='5' class='logmsg'>
Reduce the amount of dependencies which need to be externally built
before Bun can be built.

Remove clause to enforce that Bash syntax is used in Makefile, as the
clause only works on a few Linux distros which assume bash is located at
/bin/bash.


</td></tr>
<tr class='logheader'><td><span title='2021-11-23 16:55:26 -0500'>2021-11-23</span></td><td class='logsubject'><a href='/ansg191/bun/commit/packages/bun-framework-next/client.development.tsx?h=lithdew/picohttp-mimalloc&amp;id=7660f727450908f322211e04a32dc69207d54044&amp;follow=1'>update Next version on global</a></td><td><img src='//www.gravatar.com/avatar/194e8d62c38b487ec0744d2ccc0931fc?s=13&amp;d=retro' width='13' height='13' alt='Gravatar' /> Jack Hanford
</td><td>1</td><td><span class='deletions'>-1</span>/<span class='insertions'>+1</span></td></tr>
<tr class='nohover-highlight'><td/><td colspan='5' class='logmsg'>
</td></tr>
<tr class='logheader'><td><span title='2021-11-23 13:54:41 -0800'>2021-11-23</span></td><td class='logsubject'><a href='/ansg191/bun/commit/build-id?h=lithdew/picohttp-mimalloc&amp;id=4e4f2058c2f086beefda1368c2db53f46fc4f6a0&amp;follow=1'>Update build-id</a></td><td><img src='//www.gravatar.com/avatar/764ea52180f7f5236744034dd5e1a3c0?s=13&amp;d=retro' width='13' height='13' alt='Gravatar' /> Jarred Sumner
</td><td>1</td><td><span class='deletions'>-1</span>/<span class='insertions'>+1</span></td></tr>
<tr class='nohover-highlight'><td/><td colspan='5' class='logmsg'>
</td></tr>
<tr class='logheader'><td><span title='2021-11-23 16:23:31 -0500'>2021-11-23</span></td><td class='logsubject'><a href='/ansg191/bun/commit/packages/bun-framework-next/client.development.tsx?h=lithdew/picohttp-mimalloc&amp;id=ca25e32c0f4de45584982eeb4bad9f2fa5422b8a&amp;follow=1'>fix shallow routing</a></td><td><img src='//www.gravatar.com/avatar/194e8d62c38b487ec0744d2ccc0931fc?s=13&amp;d=retro' width='13' height='13' alt='Gravatar' /> Jack Hanford
</td><td>1</td><td><span class='deletions'>-4</span>/<span class='insertions'>+6</span></td></tr>
<tr class='nohover-highlight'><td/><td colspan='5' class='logmsg'>
</td></tr>
<tr class='logheader'><td><span title='2021-11-23 11:05:05 -0500'>2021-11-23</span></td><td class='logsubject'><a href='/ansg191/bun/commit/packages/bun-framework-next/client.development.tsx?h=lithdew/picohttp-mimalloc&amp;id=5a1f60c729f9e9d4b987d552a45f510a738a1805&amp;follow=1'>remove comment</a></td><td><img src='//www.gravatar.com/avatar/194e8d62c38b487ec0744d2ccc0931fc?s=13&amp;d=retro' width='13' height='13' alt='Gravatar' /> Jack Hanford
</td><td>1</td><td><span class='deletions'>-1</span>/<span class='insertions'>+0</span></td></tr>
<tr class='nohover-highlight'><td/><td colspan='5' class='logmsg'>
</td></tr>
<tr class='logheader'><td><span title='2021-11-23 10:53:44 -0500'>2021-11-23</span></td><td class='logsubject'><a href='/ansg191/bun/commit/packages/bun-framework-next/package.json?h=lithdew/picohttp-mimalloc&amp;id=ac242f94ca358365cb6eb0a8d9ab2ed62b6470b9&amp;follow=1'>add react-dom as devDep</a></td><td><img src='//www.gravatar.com/avatar/194e8d62c38b487ec0744d2ccc0931fc?s=13&amp;d=retro' width='13' height='13' alt='Gravatar' /> Jack Hanford
</td><td>1</td><td><span class='deletions'>-0</span>/<span class='insertions'>+1</span></td></tr>
<tr class='nohover-highlight'><td/><td colspan='5' class='logmsg'>
</td></tr>
<tr class='logheader'><td><span title='2021-11-23 10:51:08 -0500'>2021-11-23</span></td><td class='logsubject'><a href='/ansg191/bun/commit/packages/bun-framework-next/text-encoder-polyfill.js?h=lithdew/picohttp-mimalloc&amp;id=9ab9b01a640cd157634494b6b703d095897ac412&amp;follow=1'>stop installing textencoder</a></td><td><img src='//www.gravatar.com/avatar/194e8d62c38b487ec0744d2ccc0931fc?s=13&amp;d=retro' width='13' height='13' alt='Gravatar' /> Jack Hanford
</td><td>3</td><td><span class='deletions'>-2</span>/<span class='insertions'>+308</span></td></tr>
<tr class='nohover-highlight'><td/><td colspan='5' class='logmsg'>
</td></tr>
<tr class='logheader'><td><span title='2021-11-23 10:42:04 -0500'>2021-11-23</span></td><td class='logsubject'><a href='/ansg191/bun/commit/packages/bun-framework-next/renderDocument.tsx?h=lithdew/picohttp-mimalloc&amp;id=cdeb39bd7f0f2fa280f2c0ab833b789d3099f3ed&amp;follow=1'>begin addressing more feedback</a></td><td><img src='//www.gravatar.com/avatar/194e8d62c38b487ec0744d2ccc0931fc?s=13&amp;d=retro' width='13' height='13' alt='Gravatar' /> Jack Hanford
</td><td>1</td><td><span class='deletions'>-3</span>/<span class='insertions'>+3</span></td></tr>
<tr class='nohover-highlight'><td/><td colspan='5' class='logmsg'>
</td></tr>
<tr class='logheader'><td><span title='2021-11-22 22:19:53 -0500'>2021-11-22</span></td><td class='logsubject'><a href='/ansg191/bun/commit/packages/bun-framework-next/renderDocument.tsx?h=lithdew/picohttp-mimalloc&amp;id=1d01ffeb9e3fa90b009afe08a43b34843856c074&amp;follow=1'>remove .then</a></td><td><img src='//www.gravatar.com/avatar/194e8d62c38b487ec0744d2ccc0931fc?s=13&amp;d=retro' width='13' height='13' alt='Gravatar' /> Jack Hanford
</td><td>1</td><td><span class='deletions'>-12</span>/<span class='insertions'>+3</span></td></tr>
<tr class='nohover-highlight'><td/><td colspan='5' class='logmsg'>
</td></tr>
<tr class='logheader'><td><span title='2021-11-22 21:58:35 -0500'>2021-11-22</span></td><td class='logsubject'><a href='/ansg191/bun/commit/packages/bun-framework-next/renderDocument.tsx?h=lithdew/picohttp-mimalloc&amp;id=25a49005cc4e8ec46323c2edb5bc2826117c5266&amp;follow=1'>another try</a></td><td><img src='//www.gravatar.com/avatar/194e8d62c38b487ec0744d2ccc0931fc?s=13&amp;d=retro' width='13' height='13' alt='Gravatar' /> Jack Hanford
</td><td>2</td><td><span class='deletions'>-5</span>/<span class='insertions'>+1</span></td></tr>
<tr class='nohover-highlight'><td/><td colspan='5' class='logmsg'>
</td></tr>
<tr class='logheader'><td><span title='2021-11-22 21:55:57 -0500'>2021-11-22</span></td><td class='logsubject'><a href='/ansg191/bun/commit/packages/bun-framework-next/libs.d.ts?h=lithdew/picohttp-mimalloc&amp;id=875b60dfe5c8eeeffee958ca474f9f6be69a342d&amp;follow=1'>add type fns</a></td><td><img src='//www.gravatar.com/avatar/194e8d62c38b487ec0744d2ccc0931fc?s=13&amp;d=retro' width='13' height='13' alt='Gravatar' /> Jack Hanford
</td><td>1</td><td><span class='deletions'>-1</span>/<span class='insertions'>+4</span></td></tr>
<tr class='nohover-highlight'><td/><td colspan='5' class='logmsg'>
</td></tr>