Implement ipc get_clipboard

1 files changed, 109 insertions, 0 deletions

diff --git a/sway/ipc-server.c b/sway/ipc-server.c
index dca881fa..5bc64b1f 100644
--- a/sway/ipc-server.c
+++ b/sway/ipc-server.c
@@ -55,6 +55,11 @@ struct get_pixels_request {
 	struct wlc_geometry geo;
 };
 
+struct get_clipboard_request {
+   struct ipc_client *client;
+   struct wlc_event_source *event_source;
+};
+
 struct sockaddr_un *ipc_user_sockaddr(void);
 int ipc_handle_connection(int fd, uint32_t mask, void *data);
 int ipc_client_handle_readable(int client_fd, uint32_t mask, void *data);
@@ -322,6 +327,46 @@ void ipc_get_pixels(wlc_handle output) {
 	ipc_get_pixel_requests = unhandled;
 }
 
+static int ipc_selection_data_cb(int fd, uint32_t mask, void *data)
+{
+	assert(data);
+	struct get_clipboard_request *req = (struct get_clipboard_request*) data;
+
+	if (mask & WLC_EVENT_ERROR) {
+		sway_log(L_ERROR, "Selection data fd error");
+		const char *error = "{ \"success\": false, \"error\": "
+			"\"Could not receive text data from clipboard\" }";
+		ipc_send_reply(req->client, error, (uint32_t)strlen(error));
+		goto cleanup;
+	}
+
+	if (mask & WLC_EVENT_READABLE) {
+		char buf[512];
+		int ret = read(fd, buf, 511);
+		if (ret < 0) {
+			sway_log_errno(L_ERROR, "Reading from selection data fd failed");
+			const char *error = "{ \"success\": false, \"error\": "
+				"\"Could not receive text data from clipboard\" }";
+			ipc_send_reply(req->client, error, (uint32_t)strlen(error));
+			goto cleanup;
+		}
+
+		buf[ret] = '\0';
+		json_object *obj = json_object_new_object();
+		json_object_object_add(obj, "success", json_object_new_boolean(true));
+		json_object_object_add(obj, "content", json_object_new_string(buf));
+		const char *str = json_object_to_json_string(obj);
+		ipc_send_reply(req->client, str, (uint32_t)strlen(str));
+		json_object_put(obj);
+	}
+
+cleanup:
+   wlc_event_source_remove(req->event_source);
+   close(fd);
+   free(req);
+   return 0;
+}
+
 void ipc_client_handle_command(struct ipc_client *client) {
 	if (!sway_assert(client != NULL, "client != NULL")) {
 		return;
@@ -565,6 +610,70 @@ void ipc_client_handle_command(struct ipc_client *client) {
 		goto exit_cleanup;
 	}
 
+   case IPC_GET_CLIPBOARD:
+   {
+		if (!(client->security_policy & IPC_FEATURE_GET_CLIPBOARD)) {
+			goto exit_denied;
+		}
+
+		size_t size;
+		bool found = false;
+		const char **types = wlc_get_selection_types(&size);
+		if (types == NULL || size == 0) {
+			const char *error = "{ \"success\": false, \"error\": "
+				"\"Empty clipboard\" }";
+			ipc_send_reply(client, error, (uint32_t)strlen(error));
+			goto exit_cleanup;
+		}
+
+		for (size_t i = 0; i < size; ++i) {
+			if (strcmp(types[i], "text/plain;charset=utf-8") != 0
+					&& strcmp(types[i], "text/plain") != 0) {
+				continue;
+			}
+
+			struct get_clipboard_request *req = malloc(sizeof(*req));
+			if (!req) {
+				sway_log(L_ERROR, "Unable to allocate get_clipboard_request");
+				goto exit_cleanup;
+			}
+
+			int pipes[2];
+			if (pipe(pipes) == -1) {
+				sway_log_errno(L_ERROR, "pipe call failed");
+				free(req);
+				break;
+			}
+
+			fcntl(pipes[0], F_SETFD, FD_CLOEXEC | O_NONBLOCK);
+			fcntl(pipes[1], F_SETFD, FD_CLOEXEC | O_NONBLOCK);
+			if (!wlc_get_selection_data(types[i], pipes[1])) {
+				close(pipes[0]);
+				close(pipes[1]);
+				free(req);
+				sway_log(L_ERROR, "wlc_get_selection_data failed");
+				break;
+			}
+
+			req->client = client;
+			req->event_source = wlc_event_loop_add_fd(pipes[0],
+				WLC_EVENT_READABLE | WLC_EVENT_ERROR | WLC_EVENT_HANGUP,
+				&ipc_selection_data_cb, req);
+			found = true;
+			break;
+		}
+
+		if (!found) {
+			sway_log(L_INFO, "Clipboard has to text data");
+			const char *error = "{ \"success\": false, \"error\": "
+				"\"Could not receive text data from clipboard\" }";
+			ipc_send_reply(client, error, (uint32_t)strlen(error));
+		}
+
+		free(types);
+		goto exit_cleanup;
+	}
+
 	default:
 		sway_log(L_INFO, "Unknown IPC command type %i", client->current_command);
 		goto exit_cleanup;