aboutsummaryrefslogtreecommitdiff
blob: 0691925fe05846fc10db9868b81583f861a3d40b (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
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
# -*-eselect-*-  vim: ft=eselect
# Copyright 2014 Gentoo Foundation
# Distributed under the terms of the GNU GPL version 2 or later

DESCRIPTION="Manage the Rust compiler versions"
MAINTAINER="jauhien@gentoo.org"
VERSION="0.2"

inherit package-manager path-manipulation

SITELISP=/usr/share/emacs/site-lisp
SITEVIM=/usr/share/vim/vimfiles
SITEZSH=/usr/share/zsh/site-functions

# find a list of installed rust compilers
# each compiler provider should install
# a config file named provider-$pkgname-$pkgver
# in "${EROOT}"/etc/env.d/rust directory
# this function prints list of $pkgname-$pkgver values
find_targets() {
	local f
	local -a providers
	for f in "${EROOT}"/etc/env.d/rust/provider-*; do
		[[ -f ${f} ]] || continue
		providers=("${providers[@]}" "${f##*/provider-}")
	done
	echo "${providers[@]}"
}

#get rustc postfix
get_postfix() {
	local target=$1
	echo "${target}" | cut -d- -f2-
}

# remove symlink if exists
remove_symlink() {
	local symlink=$1

	if [[ -L ${symlink} ]]; then
		# existing symlink
		rm ${symlink} || die -q "Couldn't remove existing symlink ${symlink}"
	elif [[ -e ${symlink} ]]; then
		# we have something strange
		die -q "${symlink} exists but is not a symlink"
	fi
}


# set symlink if source exists
set_symlink() {
	local source=$1
	local dest=$2

	if [[ -e ${source} ]]; then
		mkdir -p "$(dirname ${dest})" || die -q "directory creation failed for $(dirname ${dest})"
		ln -s "${source}" "${dest}" || die -q "${dest} symlink setting failed"
	else
		false
	fi
}

# unset the rust version
unset_version() {
	remove_symlink "${EROOT}"/usr/bin/rustc
	remove_symlink "${EROOT}"/usr/bin/rustdoc
}

# set the rust version
# each compiler provider should install
# files named rustc-$postfix and rustdoc-$postfix
# in ${EROOT}/usr/bin directory
# $postfix is defined as the part of $pkgname-$pkgver after the first -
# for dev-lang/rust-bin-9999 ebuild it would be bin-9999
set_version() {
	local target=$1

	if is_number "${target}"; then
		local targets=( $(find_targets) )
		target=${targets[target-1]}
	fi

	target=$(get_postfix ${target})

	[[ -z ${target} || ! -x "${EROOT}/usr/bin/rustc-${target}" ]] \
		&& die -q "Target \"$1\" doesn't appear to be valid!"

	unset_version

	set_symlink "${EROOT}/usr/bin/rustc-${target}" "${EROOT}/usr/bin/rustc" || die -q "rustc symlink setting failed"
	set_symlink "${EROOT}/usr/bin/rustdoc-${target}" "${EROOT}/usr/bin/rustdoc" || die -q "rustdoc symlink setting failed"
}

### list action ###

describe_list() {
	echo "List available Rust versions"
}

do_list() {
	local i targets=( $(find_targets) )

	write_list_start "Available Rust versions:"
    for (( i = 0; i < ${#targets[@]}; i++ )); do
        # highlight the target where the symlink is pointing to
        [[ rustc-$(get_postfix ${targets[i]}) = \
            $(basename "$(canonicalise "${EROOT}/usr/bin/rustc")") ]] \
            && targets[i]=$(highlight_marker "${targets[i]}")
    done
    write_numbered_list -m "(none found)" "${targets[@]}"
}

### set action ###

describe_set() {
	echo "Set active Rust version"
}

describe_set_parameters() {
    echo "<target>"
}

describe_set_options() {
    echo "target : Target number (from 'list' action)"
}

do_set() {
    [[ -z $1 ]] && die -q "You didn't tell me what to set the version to"
    [[ $# -gt 1 ]] && die -q "Too many parameters"

	set_version "$1" || die -q "Couldn't set new active version"
}

### update action ###

describe_update() {
	echo "Switch to the most recent Rust compiler"
}

describe_update_options() {
	echo "--if-unset    : Do not override existing implementation"
}

do_update() {
	local if_unset="0"
	while [[ $# > 0 ]]; do
		case "$1" in
			--if-unset)
				if_unset="1"
				;;
			*)
				die -q "Unrecognized argument '$1'"
				;;
		esac
		shift
	done

	if [[ "${if_unset}" == "1" && -f "${EROOT}"/usr/bin/rustc ]]; then
		return
	fi

	local targets=( $(find_targets) )
	do_set ${#targets[@]}
}

### unset action ###

describe_unset() {
	echo "Unset active Rust version"
}

describe_unset_options() {
	echo "--if-invalid    : Unset only if symlink is invalid (e.g. package was uninstalled)"
}

do_unset() {
	local if_invalid="0"
	while [[ $# > 0 ]]; do
		case "$1" in
			--if-invalid)
				if_invalid="1"
				;;
			*)
				die -q "Unrecognized argument '$1'"
				;;
		esac
		shift
	done

	if [[ "${if_invalid}" == "1" && -e "${EROOT}"/usr/bin/rustc ]]; then
		return
	fi

	unset_version || die -q "Couldn't unset active version"
}