diff --git a/libexec/swiftenv-prefix b/libexec/swiftenv-prefix index 136895f..bcf95b5 100755 --- a/libexec/swiftenv-prefix +++ b/libexec/swiftenv-prefix @@ -25,18 +25,24 @@ if [ -d "$SWIFTENV_ROOT/versions/swift-$SWIFT_VERSION" ]; then fi # OS X Swift Toolchain -TOOLCHAIN_DIR="/Library/Developer/Toolchains" -if [ -d "$TOOLCHAIN_DIR/swift-$SWIFT_VERSION.xctoolchain" ]; then - echo "$TOOLCHAIN_DIR/swift-$SWIFT_VERSION.xctoolchain" - exit -fi -# OS X User Swift Toolchain -TOOLCHAIN_DIR="$HOME/Library/Developer/Toolchains" -if [ -d "$TOOLCHAIN_DIR/swift-$SWIFT_VERSION.xctoolchain" ]; then - echo "$TOOLCHAIN_DIR/swift-$SWIFT_VERSION.xctoolchain" - exit -fi +check_toolchain() { + local TOOLCHAIN_DIR + TOOLCHAIN_DIR=$1 + + if [ -d "$TOOLCHAIN_DIR/swift-$SWIFT_VERSION-RELEASE.xctoolchain" ]; then + echo "$TOOLCHAIN_DIR/swift-$SWIFT_VERSION-RELEASE.xctoolchain" + exit + fi + + if [ -d "$TOOLCHAIN_DIR/swift-$SWIFT_VERSION.xctoolchain" ]; then + echo "$TOOLCHAIN_DIR/swift-$SWIFT_VERSION.xctoolchain" + exit + fi +} + +check_toolchain "$HOME/Library/Developer/Toolchains" +check_toolchain "/Library/Developer/Toolchains" # OS X Xcode Swift's if command -v "mdfind" >/dev/null 2>&1; then diff --git a/libexec/swiftenv-versions b/libexec/swiftenv-versions index 4e9e4c5..61323d3 100755 --- a/libexec/swiftenv-versions +++ b/libexec/swiftenv-versions @@ -61,7 +61,7 @@ check_toolchains() { if [[ "$path" == *".xctoolchain" ]] && [ -d "$path" ]; then version="$(basename ${path##*/} .xctoolchain)" if [ "$version" != "swift-latest" ]; then - print_version "$version" "toolchain" + print_version "${version%-RELEASE}" "toolchain" fi fi done