src/librustc_incremental/persist/util.rs

// Copyright 2014 The Rust Project Developers. See the COPYRIGHT
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

use rustc::middle::cstore::LOCAL_CRATE;
use rustc::session::Session;
use rustc::ty::TyCtxt;

use std::fs;
use std::io;
use std::path::{Path, PathBuf};
use syntax::ast;

pub fn dep_graph_path(tcx: TyCtxt) -> Option<PathBuf> {
    tcx_path(tcx, LOCAL_CRATE, "local")
}

pub fn metadata_hash_path(tcx: TyCtxt, cnum: ast::CrateNum) -> Option<PathBuf> {
    tcx_path(tcx, cnum, "metadata")
}

pub fn tcx_work_products_path(tcx: TyCtxt) -> Option<PathBuf> {
    let crate_name = tcx.crate_name(LOCAL_CRATE);
    sess_work_products_path(tcx.sess, &crate_name)
}

pub fn sess_work_products_path(sess: &Session,
                               local_crate_name: &str)
                               -> Option<PathBuf> {
    let crate_disambiguator = sess.local_crate_disambiguator();
    path(sess, local_crate_name, &crate_disambiguator, "work-products")
}

pub fn in_incr_comp_dir(sess: &Session, file_name: &str) -> Option<PathBuf> {
    sess.opts.incremental.as_ref().map(|incr_dir| incr_dir.join(file_name))
}

fn tcx_path(tcx: TyCtxt,
            cnum: ast::CrateNum,
            middle: &str)
            -> Option<PathBuf> {
    path(tcx.sess, &tcx.crate_name(cnum), &tcx.crate_disambiguator(cnum), middle)
}

fn path(sess: &Session,
        crate_name: &str,
        crate_disambiguator: &str,
        middle: &str)
        -> Option<PathBuf> {
    // For now, just save/load dep-graph from
    // directory/dep_graph.rbml
    sess.opts.incremental.as_ref().and_then(|incr_dir| {
        match create_dir_racy(&incr_dir) {
            Ok(()) => {}
            Err(err) => {
                sess.err(
                    &format!("could not create the directory `{}`: {}",
                             incr_dir.display(), err));
                return None;
            }
        }

        let file_name = format!("{}-{}.{}.bin", crate_name, crate_disambiguator, middle);

        Some(incr_dir.join(file_name))
    })
}

// Like std::fs::create_dir_all, except handles concurrent calls among multiple
// threads or processes.
fn create_dir_racy(path: &Path) -> io::Result<()> {
    match fs::create_dir(path) {
        Ok(()) => return Ok(()),
        Err(ref e) if e.kind() == io::ErrorKind::AlreadyExists => return Ok(()),
        Err(ref e) if e.kind() == io::ErrorKind::NotFound => {}
        Err(e) => return Err(e),
    }
    match path.parent() {
        Some(p) => try!(create_dir_racy(p)),
        None => return Err(io::Error::new(io::ErrorKind::Other,
                                          "failed to create whole tree")),
    }
    match fs::create_dir(path) {
        Ok(()) => Ok(()),
        Err(ref e) if e.kind() == io::ErrorKind::AlreadyExists => Ok(()),
        Err(e) => Err(e),
    }
}