about summary refs log tree commit diff
path: root/scripts/nix-pull-prebuilts
blob: 9cc6683378df88722857654d5e8588a286c29d04 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
#! /usr/bin/perl -w

my $prefix = $ENV{"NIX"} || "/nix"; # !!! use prefix
my $etcdir = "$prefix/etc/nix";
my $knowns = "$prefix/var/nix/known-prebuilts";
my $tmpfile = "$prefix/var/nix/prebuilts.tmp";

my $conffile = "$etcdir/prebuilts.conf";

sub register {
    my $fn = shift;
    my $url = shift;
    return unless $fn =~ /([^\/]*)-([0-9a-z]{32})-([0-9a-z]{32})\.tar\.bz2/;
    my $id = $1;
    my $pkghash = $2;
    my $prebuilthash = $3;

    print "$pkghash => $prebuilthash ($id)\n";

    system "nix regprebuilt $pkghash $prebuilthash";
    if ($?) { die "`nix regprebuilt' failed"; }

    if ($url =~ /^\//) {
	system "nix regfile $url";
	if ($?) { die "`nix regfile' failed"; }
    } else {
	system "nix regurl $prebuilthash $url";
	if ($?) { die "`nix regurl' failed"; }
    }

    print KNOWNS "$pkghash\n";
}

open KNOWNS, ">$knowns";

open CONFFILE, "<$conffile";

while (<CONFFILE>) {
    chomp;
    if (/^\s*(\S+)\s*(\#.*)?$/) {
        my $url = $1;

        print "obtaining prebuilt list from $url...\n";

        if ($url =~ /^\//) {

            # It's a local path.

            foreach my $fn (glob "$url/*") {
                register($fn, $fn);
            }

        } else {

            # It's a URL.

            system "wget '$url' -O '$tmpfile' 2> /dev/null"; # !!! escape
            if ($?) { die "`wget' failed"; }

            open INDEX, "<$tmpfile";

            while (<INDEX>) {
                # Get all links to prebuilts, that is, file names of the
                # form foo-HASH-HASH.tar.bz2.
                next unless (/HREF=\"([^\"]*)\"/);
                my $fn = $1;
                next if $fn =~ /\.\./;
                next if $fn =~ /\//;
                register($fn, "$url/$fn");
            }

            close INDEX;

            unlink $tmpfile;
        }
    }
}

close CONFFILE;

close KNOWNS;