root/branches/dsss-smi/updateSources.d

/**
 * Update the sources list from the given path, specifically modifying the
 * given tool
 * 
 * Authors:
 *  Gregor Richards
 * 
 * License:
 *  Copyright (c) 2006  Gregor Richards
 *  
 *  Permission is hereby granted, free of charge, to any person obtaining a
 *  copy of this software and associated documentation files (the "Software"),
 *  to deal in the Software without restriction, including without limitation
 *  the rights to use, copy, modify, merge, publish, distribute, sublicense,
 *  and/or sell copies of the Software, and to permit persons to whom the
 *  Software is furnished to do so, subject to the following conditions:
 *  
 *  The above copyright notice and this permission notice shall be included in
 *  all copies or substantial portions of the Software.
 *  
 *  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 *  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 *  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 *  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 *  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 *  FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 *  DEALINGS IN THE SOFTWARE.
 */

module updateSources;

import std.file;
import std.path;
import std.process;
import std.stdio;
import std.string;

version (Unix) {
    import std.c.unix.unix;
} else version (linux) {
    import std.c.linux.linux;
} else static assert(0);

alias std.file.chdir chdir;
alias std.file.getcwd getcwd;

extern (C) int daemon(int, int);
extern (C) int geteuid();
extern (C) int setreuid(int, int);

int main(char[][] args)
{
    // args: directory tool
    if (args.length != 3) {
        writefln("Improper invocation.");
        return 1;
    }

    char[] srcDir = args[1];
    char[] updPkg = args[2];
    
    // get our correct home directory
    passwd *pwbuf;
    pwbuf = getpwuid(geteuid());
    if (pwbuf is null) {
        writefln("Failed to get home directory");
        return 1;
    }
    chdir(toString(pwbuf.pw_dir));
    setreuid(geteuid(), -1);
    
    // 1) sanitize
    if (updPkg[0] == '.' ||
        find(updPkg, '/') != -1) {
        writefln("Invalid package name.");
        return 1;
    }
    
    // 2) daemonize (if possible, ignore error but will always be running /something/)
    daemon(1, 0);
    
    // 3) make sure nothing is out of date
    system("svn up sources");
    
    // 4) remove old mirrored cruft
    system("rm -rf mirror/" ~ updPkg ~ "*/");
    
    // 5) Update the source.list
    char[] sourceList;
    foreach (pkg; listdir(srcDir)) {
        if (pkg.length < 1 ||
            pkg[0] == '.' ||
            pkg == "privileges") continue;
        
        // add it
        sourceList ~= cast(char[]) std.file.read(
            srcDir ~ "/" ~ pkg) ~ "\n";
    }

    char[] origcwd = getcwd();
    chdir("sources");
    std.file.write("source.list", sourceList);
    
    // 6) Update pkgs.list (and in turn the mirror)
    system("./pkgslist");
    chdir(origcwd);
    
    // 7) svn commit everything
    system("svn commit sources -m ''");
    system("./up.sh");
    
    return 0;
}