Skip to content

Commit

Permalink
Make <nix/unpack-channel.nix> a builtin builder
Browse files Browse the repository at this point in the history
This was the last function using a shell script, so this allows us to
get rid of tar, coreutils, bash etc.
  • Loading branch information
edolstra committed Nov 26, 2019
1 parent e60f6bd commit 045708d
Show file tree
Hide file tree
Showing 9 changed files with 55 additions and 84 deletions.
10 changes: 1 addition & 9 deletions corepkgs/config.nix.in
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
# FIXME: remove this file?
let
fromEnv = var: def:
let val = builtins.getEnv var; in
Expand All @@ -17,13 +18,4 @@ in rec {
nixLocalstateDir = "@localstatedir@";
nixSysconfDir = "@sysconfdir@";
nixStoreDir = fromEnv "NIX_STORE_DIR" "@storedir@";

# If Nix is installed in the Nix store, then automatically add it as
# a dependency to the core packages. This ensures that they work
# properly in a chroot.
chrootDeps =
if dirOf nixPrefix == builtins.storeDir then
[ (builtins.storePath nixPrefix) ]
else
[ ];
}
35 changes: 4 additions & 31 deletions corepkgs/unpack-channel.nix
Original file line number Diff line number Diff line change
@@ -1,39 +1,12 @@
with import <nix/config.nix>;

let

builder = builtins.toFile "unpack-channel.sh"
''
mkdir $out
cd $out
xzpat="\.xz\$"
gzpat="\.gz\$"
if [[ "$src" =~ $xzpat ]]; then
${xz} -d < $src | ${tar} xf - ${tarFlags}
elif [[ "$src" =~ $gzpat ]]; then
${gzip} -d < $src | ${tar} xf - ${tarFlags}
else
${bzip2} -d < $src | ${tar} xf - ${tarFlags}
fi
if [ * != $channelName ]; then
mv * $out/$channelName
fi
'';

in

{ name, channelName, src }:

derivation {
system = builtins.currentSystem;
builder = shell;
args = [ "-e" builder ];
inherit name channelName src;
builder = "builtin:unpack-channel";

system = "builtin";

PATH = "${nixBinDir}:${coreutils}";
inherit name channelName src;

# No point in doing this remotely.
preferLocalBuild = true;

inherit chrootDeps;
}
5 changes: 4 additions & 1 deletion nix-rust/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ pub extern "C" fn unpack_tarfile(source: Source, dest_dir: &str) -> bool {
for file in tar.entries().unwrap() {
let mut file = file.unwrap();

let dest_file = dest_dir.join(file.header().path().unwrap());
let dest_file = dest_dir.join(file.path().unwrap());

fs::create_dir_all(dest_file.parent().unwrap()).unwrap();

Expand All @@ -55,6 +55,9 @@ pub extern "C" fn unpack_tarfile(source: Source, dest_dir: &str) -> bool {
.unwrap();
io::copy(&mut file, &mut f).unwrap();
}
tar::EntryType::Symlink => {
std::os::unix::fs::symlink(file.header().link_name().unwrap().unwrap(), dest_file).unwrap();
}
t => panic!("Unsupported tar entry type '{:?}'.", t),
}
}
Expand Down
2 changes: 2 additions & 0 deletions src/libstore/build.cc
Original file line number Diff line number Diff line change
Expand Up @@ -3128,6 +3128,8 @@ void DerivationGoal::runChild()
builtinFetchurl(drv2, netrcData);
else if (drv->builder == "builtin:buildenv")
builtinBuildenv(drv2);
else if (drv->builder == "builtin:unpack-channel")
builtinUnpackChannel(drv2);
else
throw Error(format("unsupported builtin function '%1%'") % string(drv->builder, 8));
_exit(0);
Expand Down
1 change: 1 addition & 0 deletions src/libstore/builtins.hh
Original file line number Diff line number Diff line change
Expand Up @@ -7,5 +7,6 @@ namespace nix {
// TODO: make pluggable.
void builtinFetchurl(const BasicDerivation & drv, const std::string & netrcData);
void builtinBuildenv(const BasicDerivation & drv);
void builtinUnpackChannel(const BasicDerivation & drv);

}
39 changes: 39 additions & 0 deletions src/libstore/builtins/unpack-channel.cc
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
#include "rust.hh"
#include "builtins.hh"
#include "compression.hh"

