add non-consuming spec arg types for privateonly and channelonly

1 files changed, 8 insertions, 0 deletions

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