wgtpkg-unit: install/uninstall functions added

Change-Id: Ib147de12d0b2c8e783ddb6aae2f9978f288d98fd
Signed-off-by: José Bollo <jose.bollo@iot.bzh>

2 files changed, 129 insertions, 1 deletions

diff --git a/src/wgtpkg-unit.c b/src/wgtpkg-unit.c
index 26cf324..93cfe64 100644
--- a/src/wgtpkg-unit.c
+++ b/src/wgtpkg-unit.c
@@ -25,13 +25,23 @@
 #include <unistd.h>
 #include <string.h>
 #include <errno.h>
+#include <limits.h>
+
+#include <json-c/json.h>
  
 #include "verbose.h"
 #include "utils-file.h"
 
 #include "wgtpkg-mustach.h"
+#include "utils-json.h"
+#include "wgt-json.h"
+
 #include "wgtpkg-unit.h"
 
+#if !defined(SYSTEMD_UNITS_ROOT)
+# define SYSTEMD_UNITS_ROOT "/usr/local/lib/systemd"
+#endif
+
 #if 0
 #include <ctype.h>
 #else
@@ -299,7 +309,7 @@ void unit_generator_off()
 }
 
 /*
- * Initialises the unit generator with 'filename'.
+ * Initialises the unit generator with the content of the file of path 'filename'.
  * Returns 0 in case of success or a negative number in case of error.
  */
 int unit_generator_on(const char *filename)
@@ -345,3 +355,117 @@ int unit_generator_process(struct json_object *jdesc, int (*process)(void *closu
 	return rc;
 }
 
+/**************** SPECIALIZED PART *****************************/
+
+
+static int get_unit_path(char *path, size_t pathlen, const struct unitdesc *desc)
+{
+	int rc;
+
+	if (desc->scope == unitscope_unknown || desc->type == unittype_unknown || desc->name == NULL) {
+		if (desc->scope == unitscope_unknown)
+			ERROR("unit of unknown scope");
+		if (desc->type == unittype_unknown)
+			ERROR("unit of unknown type");
+		if (desc->name == NULL)
+			ERROR("unit of unknown name");
+		errno = EINVAL;
+		rc = -1;
+	}
+	else {
+		rc = snprintf(path, pathlen, "%s/%s/%s.%s", 
+				SYSTEMD_UNITS_ROOT,
+				desc->scope == unitscope_system ? "system" : "user",
+				desc->name,
+				desc->type == unittype_socket ? "socket" : "service");
+
+		if (rc >= 0 && (size_t)rc >= pathlen) {
+			ERROR("can't set the unit name");
+			errno = EINVAL;
+			rc = -1;
+		}
+	}
+	return rc;
+}
+
+static int do_install_units(void *closure, const struct unitdesc descs[], unsigned count)
+{
+	int rc;
+	unsigned i;
+	char path[PATH_MAX + 1];
+
+	for (i = 0 ; i < count ; i++) {
+		rc = get_unit_path(path, sizeof path, &descs[i]);
+		if (rc >= 0) {
+			rc = putfile(path, descs[i].content, descs[i].content_length);
+		}
+	}
+	return 0;
+}
+
+static int do_uninstall_units(void *closure, const struct unitdesc descs[], unsigned count)
+{
+	int rc;
+	unsigned i;
+	char path[PATH_MAX];
+
+	for (i = 0 ; i < count ; i++) {
+		rc = get_unit_path(path, sizeof path, &descs[i]);
+		if (rc >= 0) {
+			rc = unlink(path);
+		}
+	}
+	return 0;
+}
+
+static int add_metadata(struct json_object *jdesc, const char *installdir, const char *icondir, int port)
+{
+	char portstr[30];
+
+	sprintf(portstr, "%d", port);
+	return 	j_add_many_strings_m(jdesc,
+		"#metadata.install-dir", installdir,
+		"#metadata.app-data-dir", "%h/app-data",
+		"#metadata.icons-dir", icondir,
+		"#metadata.http-port", portstr,
+		NULL) ? 0 : -1;
+}
+
+static int do_install_uninstall(
+		struct wgt_info *ifo,
+		const char *installdir,
+		const char *icondir,
+		int port,
+		int (*doer)(void *, const struct unitdesc[], unsigned)
+)
+{
+	int rc;
+	struct json_object *jdesc;
+
+	jdesc = wgt_info_to_json(ifo);
+	if (!jdesc)
+		rc = -1;
+	else {
+		rc = add_metadata(jdesc, installdir, icondir, port);
+		if (rc)
+			ERROR("can't set the metadata. %m");
+		else {
+			rc = unit_generator_process(jdesc, doer, NULL);
+			if (rc)
+				ERROR("can't install units, error %d", rc);
+		}
+		json_object_put(jdesc);
+	}
+	return rc;
+}
+
+int unit_install(struct wgt_info *ifo, const char *installdir, const char *icondir, int port)
+{
+	return do_install_uninstall(ifo, installdir, icondir, port, do_install_units);
+}
+
+int unit_uninstall(struct wgt_info *ifo)
+{
+	return do_install_uninstall(ifo, "", "", 0, do_uninstall_units);
+}
+
diff --git a/src/wgtpkg-unit.h b/src/wgtpkg-unit.h
index 4843685..1d56eb3 100644
--- a/src/wgtpkg-unit.h
+++ b/src/wgtpkg-unit.h
@@ -18,6 +18,7 @@
 
 
 struct json_object;
+struct wgt_info;
 
 enum unitscope {
 	unitscope_unknown = 0,
@@ -43,3 +44,6 @@ struct unitdesc {
 extern int unit_generator_on(const char *filename);
 extern void unit_generator_off();
 extern int unit_generator_process(struct json_object *jdesc, int (*process)(void *closure, const struct unitdesc descs[], unsigned count), void *closure);
+extern int unit_install(struct wgt_info *ifo, const char *installdir, const char *icondir, int port);
+extern int unit_uninstall(struct wgt_info *ifo);
+