1 files changed, 56 insertions, 0 deletions

diff --git a/src/rest.c b/src/rest.c
index 7ffb79c..2c1018c 100644
--- a/src/rest.c
+++ b/src/rest.c
@@ -270,6 +270,7 @@ static const char *og_cmd_to_uri[OG_CMD_MAX] = {
 	[OG_CMD_RUN_SCHEDULE]	= "run/schedule",
 	[OG_CMD_IMAGES]		= "images",
 	[OG_CMD_CACHE_DELETE]  = "cache/delete",
+	[OG_CMD_CACHE_FETCH]  = "cache/fetch",
 };
 
 static bool og_client_is_busy(const struct og_client *cli,
@@ -4334,6 +4335,47 @@ static int og_cmd_cache_delete(json_t *element, struct og_msg_params *params,
 	return og_send_request(OG_METHOD_POST, OG_CMD_CACHE_DELETE, params, body);
 }
 
+static int og_cmd_cache_fetch(json_t *element, struct og_msg_params *params,
+			    char *buffer_reply)
+{
+	json_t *value, *body, *image, *repository = NULL;
+	const char *key;
+	int err = 0;
+
+	if (json_typeof(element) != JSON_OBJECT)
+		return -1;
+
+	json_object_foreach(element, key, value) {
+		if (!strcmp(key, "clients")) {
+			err = og_json_parse_clients(value, params);
+		} else if (!strcmp(key, "image")) {
+			if (json_typeof(value) != JSON_STRING) {
+				err = -1;
+			}
+			image = value;
+		} else if (!strcmp(key, "repository")) {
+			if (json_typeof(value) != JSON_STRING) {
+				err = -1;
+			}
+			repository = value;
+		}
+
+		if (err < 0)
+			return err;
+	}
+
+	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
+		return -1;
+
+	if (!image || !repository)
+		return -1;
+
+	body = json_copy(element);
+	json_object_del(body, "clients");
+
+	return og_send_request(OG_METHOD_POST, OG_CMD_CACHE_FETCH, params, body);
+}
+
 static int og_cmd_setup(json_t *element, struct og_msg_params *params)
 {
 	json_t *value, *clients;
@@ -8193,6 +8235,7 @@ struct {
 	[OG_URI_IMAGE_RESTRICT]	= { "image/restrict", },
 	[OG_URI_CACHE_LIST]	= { "cache/list", },
 	[OG_URI_CACHE_DELETE]	= { "cache/delete", },
+	[OG_URI_CACHE_FETCH]	= { "cache/fetch", },
 	[OG_URI_PART_SETUP]	= { "setup", },
 	[OG_URI_RUN_SCHEDULE]	= { "run/schedule", },
 	[OG_URI_TASK_RUN]	= { "task/run", },
@@ -8831,6 +8874,19 @@ int og_client_state_process_payload_rest(struct og_client *cli)
 			goto err_process_rest_payload;
 		}
 		err = og_cmd_cache_delete(root, &params, buf_reply);
+	} else if (!strncmp(cmd, "cache/fetch", strlen("cache/fetch"))) {
+		if (method != OG_METHOD_POST) {
+			err = og_client_method_not_found(cli);
+			goto err_process_rest_payload;
+		}
+
+		if (!root) {
+			syslog(LOG_ERR,
+			       "command cache fetch with no payload\n");
+			err = og_client_bad_request(cli);
+			goto err_process_rest_payload;
+		}
+		err = og_cmd_cache_fetch(root, &params, buf_reply);
 	} else if (!strncmp(cmd, "setup", strlen("setup"))) {
 		if (method != OG_METHOD_POST) {
 			err = og_client_method_not_found(cli);