86 lines
		
	
	
	
		
			2.4 KiB
		
	
	
	
		
			Perl
		
	
	
		
			Executable file
		
	
	
	
	
			
		
		
	
	
			86 lines
		
	
	
	
		
			2.4 KiB
		
	
	
	
		
			Perl
		
	
	
		
			Executable file
		
	
	
	
	
| #! /usr/bin/perl -w
 | |
| 
 | |
| use strict;
 | |
| 
 | |
| my $keep = 0;
 | |
| my $sourceroot = 1;
 | |
| my $name = "current";
 | |
| my $srcid;
 | |
| 
 | |
| my $argnr = 0;
 | |
| while ($argnr < scalar @ARGV) {
 | |
|     my $arg = $ARGV[$argnr++];
 | |
|     if ($arg eq "--keep") { $keep = 1; }
 | |
|     elsif ($arg eq "--no-source") { $sourceroot = 0; }
 | |
|     elsif ($arg eq "--name") { $name = $ARGV[$argnr++]; }
 | |
|     elsif ($arg =~ /^\//) { $srcid = $arg; }
 | |
|     else { die "unknown argument `$arg'" };
 | |
| }
 | |
| 
 | |
| my $linkdir = "@localstatedir@/nix/links";
 | |
| 
 | |
| # Build the specified package, and all its dependencies.
 | |
| my $nfid = `nix --install $srcid`;
 | |
| if ($?) { die "`nix --install' failed"; }
 | |
| chomp $nfid;
 | |
| die unless $nfid =~ /^\//;
 | |
| 
 | |
| my $pkgdir = `nix --query --list $nfid`;
 | |
| if ($?) { die "`nix --query --list' failed"; }
 | |
| chomp $pkgdir;
 | |
| 
 | |
| # Figure out a generation number.
 | |
| opendir(DIR, $linkdir);
 | |
| my $nr = 0;
 | |
| foreach my $n (sort(readdir(DIR))) {
 | |
|     next if (!($n =~ /^\d+$/));
 | |
|     $nr = $n + 1 if ($n >= $nr);
 | |
| }
 | |
| closedir(DIR);
 | |
| 
 | |
| my $link = "$linkdir/$nr";
 | |
| 
 | |
| # Create a symlink from $link to $pkgdir.
 | |
| symlink($pkgdir, $link) or die "cannot create $link: $!";
 | |
| 
 | |
| # Store the id of the normal form.  This is useful for garbage
 | |
| # collection and the like.
 | |
| my $idfile = "$linkdir/$nr.id";
 | |
| open ID, "> $idfile" or die "cannot create $idfile";
 | |
| print ID "$nfid\n";
 | |
| close ID;
 | |
| 
 | |
| # Optionally store the source id.
 | |
| if ($sourceroot) {
 | |
|     $idfile = "$linkdir/$nr-src.id";
 | |
|     open ID, "> $idfile" or die "cannot create $idfile";
 | |
|     print ID "$srcid\n";
 | |
|     close ID;
 | |
| }
 | |
| 
 | |
| my $current = "$linkdir/$name";
 | |
| 
 | |
| # Read the current generation so that we can delete it (if --keep
 | |
| # wasn't specified).
 | |
| my $oldlink = readlink($current);
 | |
| 
 | |
| # Make $link the current generation by pointing $linkdir/current to
 | |
| # it.  The rename() system call is supposed to be essentially atomic
 | |
| # on Unix.  That is, if we have links `current -> X' and `new_current
 | |
| # -> Y', and we rename new_current to current, a process accessing
 | |
| # current will see X or Y, but never a file-not-found or other error
 | |
| # condition.  This is sufficient to atomically switch the current link
 | |
| # tree.
 | |
| 
 | |
| print "switching $current to $link\n";
 | |
| 
 | |
| my $tmplink = "$linkdir/.new_$name";
 | |
| symlink($link, $tmplink) or die "cannot create $tmplink";
 | |
| rename($tmplink, $current) or die "cannot rename $tmplink";
 | |
| 
 | |
| if (!$keep && defined $oldlink) {
 | |
|     print "deleting old $oldlink\n";
 | |
|     unlink($oldlink) == 1 or print "cannot delete $oldlink\n";
 | |
|     unlink("$oldlink.id") == 1 or print "cannot delete $oldlink.id\n";
 | |
|     unlink("$oldlink-src.id");
 | |
| }
 |