namespace nix {

void builtinUnpackChannel(const BasicDerivation & drv)
{
auto getAttr = [&](const string & name) {
auto i = drv.env.find(name);
if (i == drv.env.end()) throw Error("attribute '%s' missing", name);
return i->second;
};

Path out = getAttr("out");
auto channelName = getAttr("channelName");
auto src = getAttr("src");

createDirs(out);

auto source = sinkToSource([&](Sink & sink) {
auto decompressor =
hasSuffix(src, ".bz2") ? makeDecompressionSink("bzip2", sink) :
hasSuffix(src, ".xz") ? makeDecompressionSink("xz", sink) :
makeDecompressionSink("none", sink);
readFile(src, *decompressor);
decompressor->finish();
});

unpack_tarfile(*source, out);

auto entries = readDirectory(out);
if (entries.size() != 1)
throw Error("channel tarball '%s' contains more than one file", src);
if (rename((out + "/" + entries[0].name).c_str(), (out + "/" + channelName).c_str()) == -1)
throw SysError("renaming channel directory");
}

}
2 changes: 1 addition & 1 deletion src/libstore/local.mk
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ libstore_DIR := $(d)

libstore_SOURCES := $(wildcard $(d)/*.cc $(d)/builtins/*.cc)

libstore_LIBS = libutil
libstore_LIBS = libutil libnixrust

libstore_LDFLAGS = $(SQLITE3_LIBS) -lbz2 $(LIBCURL_LIBS) $(SODIUM_LIBS) -pthread
ifneq ($(OS), FreeBSD)
Expand Down
41 changes: 1 addition & 40 deletions src/nix/test.cc → src/libstore/rust.hh
Original file line number Diff line number Diff line change
@@ -1,9 +1,4 @@
#include "command.hh"
#include "store-api.hh"
#include "common-args.hh"
#include "compression.hh"

using namespace nix;
#include "serialise.hh"

namespace rust {

Expand Down Expand Up @@ -47,37 +42,3 @@ struct Source
extern "C" {
bool unpack_tarfile(rust::Source source, rust::StringSlice dest_dir);
}

struct CmdTest : StoreCommand
{
CmdTest()
{
}

std::string name() override
{
return "test";
}

std::string description() override
{
return "bla bla";
}

void run(ref<Store> store) override
{
auto source = sinkToSource([&](Sink & sink) {
auto decompressor = makeDecompressionSink("bzip2", sink);
readFile("./nix-2.2.tar.bz2", *decompressor);
decompressor->finish();
});

std::string destDir = "./dest";

deletePath(destDir);

unpack_tarfile(*source, destDir);
}
};

static RegisterCommand r(make_ref<CmdTest>());
4 changes: 2 additions & 2 deletions src/nix/local.mk
Original file line number Diff line number Diff line change
Expand Up @@ -15,9 +15,9 @@ nix_SOURCES := \
$(wildcard src/nix-prefetch-url/*.cc) \
$(wildcard src/nix-store/*.cc) \

nix_LIBS = libexpr libmain libstore libutil libnixrust
nix_LIBS = libexpr libmain libstore libutil

nix_LDFLAGS = -pthread $(SODIUM_LIBS) $(EDITLINE_LIBS) $(BOOST_LDFLAGS) -lboost_context -lboost_thread -lboost_system -Lnix-rust/target/release
nix_LDFLAGS = -pthread $(SODIUM_LIBS) $(EDITLINE_LIBS) $(BOOST_LDFLAGS) -lboost_context -lboost_thread -lboost_system

$(foreach name, \
nix-build nix-channel nix-collect-garbage nix-copy-closure nix-daemon nix-env nix-hash nix-instantiate nix-prefetch-url nix-shell nix-store, \
Expand Down

0 comments on commit 045708d

Please sign in to comment.