add: [bot] command

DavidCruciani · DavidCruciani · commit 8b9bfe5fbb21 · 2024-04-11T14:04:51.000+02:00
diff --git a/bin/case_function.py b/bin/case_function.py
@@ -9,80 +9,160 @@
 
 headers={'X-API-KEY': Config.FLOWINTEL_API_KEY}
 
-def request_post_flowintel(url, data, content_type = 'application/json'):
+def request_post_flowintel(url, data, content_type = 'application/json', matrix_id = None):
     headers["Content-Type"] = content_type
+    if matrix_id:
+        headers["MATRIX-ID"] = matrix_id
     res = requests.post(f"{Config.FLOWINTEL_URL}/api/case/{url}", json=data, headers=headers)
     return res
 
-def request_get_flowintel(url):
+def request_get_flowintel(url, matrix_id = None):
+    if matrix_id:
+        headers["MATRIX-ID"] = matrix_id
     res = requests.get(f"{Config.FLOWINTEL_URL}/api/case/{url}", headers=headers)
     return res
 
+async def check_format(bot, match, message, room, size):
+    if not len(match.args()) >= size:
+        await bot.api.send_text_message(room.room_id, "Not enough argument", reply_to=message.event_id)
+        return False
+    return True
 
-async def modify_case(bot, match, message, room, type):
-    """Modify title or description of a case"""
-    args_list = match.args()[1:]
-    case_id = match.args()[0]
+async def check_matrix_id(bot, message, room):
+    res = requests.get(f"{Config.FLOWINTEL_URL}/api/admin/user_matrix_id?matrix_id={message.sender}", headers=headers)
+    if "message" in res.json():
+        loc_message = res.json()["message"]
+        await bot.api.send_markdown_message(room.room_id, loc_message, reply_to=message.event_id)
+        return False
+    return res
 
-    data = {type: " ".join(arg for arg in args_list)}
-    res = request_post_flowintel(f"{case_id}/edit", data)
+
+async def send_markdown(bot, message, room, res):
     if "message" in res.json():
         loc_message = res.json()["message"]
     else:
         loc_message = f"```JSON\n{json.dumps(res.json(), indent=2)}\n```"
-    last_command = message
     await bot.api.send_markdown_message(room.room_id, loc_message, reply_to=message.event_id)
-    return last_command
+
+
+async def modify_case(bot, match, message, room, type):
+    """Modify title or description of a case"""
+    args_list = match.args()[1:]
+    case_id = match.args()[0]
+
+    data = {type: " ".join(arg for arg in args_list)}
+    res = request_post_flowintel(f"{case_id}/edit", data, matrix_id=message.sender)
+    return send_markdown(bot, message, room, res)
 
 
 async def case_check(bot, match, message, room):
     """All ckeck for case part"""
     last_command = ""
     if match.command("create_case"):
-        data = {"title": " ".join(arg for arg in match.args())}
-        res = request_post_flowintel("create", data)
-        if "message" in res.json():
-            loc_message = res.json()["message"]
-        else:
-            loc_message = f"```JSON\n{json.dumps(res.json(), indent=2)}\n```"
         last_command = message
-        await bot.api.send_text_message(room.room_id, loc_message, reply_to=message.event_id)
+        if await check_format(bot, match, message, room, 1):
+            data = {"title": " ".join(arg for arg in match.args())}
+            res = request_post_flowintel("create", data, matrix_id=message.sender)
+            if "message" in res.json():
+                loc_message = res.json()["message"]
+            else:
+                loc_message = f"```JSON\n{json.dumps(res.json(), indent=2)}\n```"
+            await bot.api.send_text_message(room.room_id, loc_message, reply_to=message.event_id)
 
     if match.command("case_title"):
-        data = {"title": " ".join(arg for arg in match.args())}
-        res = request_post_flowintel("title", data)
-        if "message" in res.json():
-            loc_message = res.json()["message"]
-        else:
-            loc_message = f"```JSON\n{json.dumps(res.json(), indent=2)}\n```"
         last_command = message
-        await bot.api.send_markdown_message(room.room_id, loc_message, reply_to=message.event_id)
+        if await check_format(bot, match, message, room, 1):
+            data = {"title": " ".join(arg for arg in match.args())}
+            res = request_post_flowintel("title", data)
+            await send_markdown(bot, message, room, res)            
 
