introduced pre-defined default roles

py/context.py

@@ -74,12 +74,7 @@ def parse_role_section(role):
     return result
 
 def load_role_config(role):
-    roles_config_path = os.path.expanduser(vim.eval("g:vim_ai_roles_config_file"))
-    if not os.path.exists(roles_config_path):
-        raise Exception(f"Role config file does not exist: {roles_config_path}")
-
-    roles = configparser.ConfigParser()
-    roles.read(roles_config_path)
+    roles = read_role_files()
     roles = dict(roles)
 
     enhance_roles_with_custom_function(roles)

py/roles.py

@@ -7,13 +7,7 @@
 roles_py_imported = True
 
 def load_ai_role_names(command_type):
-    roles_config_path = os.path.expanduser(vim.eval("g:vim_ai_roles_config_file"))
-    if not os.path.exists(roles_config_path):
-        raise Exception(f"Role config file does not exist: {roles_config_path}")
-
-    roles = configparser.ConfigParser()
-    roles.read(roles_config_path)
-
+    roles = read_role_files()
     enhance_roles_with_custom_function(roles)
 
     role_names = set()

py/utils.py

@@ -317,3 +317,14 @@ def map_chunk_stream(resp):
     map_chunk = map_chunk_stream if openai_options['stream'] else map_chunk_no_stream
 
     return map(map_chunk, response)
+
+def read_role_files():
+    plugin_root = vim.eval("s:plugin_root")
+    default_roles_config_path = str(os.path.join(plugin_root, "roles-default.ini"))
+    roles_config_path = os.path.expanduser(vim.eval("g:vim_ai_roles_config_file"))
+    if not os.path.exists(roles_config_path):
+        raise Exception(f"Role config file does not exist: {roles_config_path}")
+
+    roles = configparser.ConfigParser()
+    roles.read([default_roles_config_path, roles_config_path])
+    return roles

roles-default.ini

@@ -0,0 +1,10 @@
+# predefined roles
+
+[right.chat]
+ui.open_chat_command = preset_right
+
+[below.chat]
+ui.open_chat_command = preset_below
+
+[tab.chat]
+ui.open_chat_command = preset_tab

tests/context_test.py

@@ -126,6 +126,20 @@ def test_chat_only_role():
     actual_config = context['config']
     assert 'preset_tab' == actual_config['options']['open_chat_command']
 
+def test_default_roles():
+    base = {
+        'config_default': default_config,
+        'config_extension': {},
+        'user_instruction': '/chat-only-role',
+        'user_selection': '',
+        'command_type': 'chat',
+    }
+    context = make_ai_context({ **base, 'user_instruction': '/right hello world!' })
+    assert 'preset_right' == context['config']['ui']['open_chat_command']
+
+    context = make_ai_context({ **base, 'user_instruction': '/tab' })
+    assert 'preset_tab' == context['config']['ui']['open_chat_command']
+
 def test_user_prompt():
     assert 'fix grammar: helo word' == make_prompt( '', 'fix grammar: helo word', '', '')
     assert 'fix grammar:\nhelo word' == make_prompt( '', 'fix grammar', 'helo word', '')

tests/mocks/vim.py

@@ -7,7 +7,9 @@ def eval(cmd):
         case 'g:vim_ai_debug_log_file':
             return '/tmp/vim_ai_debug.log'
         case 'g:vim_ai_roles_config_file':
-            return os.path.join(dirname, '..',  'resources/roles.ini')
+            return os.path.join(dirname, '../resources/roles.ini')
+        case 's:plugin_root':
+            return os.path.join(dirname, '../..')
         case _:
             return None
 

tests/roles_test.py

@@ -17,4 +17,8 @@ def test_role_chat_only():
         'chat-only-role',
         'deprecated-test-role-simple',
         'deprecated-test-role',
+        # default roles
+        'right',
+        'below',
+        'tab',
     }