summaryrefslogtreecommitdiff
path: root/src/get.sh
blob: b99340f01376f545f7e428a391cc1e8e4a273261 (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
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
#!/bin/sh


# list all direct dependencies of a package
#
list_deps() {
    local name=$1
    local tree_file="${DEP_DIR}/$name"
    [ -f $tree_file ] &&
        for dep in $(cat "$tree_file"); do
            echo $dep
        done | sort -u
}

# list all dependencies and sub dependencies of a package
#
resolve_deps () {
    local out="${CACHE_DIR}/deps"
    local deps=""
    local i=0

    if ! ${RESOLVE_DEPS}; then
        echo $@ > $out
        return 0
    fi

    hbar
    while [ "$#" != "0" ]; do

        # pop a value from the args
        local package=$1

        #only add if not already added
        if ! echo ${deps} | grep -q "\b$package\b"; then
            deps="$deps $package"
            i=$((i+1))
        fi

        for dep in $(list_deps $package); do
            # if not already checked
            if echo $@ | grep -qv "\b$dep\b"; then
                set -- $@ $dep
            fi
        done

        shift

        ${QUIET} || hbar -T "${CHECKMARK} resolving dependencies" $i $((i + $#))
    done

    ${QUIET} || hbar -t ${HBAR_COMPLETE} -T "${CHECKMARK} resolved dependencies" $i $((i + $#))
    echo ${deps} > $out
}

get_package_download_info() {
    sed 1q ${PACKAGES_DIR}/*/$1
}

is_installed() {
    [ -f "${INSTALLED_DIR}/$1/checksum" ]
}

get_installed_version () {
    local name=$1
    local file="${INSTALLED_DIR}/$name/checksum"
    [ -f $file ] &&
        cat $file
}

# bad implementation
package_exists () {
    [ "$(find ${PACKAGES_DIR} -mindepth 2 -name "$1" | wc -l)" != "0" ]
}

download_package () {
    local package=$1
    local output=$2

    local info=$(get_package_download_info $package)
    set -- $info

    local url=$1
    local checksum=$2

    local output_info="${output}.info"

    if validate_checksum $output $checksum; then
        ${VERBOSE} && printf "${LIGHT_BLACK}skipping download for %s already exists with checksum %s${RESET}\n" $package $checksum
        return 0
    fi
    ${VERBOSE} && printf "${LIGHT_BLACK}downloading $package from $url\n" $package $checksum
    touch $output

    (curl ${CURL_OPTS} -o "$output_info" "$url.info" 2>> ${LOG_FILE} || printf "${RED}Failed to download info for %s\n" $package) &
    (curl ${CURL_OPTS} -o "$output" "$url" 2>> ${LOG_FILE} || printf "${RED}Failed to download %s\n" $package) &
}

download_packages () {
    local total_download=$1; shift
    local packages=$@
    local outputs=""

    local out_dir="${PACKAGE_CACHE}"
    mkdir -p "$out_dir"

    for package in $packages; do 
        local output="${out_dir}/${checksum}.${package}.xipkg"
        download_package $package $output
        outputs="$outputs $output"
    done

    wait_for_download $total_download ${outputs}

    set -- $outputs

    local i=0
    ${UNSAFE} || for pkg_file in ${outputs}; do 

        ${QUIET} || hbar -T "${LARGE_CIRCLE} validating downloads..." $i $#

        info_file="${pkg_file}.info"
        if ! validate_sig $pkg_file $info_file; then
            printf "${RED}Failed to verify signature for ${LIGHT_RED}%s${RED}\n" $(basename $pkg_file .xipkg)
            mv "$pkg_file" "${pkg_file}.invalid"
        else
            i=$((i+1))
        fi
    done &&
    ${QUIET} || hbar -t ${HBAR_COMPLETE} -T "${CHECKMARK} validated downloads" $i $#

    install $@
}

get () {
    local requested=$@
    local missing="" already="" install="" update="" urls=""
    local total_download=0
    local out="${CACHE_DIR}/deps"

    touch $out
    resolve_deps $@

    for package in $(cat $out); do
        if ! package_exists $package; then
            missing="$missing $package"
            continue
        fi

        set -- $(get_package_download_info $package)
        checksum=$2
        size=$3
        
        if ! is_installed $package; then
            install="$install $package"
            total_download=$((total_download+size))
            continue
        fi

        if [ "$(get_installed_version $package)" = "$checksum" ]; then
            already="$already $package"
            continue
        fi
        update="$update $package"
        total_download=$((total_download+size))
    done

    ${QUIET} || {
        [ "${missing}" ] && 
            printf "${LIGHT_RED}The following packages could not be located:${RED} $missing\n${RESET}"

        [ "${update}" ] &&
            printf "${LIGHT_GREEN}The following packages will be updated:\n\t${GREEN}$update\n${RESET}"

        [ "${install}" ] &&
            printf "${LIGHT_BLUE}The following packages will be installed:\n\t${BLUE}$install\n${RESET}"

        [ ! "${install}" ] && [ ! "${update}" ] && [ "${already}" ] &&
            printf "${LIGHT_WHITE}The following packages are already up to date:\n\t${WHITE}$already\n${RESET}"
    }

    [ "${#install}" = "0" ] && [ "${#update}" = 0 ] && {
        printf "${LIGHT_RED}Nothing to do!\n"
        return 0
    }
    
    ${QUIET} || {
        [ "${SYSROOT}" = "/" ] || printf "${WHITE}To install to ${LIGHT_WHITE}${SYSROOT}${RESET}\n"
        printf "${WHITE}Total download size:${LIGHT_WHITE} $(format_bytes $total_download)\n"
    }

    prompt_question "${WHITE}Continue?" &&
    download_packages $total_download ${install} ${update} || {
        ${QUIET} || printf "${RED}Action canceled by user\n"
    }
}

fetch () {
    local packages=$@
    local outputs=""

    local total_download=0
    for package in $packages; do 
        if package_exists $package; then
            set -- $(get_package_download_info $package)
            size=$3
            total_download=$((total_download+size))

            local output="${package}.xipkg"
            download_package $package $output
            outputs="$outputs $output"
        fi
    done

    wait_for_download $total_download ${outputs}
}