-    if match.command("case_id"):
-        res = request_get_flowintel(match.args()[0])
-        if "message" in res.json():
-            loc_message = res.json()["message"]
-        else:
-            loc_message = f"```JSON\n{json.dumps(res.json(), indent=2)}\n```"
+    if match.command("case"):
         last_command = message
-        await bot.api.send_markdown_message(room.room_id, loc_message, reply_to=message.event_id)
+        if await check_format(bot, match, message, room, 1):
+            res = request_get_flowintel(match.args()[0])
+            await send_markdown(bot, message, room, res)
+            
 
     if match.command("modify_title"):
-        last_command = modify_case(bot, match, message, room, "title")
-
+        last_command = message
+        if await check_format(bot, match, message, room, 2):
+            await modify_case(bot, match, message, room, "title")
+            
     if match.command("modify_description"):
-        last_command = modify_case(bot, match, message, room, "description")
+        if await check_format(bot, match, message, room, 2):
+            await modify_case(bot, match, message, room, "description")
 
     if match.command("complete_case"):
-        res = request_get_flowintel(f"{match.args()[0]}/complete")
-        if "message" in res.json():
-            loc_message = res.json()["message"]
-        else:
-            loc_message = f"```JSON\n{json.dumps(res.json(), indent=2)}\n```"
         last_command = message
-        await bot.api.send_markdown_message(room.room_id, loc_message, reply_to=message.event_id)
+        if await check_format(bot, match, message, room, 1):
+            res = request_get_flowintel(f"{match.args()[0]}/complete", matrix_id=message.sender)
+            await send_markdown(bot, message, room, res)
+
+    if match.command("case_tasks"):
+        last_command = message
+        if await check_format(bot, match, message, room, 1):
+            res = request_get_flowintel(f"{match.args()[0]}/tasks")
+            await send_markdown(bot, message, room, res)
+
+    if match.command("case_note"):
+        last_command = message
+        if await check_format(bot, match, message, room, 1):
+            res = request_get_flowintel(f"{match.args()[0]}/get_note")
+            await send_markdown(bot, message, room, res)
+
+    if match.command("case_modify_note"):
+        last_command = message
+        if await check_format(bot, match, message, room, 2):
+            args_list = match.args()[1:]
+            case_id = match.args()[0]
+            data = {"note": " ".join(arg for arg in args_list)}
+            res = request_post_flowintel(f"{case_id}/modif_case_note", data, matrix_id=message.sender)
+            await send_markdown(bot, message, room, res)
+            
+
+    if match.command("case_all_notes"):
+        last_command = message
+        if await check_format(bot, match, message, room, 1):
+            res = request_get_flowintel(f"{match.args()[0]}/all_notes")
+            await send_markdown(bot, message, room, res)
+
+    if match.command("case_all_users"):
+        last_command = message
+        if await check_format(bot, match, message, room, 1):
+            res = request_get_flowintel(f"{match.args()[0]}/get_all_users")
+            await send_markdown(bot, message, room, res)
+
+    if match.command("list_status"):
+        last_command = message
+        if await check_format(bot, match, message, room, 1):
+            res = request_get_flowintel(f"list_status")
+            await send_markdown(bot, message, room, res)
+
+    if match.command("case_status"):
+        last_command = message
+        if await check_format(bot, match, message, room, 1):
+            res = request_get_flowintel(match.args()[0])
+            if "message" in res.json():
+                loc_message = res.json()["message"]
+            else:
+                res_status = request_get_flowintel(f"list_status")
+                loc_message = "Status not found"
+                for status in res_status.json():
+                    if status["id"] == res.json()["status_id"]:
+                        loc_message = f"Case is `{status['name']}`"
+            await bot.api.send_markdown_message(room.room_id, loc_message, reply_to=message.event_id)
+
+    if match.command("case_modify_status"):
+        last_command = message
+        if await check_format(bot, match, message, room, 2):
+            case_id = match.args()[0]
+            status_id = match.args()[1]
+            data = {"status_id": status_id}
 
+            res = request_post_flowintel(f"{case_id}/change_status", data, matrix_id=message.sender)
+            await send_markdown(bot, message, room, res)
 
 
     if last_command:
-        return last_command
+        return last_command
+    
diff --git a/bin/help.py b/bin/help.py
@@ -1,21 +1,74 @@
 def help():
     help = "```\n"
 
+    help += "help: Show this output.\n"
+    help += "help_case: Show command for case part.\n"
+    help += "help_task: Show command for task part.\n"
+
     ## Basic command
     help += "last_command: Give the last command used.\n"
     help += "url: Give the url for flowintel-cm instance.\n"
 
     ## Case command
