src/download.sh

2 files changed, 86 insertions, 0 deletions

diff --git a/src/download.sh b/src/download.sh
new file mode 100644
index 0000000..a95a94b
--- /dev/null
+++ b/src/download.sh
@@ -0,0 +1,85 @@
+# Downloading and temporary file management
+#
+# Copyright (C) 2015  Patrick "P. J." McDermott
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+temp_dir=''
+
+init_temp_dir()
+{
+	local tmpdir=''
+
+	if [ "x${TMPDIR:+set}" = 'xset' ]; then
+		tmpdir="${TMPDIR}"
+	else
+		tmpdir='/tmp'
+	fi
+
+	rand
+	temp_dir="${tmpdir}/firman-$(printf '%010d' ${rand_x})"
+	if ! mkdir "${temp_dir}"; then
+		err 'Failed to create temporary directory'
+		return 1
+	fi
+
+	return 0
+}
+
+fini_temp_dir()
+{
+	if ! rmdir "${temp_dir}"; then
+		err 'Failed to remove temporary directory'
+		return 1
+	fi
+
+	return 0
+}
+
+dl()
+{
+	local url="${1}"
+	shift 1
+
+	if ! ${WGET} -q -O "${temp_dir}/${file}" "${url}"; then
+		err 'Failed to download "%s"' "${url}"
+		return 1
+	fi
+
+	return 0
+}
+
+dl_temp()
+{
+	local url="${1}"
+	local file="${2}"
+	shift 2
+
+	if ! ${WGET} -q -O "${temp_dir}/${file}" "${url}"; then
+		err 'Failed to download "%s"' "${url}"
+		return 1
+	fi
+
+	return 0
+}
+
+rm_temp()
+{
+	local file="${1}"
+	shift 1
+
+	rm -f "${temp_dir}/${file}"
+
+	return 0
+}
diff --git a/src/local.mk b/src/local.mk
index 0734fa7..72b52b6 100644
--- a/src/local.mk
+++ b/src/local.mk
@@ -1,6 +1,7 @@
 firman_SOURCES += \
 	src/board.sh \
 	src/dist.sh \
+	src/download.sh \
 	src/flashrom.sh \
 	src/main.sh \
 	src/rand.sh \