From 1c8b6798ef59f227a47a7ef41c8e622af88ad417 Mon Sep 17 00:00:00 2001
From: Patrick McDermott <pj@pehjota.net>
Date: Mon, 30 Oct 2017 13:21:35 -0400
Subject: mq_web_view_scheme_set_methods(): New function

---
diff --git a/src/web-view-schemes/local.mk b/src/web-view-schemes/local.mk
index af53bc1..4728608 100644
--- a/src/web-view-schemes/local.mk
+++ b/src/web-view-schemes/local.mk
@@ -1,3 +1,4 @@
 marquee_SOURCES += \
+	%reldir%/methods.c \
 	%reldir%/normal.c \
 	%reldir%/view-source.c
diff --git a/src/web-view-schemes/methods.c b/src/web-view-schemes/methods.c
new file mode 100644
index 0000000..f56cbfe
--- /dev/null
+++ b/src/web-view-schemes/methods.c
@@ -0,0 +1,55 @@
+/*
+ * Web view scheme-specific methods
+ *
+ * Copyright (C) 2017  Patrick McDermott
+ *
+ * This file is part of Marquee.
+ *
+ * Marquee 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.
+ *
+ * Marquee 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 Marquee.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "schemes.h"
+
+#define N_SCHEMES 2
+
+static gboolean
+check_and_set_methods(MqWebViewSchemeMethods **methods,
+	MqWebViewScheme *scheme, const gchar *uri,
+	MqWebViewSchemeMethods *scheme_methods)
+{
+	if (scheme_methods->match_uri(uri)) {
+		if (*methods != scheme_methods) {
+			(*methods)->finalize(scheme);
+			*methods = scheme_methods;
+		}
+		return TRUE;
+	} else {
+		return FALSE;
+	}
+}
+
+void
+mq_web_view_scheme_set_methods(MqWebViewSchemeMethods **methods,
+	MqWebViewScheme *scheme, const gchar *uri)
+{
+#pragma GCC diagnostic push
+#pragma GCC diagnostic ignored "-Wunused-value"
+
+	check_and_set_methods(methods, scheme, uri,
+		&mq_web_view_normal_scheme_methods) ||
+	check_and_set_methods(methods, scheme, uri,
+		&mq_web_view_view_source_scheme_methods);
+
+#pragma GCC diagnostic pop
+}
diff --git a/src/web-view-schemes/schemes.h b/src/web-view-schemes/schemes.h
index adef875..555f30b 100644
--- a/src/web-view-schemes/schemes.h
+++ b/src/web-view-schemes/schemes.h
@@ -62,6 +62,10 @@ typedef struct {
 extern MqWebViewSchemeMethods mq_web_view_normal_scheme_methods;
 extern MqWebViewSchemeMethods mq_web_view_view_source_scheme_methods;
 
+void
+mq_web_view_scheme_set_methods(MqWebViewSchemeMethods **methods,
+	MqWebViewScheme *scheme, const gchar *uri);
+
 G_END_DECLS
 
 #endif  /* MQ_WEB_VIEW_SCHEMES_H */
--
cgit v0.9.1