src/temp.sh: New file

2 files changed, 55 insertions, 0 deletions

diff --git a/src/local.mk b/src/local.mk
index 0ce1d77..0734fa7 100644
--- a/src/local.mk
+++ b/src/local.mk
@@ -4,6 +4,7 @@ firman_SOURCES += \
 	src/flashrom.sh \
 	src/main.sh \
 	src/rand.sh \
+	src/temp.sh \
 	src/ui.sh
 
 include $(top_srcdir)/src/action/local.mk
diff --git a/src/temp.sh b/src/temp.sh
new file mode 100644
index 0000000..2e6885d
--- /dev/null
+++ b/src/temp.sh
@@ -0,0 +1,54 @@
+# 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=''
+
+temp_init()
+{
+	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
+}
+
+temp_fini()
+{
+	if ! rmdir "${temp_dir}"; then
+		err 'Failed to remove temporary directory'
+		return 1
+	fi
+
+	return 0
+}
+
+get_temp_dir()
+{
+	printf '%s' "${temp_dir}"
+	return 0
+}