about summary refs log tree commit diff
path: root/src/rustc/util/filesearch.rs
blob: 75327597ce92959a4d684f828aacae7b2df7a46d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
// A module for searching for libraries
// FIXME: I'm not happy how this module turned out. Should probably
// just be folded into cstore.

import result::result;
export filesearch;
export mk_filesearch;
export pick;
export pick_file;
export search;
export relative_target_lib_path;
export get_cargo_sysroot;
export get_cargo_root;
export get_cargo_root_nearest;
export libdir;

import path::path;

type pick<T> = fn(path: path) -> option<T>;

fn pick_file(file: path, path: path) -> option<path> {
    if path::basename(path) == file { option::some(path) }
    else { option::none }
}

iface filesearch {
    fn sysroot() -> path;
    fn lib_search_paths() -> [path];
    fn get_target_lib_path() -> path;
    fn get_target_lib_file_path(file: path) -> path;
}

fn mk_filesearch(maybe_sysroot: option<path>,
                 target_triple: str,
                 addl_lib_search_paths: [path]) -> filesearch {
    type filesearch_impl = {sysroot: path,
                            addl_lib_search_paths: [path],
                            target_triple: str};
    impl of filesearch for filesearch_impl {
        fn sysroot() -> path { self.sysroot }
        fn lib_search_paths() -> [path] {
            self.addl_lib_search_paths
                + [make_target_lib_path(self.sysroot, self.target_triple)]
                + alt get_cargo_lib_path_nearest() {
                  result::ok(p) { [p] }
                  result::err(p) { [] }
                }
                + alt get_cargo_lib_path() {
                  result::ok(p) { [p] }
                  result::err(p) { [] }
                }
        }
        fn get_target_lib_path() -> path {
            make_target_lib_path(self.sysroot, self.target_triple)
        }
        fn get_target_lib_file_path(file: path) -> path {
            path::connect(self.get_target_lib_path(), file)
        }
    }

    let sysroot = get_sysroot(maybe_sysroot);
    #debug("using sysroot = %s", sysroot);
    {sysroot: sysroot,
     addl_lib_search_paths: addl_lib_search_paths,
     target_triple: target_triple} as filesearch
}

// FIXME #1001: This can't be an obj method
fn search<T: copy>(filesearch: filesearch, pick: pick<T>) -> option<T> {
    for lib_search_path in filesearch.lib_search_paths() {
        #debug("searching %s", lib_search_path);
        for path in os::list_dir_path(lib_search_path) {
            #debug("testing %s", path);
            let maybe_picked = pick(path);
            if option::is_some(maybe_picked) {
                #debug("picked %s", path);
                ret maybe_picked;
            } else {
                #debug("rejected %s", path);
            }
        }
    }
    ret option::none;
}

fn relative_target_lib_path(target_triple: str) -> [path] {
    [libdir(), "rustc", target_triple, libdir()]
}

fn make_target_lib_path(sysroot: path,
                        target_triple: str) -> path {
    let path = [sysroot] + relative_target_lib_path(target_triple);
    let path = path::connect_many(path);
    ret path;
}

fn get_default_sysroot() -> path {
    alt os::self_exe_path() {
      option::some(p) { path::normalize(path::connect(p, "..")) }
      option::none {
        fail "can't determine value for sysroot";
      }
    }
}

fn get_sysroot(maybe_sysroot: option<path>) -> path {
    alt maybe_sysroot {
      option::some(sr) { sr }
      option::none { get_default_sysroot() }
    }
}

fn get_cargo_sysroot() -> result<path, str> {
    let path = [get_default_sysroot(), libdir(), "cargo"];
    result::ok(path::connect_many(path))
}

fn get_cargo_root() -> result<path, str> {
    alt os::getenv("CARGO_ROOT") {
        some(_p) { result::ok(_p) }
        none {
          alt os::homedir() {
            some(_q) { result::ok(path::connect(_q, ".cargo")) }
            none { result::err("no CARGO_ROOT or home directory") }
          }
        }
    }
}

fn get_cargo_root_nearest() -> result<path, str> {
    result::chain(get_cargo_root()) { |p|
        let cwd = os::getcwd();
        let mut dirname = path::dirname(cwd);
        let mut dirpath = path::split(dirname);
        let cwd_cargo = path::connect(cwd, ".cargo");
        let mut par_cargo = path::connect(dirname, ".cargo");
        let mut rslt = result::ok(cwd_cargo);

        if !os::path_is_dir(cwd_cargo) && cwd_cargo != p {
            while vec::is_not_empty(dirpath) && par_cargo != p {
                if os::path_is_dir(par_cargo) {
                    rslt = result::ok(par_cargo);
                    break;
                }
                vec::pop(dirpath);
                dirname = path::dirname(dirname);
                par_cargo = path::connect(dirname, ".cargo");
            }
        }
        rslt
    }
}

fn get_cargo_lib_path() -> result<path, str> {
    result::chain(get_cargo_root()) { |p|
        result::ok(path::connect(p, libdir()))
    }
}

fn get_cargo_lib_path_nearest() -> result<path, str> {
    result::chain(get_cargo_root_nearest()) { |p|
        result::ok(path::connect(p, libdir()))
    }
}

// The name of the directory rustc expects libraries to be located.
// On Unix should be "lib", on windows "bin"
fn libdir() -> str {
   let libdir = #env("CFG_LIBDIR");
   if str::is_empty(libdir) {
      fail "rustc compiled without CFG_LIBDIR environment variable";
   }
   libdir
}