+    help += help_case(False)
+
+    ## Task command
+    help += help_task(False)
+
+    help = "```\n"
+    return help
+
+
+def help_case(standalone=True):
+    ## Case command
+    help = ""
+    if standalone:
+        help = "```\n"
+
     help += "create_case: Create a new case.\n\t!create_case New case name\n"
     help += "case_title: Search a case by title.\n\t!case_title title to search\n"
-    help += "case_id: Get a case by its id.\n\t!case_id 1\n"
-    help += "modify_title: Modify a title of a case:\n\t!modify_title case_id new title\n"
+    help += "case: Get a case by its id.\n\t!case_id case_id\n"
+    help += "modify_title: Modify a title of a case.\n\t!modify_title case_id new title\n"
     help += "modify_description: Modify a description of a case:\n\t!modify_description case_id new description\n"
     help += "complete_case: Mark a case as Finished.\n\t!complete_case case_id\n"
+    help += "case_tasks: List all tasks for a case.\n\t!case_tasks case_id\n"
+    help += "case_note: Get note of a case.\n\t!case_note case_id\n"
+    help += "case_modify_note: Modify note of a case.\n\t!case_modify_note case_id new note\n"
+    help += "case_all_notes: Get notes from all tasks in a case.\n\t!case_all_notes case_id\n"
+    help += "case_all_users: Get list of user that can be assign.\n\t!case_all_users case_id\n"
+    help += "list_status: Get list of status apllicable to a case.\n\t!list_status\n"
+    help += "case_status: Get current status of the case.\n\t!case_status case_id\n"
+    help += "case_modify_status: Modify status of the case.\n\t!case_modify_status case_id status_id\n"
+
+    if standalone:
+        help = "```\n"
+
+    return help
 
+
+def help_task(standalone=True):
     ## Task command
+    help = ""
+    if standalone:
+        help = "```\n"
+
     help += "create_task: Create a new task in a case.\n\t!create_task case_id task title\n"
+    help += "task: Get a task by its id.\n\t!task_id task_id\n"
     help += "my_assignment: Give task assign to user.\n\t!my_assignment page\n"
+    help += "modify_title: Modify a title of a task.\n\t!modify_title task_id new title\n"
+    help += "modify_description: Modify a description of a task.\n\t!modify_description task_id new description\n"
+    help += "complete_task: Mark a task as Finished.\n\t!complete_task task_id\n"
+    help += "create_note: Create a note for a task.\n\t!create_note task_id new note\n"
+    help += "task_get_all_notes: Get all notes for a task.\n\t!task_get_all_notes task_id\n"
+    help += "task_get_note: Get a note for a task.\n\t!task_get_note task_id\n"
+    help += "task_modify_note: Modify a note for a task.\n\t!task_modify_note task_id new note\n"
+    help += "delete_note: Delete a note for a task.\n\t!delete_note task_id\n"
+    help += "take_task: Assign current user to the task.\n\t!take_task task_id\n"
+    help += "remove_assignment: Remove assignment of current user on the task.\n\t!remove_assignment task_id\n"
+    help += "task_status: Get current status of the task.\n\t!task_status task_id\n"
+    help += "task_modify_status: Modify status of the task.\n\t!task_modify_status task_id status_id\n"
 
-    help += "```"
-    return help
+    if standalone:
+        help = "```\n"
+
+    return help
diff --git a/bin/neo.py b/bin/neo.py
@@ -12,7 +12,7 @@
 d = dirname(dirname(abspath(__file__)))
 sys.path.append(os.path.join(d, "config"))
 import config as Config
-from help import help
+from help import help, help_task, help_case
 
 from case_function import case_check
 from task_function import task_check
@@ -40,6 +40,16 @@ async def echo(room, message):
             last_command = message
             await bot.api.send_markdown_message(room.room_id, loc_message, reply_to=message.event_id)
 
+        if match.command("help_case"):
+            loc_message = help_case()
+            last_command = message
+            await bot.api.send_markdown_message(room.room_id, loc_message, reply_to=message.event_id)
+
+        if match.command("help_task"):
+            loc_message = help_task()
+            last_command = message
+            await bot.api.send_markdown_message(room.room_id, loc_message, reply_to=message.event_id)
+
         if match.command("last_command"):
             if last_command:
                 loc_message = f"```\n{last_command.body}```"
diff --git a/bin/task_function.py b/bin/task_function.py
