From dc4adc4d713439fc6484339b321a1a1aa6af4fcb Mon Sep 17 00:00:00 2001
From: jesopo
Date: Mon, 27 Jan 2020 00:28:30 +0000
Subject: add non-consuming spec arg types for privateonly and channelonly

---
 src/core_modules/command_spec.py | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'src/core_modules')

diff --git a/src/core_modules/command_spec.py b/src/core_modules/command_spec.py
index 282af581..ae624e9d 100644
--- a/src/core_modules/command_spec.py
+++ b/src/core_modules/command_spec.py
@@ -88,6 +88,14 @@ class Module(ModuleManager.BaseModule):
                 if args:
                     value = server.get_user(args[0], create=True)
                 n = 1
+            elif argument_type.type == "channelonly":
+                if not channel == None:
+                    n = 0
+                error = "Command not valid in private message"
+            elif argument_type.type == "privateonly":
+                if channel == None:
+                    n = 0
+                error = "Command not valid in private message"
 
             options.append([argument_type, value, n, error])
         return options
-- 
cgit v1.3.1-10-gc9f91