extern crate pkg_config;

use std::fs;
use std::process::{Command, Stdio};
use std::env;
use std::path::Path;

fn main() {
    let cargo_dir = env::var("CARGO_MANIFEST_DIR").unwrap();
    let output_dir = env::var("OUT_DIR").unwrap();

    let src = Path::new(&cargo_dir[..]);
    let dst = Path::new(&output_dir[..]);

    let target = env::var("TARGET").unwrap();
    println!("target={}", target);
    match target.find("-windows-") {
        Some(..) => {
            // do not build c-code on windows, use binaries
            let prebuilt_dir = env::var("ZMQPW_DIR").unwrap();
            println!("cargo:rustc-link-search=native={}", prebuilt_dir);
            println!("cargo:rustc-flags=-L {}/lib \
                  -l zmq-pw \
                  ", prebuilt_dir);
            println!("cargo:include={}/include/zmq-pw", prebuilt_dir);
            return;
        },
        None => {}
    }

    let libs = vec!["zmq-pw"];

    let mut found = true;
    for l in libs {
        println!("searching for {}", l);
        match pkg_config::find_library(l) {
            Ok(..) => {},
            Err(..) => {
                println!("pkg-config cannot find {}", l);
                let mdata = fs::metadata(
                    &format!("{}/{}.a",
                             dst.join("pkg/lib").to_str().unwrap(),
                             l));
                match mdata {
                    Ok(md) => {
                        if !md.is_file() {
                            found = false;
                            println!("not found");
                            break;
                        }
                    },
                    Err(..) => {
                        found = false;
                        println!("error getting metadata");
                        break;
                    }
                }
            }
        }
    }

    if found {
        if target.find("darwin").is_some() {
            println!("cargo:rustc-flags=-L {}/lib \
                  -l zmq-pw \
                  -l c++ \
                  ", dst.join("pkg").display());
        }
        else {
            println!("cargo:rustc-flags=-L {}/lib \
                  -l zmq-pw \
                  -l stdc++ \
                  ", dst.join("pkg").display());
        }
        
        println!("cargo:root={}", dst.join("pkg").display());
        println!("cargo:include={}/include/zmq-pw", dst.join("pkg").display());
        return;
    }

    let root = src.join("libzmq-pw");

    let _ = fs::remove_dir_all(&dst.join("pkg"));
    let _ = fs::remove_dir_all(&dst.join("build"));
    fs::create_dir(&dst.join("build")).unwrap();

    run(Command::new("sh")
        .arg("-c")
        .arg(&format!("cd {} && \
                       cmake \
                       -DCMAKE_BUILD_TYPE=Release \
                       -DCMAKE_INSTALL_PREFIX=/ \
                       -DCMAKE_INSTALL_LIBDIR=\"/lib\" \
                       -DCMAKE_INSTALL_INCLUDEDIR=\"/include\" \
                       -DBUILD_SHARED=OFF \
                       -DBUILD_STATIC=ON \
                       -DCMAKE_POSITION_INDEPENDENT_CODE=ON {}",
                      dst.join("build").to_str().unwrap(),
                      root.as_path().to_str().unwrap()))
        .current_dir(&dst.join("build")));

    run(Command::new("make")
        .arg(&format!("-j{}", env::var("NUM_JOBS").unwrap()))
        .current_dir(&dst.join("build")));

    run(Command::new("make")
        .arg(&format!("-j{}", env::var("NUM_JOBS").unwrap()))
        .arg("install")
        .arg(&format!("DESTDIR={}", dst.join("pkg").to_str().unwrap()))
        .current_dir(&dst.join("build")));

    if target.find("darwin").is_some() {
            println!("cargo:rustc-flags=-L {}/lib \
            -l zmq-pw \
            -l c++ \
            ", dst.join("pkg").display());
    } 
    else {
        println!("cargo:rustc-flags=-L {}/lib \
            -l zmq-pw \
            -l stdc++ \
            ", dst.join("pkg").display());
    }
    
    println!("cargo:root={}", dst.join("pkg").display());
    println!("cargo:include={}/include/zmq-pw", dst.join("pkg").display());
}

fn run(cmd: &mut Command) {
    println!("running: {:?}", cmd);
    assert!(cmd.stdout(Stdio::inherit())
               .stderr(Stdio::inherit())
               .status()
               .unwrap()
               .success());
}