1 files changed, 87 insertions, 0 deletions
diff --git a/sway/commands/mark.c b/sway/commands/mark.c
new file mode 100644
index 00000000..c1d959df
--- /dev/null
+++ b/sway/commands/mark.c
@@ -0,0 +1,87 @@
+#include <string.h>
+#include <strings.h>
+#include <stdbool.h>
+#include "sway/commands.h"
+#include "list.h"
+#include "stringop.h"
+
+static void find_marks_callback(swayc_t *container, void *_mark) {
+	char *mark = (char *)_mark;
+
+	int index;
+	if (container->marks && ((index = list_seq_find(container->marks, (int (*)(const void *, const void *))strcmp, mark)) != -1)) {
+		list_del(container->marks, index);
+	}
+}
+
+struct cmd_results *cmd_mark(int argc, char **argv) {
+	struct cmd_results *error = NULL;
+	if (config->reading) return cmd_results_new(CMD_FAILURE, "mark", "Can't be used in config file.");
+	if ((error = checkarg(argc, "mark", EXPECTED_AT_LEAST, 1))) {
+		return error;
+	}
+
+	swayc_t *view = current_container;
+	bool add = false;
+	bool toggle = false;
+
+	if (strcmp(argv[0], "--add") == 0) {
+		--argc; ++argv;
+		add = true;
+	} else if (strcmp(argv[0], "--replace") == 0) {
+		--argc; ++argv;
+	}
+
+	if (argc && strcmp(argv[0], "--toggle") == 0) {
+		--argc; ++argv;
+		toggle = true;
+	}
+
+	if (argc) {
+		char *mark = join_args(argv, argc);
+
+		// Remove all existing marks of this type
+		container_map(&root_container, find_marks_callback, mark);
+
+		if (view->marks) {
+			if (add) {
+				int index;
+				if ((index = list_seq_find(view->marks, (int (*)(const void *, const void *))strcmp, mark)) != -1) {
+					if (toggle) {
+						free(view->marks->items[index]);
+						list_del(view->marks, index);
+
+						if (0 == view->marks->length) {
+							list_free(view->marks);
+							view->marks = NULL;
+						}
+					}
+					free(mark);
+				} else {
+					list_add(view->marks, mark);
+				}
+			} else {
+				if (toggle && list_seq_find(view->marks, (int (*)(const void *, const void *))strcmp, mark) != -1) {
+					// Delete the list
+					list_foreach(view->marks, free);
+					list_free(view->marks);
+					view->marks = NULL;
+				} else {
+					// Delete and replace with a new list
+					list_foreach(view->marks, free);
+					list_free(view->marks);
+
+					view->marks = create_list();
+					list_add(view->marks, mark);
+				}
+			}
+		} else {
+			view->marks = create_list();
+			list_add(view->marks, mark);
+		}
+	} else {
+		return cmd_results_new(CMD_FAILURE, "mark",
+			"Expected 'mark [--add|--replace] [--toggle] <mark>'");
+	}
+	return cmd_results_new(CMD_SUCCESS, NULL, NULL);
+}