janito 3.5.1__py3-none-any.whl → 3.7.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (161) hide show
  1. janito/README.md +1 -4
  2. janito/cli/chat_mode/bindings.py +0 -50
  3. janito/cli/chat_mode/session.py +1 -12
  4. janito/cli/chat_mode/shell/commands/multi.py +0 -5
  5. janito/cli/chat_mode/shell/commands/security/allowed_sites.py +33 -47
  6. janito/cli/cli_commands/list_plugins.py +43 -52
  7. janito/cli/core/getters.py +0 -3
  8. janito/cli/core/model_guesser.py +24 -40
  9. janito/cli/main_cli.py +12 -9
  10. janito/cli/prompt_core.py +9 -47
  11. janito/cli/rich_terminal_reporter.py +2 -2
  12. janito/docs/GETTING_STARTED.md +1 -1
  13. janito/drivers/openai/driver.py +0 -1
  14. janito/drivers/zai/driver.py +0 -1
  15. janito/i18n/it.py +46 -46
  16. janito/llm/agent.py +16 -32
  17. janito/llm/auth_utils.py +5 -14
  18. janito/llm/driver.py +0 -8
  19. janito/plugins/__init__.py +12 -31
  20. janito/plugins/auto_loader.py +11 -12
  21. janito/plugins/auto_loader_fixed.py +11 -12
  22. janito/{plugin_system → plugins}/base.py +2 -5
  23. janito/plugins/builtin.py +1 -15
  24. janito/plugins/core_adapter.py +11 -89
  25. janito/plugins/core_loader.py +120 -0
  26. janito/plugins/core_loader_fixed.py +125 -0
  27. janito/plugins/discovery.py +5 -5
  28. janito/plugins/discovery_core.py +9 -14
  29. janito/plugins/manager.py +1 -1
  30. janito/providers/__init__.py +0 -1
  31. janito/providers/moonshot/model_info.py +11 -0
  32. janito/tools/__init__.py +7 -31
  33. janito/tools/adapters/__init__.py +1 -6
  34. janito/tools/adapters/local/__init__.py +70 -7
  35. janito/{plugins/tools → tools/adapters/local}/ask_user.py +3 -3
  36. janito/{plugins/tools → tools/adapters/local}/create_file.py +6 -6
  37. janito/{plugins/tools → tools/adapters/local}/fetch_url.py +3 -3
  38. janito/{plugins/tools → tools/adapters/local}/python_code_run.py +7 -23
  39. janito/{plugins/tools → tools/adapters/local}/python_command_run.py +5 -21
  40. janito/{plugins/tools → tools/adapters/local}/python_file_run.py +5 -21
  41. janito/{plugins/tools → tools/adapters/local}/replace_text_in_file.py +4 -4
  42. janito/{plugins/tools → tools/adapters/local}/run_bash_command.py +3 -3
  43. janito/{plugins/tools → tools/adapters/local}/run_powershell_command.py +3 -3
  44. janito/{plugins/tools → tools/adapters/local}/show_image.py +6 -15
  45. janito/{plugins/core/imagedisplay/tools → tools/adapters/local}/show_image_grid.py +5 -13
  46. janito/tools/function_adapter.py +65 -0
  47. janito/tools/loop_protection_decorator.py +117 -114
  48. janito-3.7.0.dist-info/METADATA +84 -0
  49. {janito-3.5.1.dist-info → janito-3.7.0.dist-info}/RECORD +88 -157
  50. janito/cli/cli_commands/check_tools.py +0 -212
  51. janito/llm/cancellation_manager.py +0 -63
  52. janito/llm/enter_cancellation.py +0 -107
  53. janito/plugin_system/__init__.py +0 -10
  54. janito/plugin_system/core_loader.py +0 -217
  55. janito/plugin_system/core_loader_fixed.py +0 -225
  56. janito/plugins/core/__init__.py +0 -7
  57. janito/plugins/core/codeanalyzer/__init__.py +0 -43
  58. janito/plugins/core/filemanager/__init__.py +0 -124
  59. janito/plugins/core/filemanager/tools/create_file.py +0 -87
  60. janito/plugins/core/filemanager/tools/replace_text_in_file.py +0 -270
  61. janito/plugins/core/imagedisplay/__init__.py +0 -14
  62. janito/plugins/core/imagedisplay/plugin.py +0 -51
  63. janito/plugins/core/imagedisplay/tools/__init__.py +0 -1
  64. janito/plugins/core/imagedisplay/tools/show_image.py +0 -83
  65. janito/plugins/core/system/__init__.py +0 -23
  66. janito/plugins/core/system/tools/run_bash_command.py +0 -204
  67. janito/plugins/core/system/tools/run_powershell_command.py +0 -234
  68. janito/plugins/dev/__init__.py +0 -7
  69. janito/plugins/dev/pythondev/__init__.py +0 -37
  70. janito/plugins/dev/visualization/__init__.py +0 -23
  71. janito/plugins/example_plugin.py +0 -108
  72. janito/plugins/tools/__init__.py +0 -10
  73. janito/plugins/tools/copy_file.py +0 -87
  74. janito/plugins/tools/core_tools_plugin.py +0 -87
  75. janito/plugins/tools/create_directory.py +0 -70
  76. janito/plugins/tools/decorators.py +0 -19
  77. janito/plugins/tools/delete_text_in_file.py +0 -134
  78. janito/plugins/tools/find_files.py +0 -143
  79. janito/plugins/tools/get_file_outline/__init__.py +0 -7
  80. janito/plugins/tools/get_file_outline/core.py +0 -122
  81. janito/plugins/tools/get_file_outline/java_outline.py +0 -47
  82. janito/plugins/tools/get_file_outline/markdown_outline.py +0 -14
  83. janito/plugins/tools/get_file_outline/python_outline.py +0 -303
  84. janito/plugins/tools/get_file_outline/search_outline.py +0 -36
  85. janito/plugins/tools/move_file.py +0 -131
  86. janito/plugins/tools/open_html_in_browser.py +0 -51
  87. janito/plugins/tools/open_url.py +0 -37
  88. janito/plugins/tools/read_chart.py +0 -259
  89. janito/plugins/tools/read_files.py +0 -58
  90. janito/plugins/tools/remove_directory.py +0 -55
  91. janito/plugins/tools/remove_file.py +0 -58
  92. janito/plugins/tools/search_text/__init__.py +0 -7
  93. janito/plugins/tools/search_text/core.py +0 -205
  94. janito/plugins/tools/search_text/match_lines.py +0 -67
  95. janito/plugins/tools/search_text/pattern_utils.py +0 -73
  96. janito/plugins/tools/search_text/traverse_directory.py +0 -145
  97. janito/plugins/tools/show_image_grid.py +0 -85
  98. janito/plugins/tools/validate_file_syntax/__init__.py +0 -7
  99. janito/plugins/tools/validate_file_syntax/core.py +0 -114
  100. janito/plugins/tools/validate_file_syntax/css_validator.py +0 -35
  101. janito/plugins/tools/validate_file_syntax/html_validator.py +0 -100
  102. janito/plugins/tools/validate_file_syntax/jinja2_validator.py +0 -50
  103. janito/plugins/tools/validate_file_syntax/js_validator.py +0 -27
  104. janito/plugins/tools/validate_file_syntax/json_validator.py +0 -6
  105. janito/plugins/tools/validate_file_syntax/markdown_validator.py +0 -109
  106. janito/plugins/tools/validate_file_syntax/ps1_validator.py +0 -32
  107. janito/plugins/tools/validate_file_syntax/python_validator.py +0 -5
  108. janito/plugins/tools/validate_file_syntax/xml_validator.py +0 -11
  109. janito/plugins/tools/validate_file_syntax/yaml_validator.py +0 -6
  110. janito/plugins/tools/view_file.py +0 -172
  111. janito/plugins/ui/__init__.py +0 -7
  112. janito/plugins/ui/userinterface/__init__.py +0 -16
  113. janito/plugins/ui/userinterface/tools/ask_user.py +0 -110
  114. janito/plugins/web/__init__.py +0 -7
  115. janito/plugins/web/webtools/__init__.py +0 -33
  116. janito/plugins/web/webtools/tools/fetch_url.py +0 -458
  117. janito/providers/together/__init__.py +0 -1
  118. janito/providers/together/model_info.py +0 -69
  119. janito/providers/together/provider.py +0 -108
  120. janito/tools/cli_initializer.py +0 -88
  121. janito/tools/initialize.py +0 -70
  122. janito-3.5.1.dist-info/METADATA +0 -229
  123. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/copy_file.py +0 -0
  124. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/create_directory.py +0 -0
  125. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/delete_text_in_file.py +0 -0
  126. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/find_files.py +0 -0
  127. /janito/{plugins/core/codeanalyzer/tools → tools/adapters/local}/get_file_outline/__init__.py +0 -0
  128. /janito/{plugins/core/codeanalyzer/tools → tools/adapters/local}/get_file_outline/core.py +0 -0
  129. /janito/{plugins/core/codeanalyzer/tools → tools/adapters/local}/get_file_outline/java_outline.py +0 -0
  130. /janito/{plugins/core/codeanalyzer/tools → tools/adapters/local}/get_file_outline/markdown_outline.py +0 -0
  131. /janito/{plugins/core/codeanalyzer/tools → tools/adapters/local}/get_file_outline/python_outline.py +0 -0
  132. /janito/{plugins/core/codeanalyzer/tools → tools/adapters/local}/get_file_outline/search_outline.py +0 -0
  133. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/move_file.py +0 -0
  134. /janito/{plugins/web/webtools/tools → tools/adapters/local}/open_html_in_browser.py +0 -0
  135. /janito/{plugins/web/webtools/tools → tools/adapters/local}/open_url.py +0 -0
  136. /janito/{plugins/dev/visualization/tools → tools/adapters/local}/read_chart.py +0 -0
  137. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/read_files.py +0 -0
  138. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/remove_directory.py +0 -0
  139. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/remove_file.py +0 -0
  140. /janito/{plugins/core/codeanalyzer/tools → tools/adapters/local}/search_text/__init__.py +0 -0
  141. /janito/{plugins/core/codeanalyzer/tools → tools/adapters/local}/search_text/core.py +0 -0
  142. /janito/{plugins/core/codeanalyzer/tools → tools/adapters/local}/search_text/match_lines.py +0 -0
  143. /janito/{plugins/core/codeanalyzer/tools → tools/adapters/local}/search_text/pattern_utils.py +0 -0
  144. /janito/{plugins/core/codeanalyzer/tools → tools/adapters/local}/search_text/traverse_directory.py +0 -0
  145. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/validate_file_syntax/__init__.py +0 -0
  146. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/validate_file_syntax/core.py +0 -0
  147. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/validate_file_syntax/css_validator.py +0 -0
  148. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/validate_file_syntax/html_validator.py +0 -0
  149. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/validate_file_syntax/jinja2_validator.py +0 -0
  150. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/validate_file_syntax/js_validator.py +0 -0
  151. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/validate_file_syntax/json_validator.py +0 -0
  152. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/validate_file_syntax/markdown_validator.py +0 -0
  153. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/validate_file_syntax/ps1_validator.py +0 -0
  154. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/validate_file_syntax/python_validator.py +0 -0
  155. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/validate_file_syntax/xml_validator.py +0 -0
  156. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/validate_file_syntax/yaml_validator.py +0 -0
  157. /janito/{plugins/core/filemanager/tools → tools/adapters/local}/view_file.py +0 -0
  158. {janito-3.5.1.dist-info → janito-3.7.0.dist-info}/WHEEL +0 -0
  159. {janito-3.5.1.dist-info → janito-3.7.0.dist-info}/entry_points.txt +0 -0
  160. {janito-3.5.1.dist-info → janito-3.7.0.dist-info}/licenses/LICENSE +0 -0
  161. {janito-3.5.1.dist-info → janito-3.7.0.dist-info}/top_level.txt +0 -0
@@ -1,87 +0,0 @@
1
- import os
2
- from janito.tools.path_utils import expand_path
3
- import shutil
4
- from typing import List, Union
5
- from janito.plugins.tools.decorators import register_core_tool
6
- from janito.tools.tool_base import ToolBase, ToolPermissions
7
- from janito.tools.tool_utils import display_path
8
- from janito.report_events import ReportAction
9
- from janito.i18n import tr
10
-
11
-
12
- @register_core_tool
13
- class CopyFile(ToolBase):
14
- """
15
- Copy one or more files to a target directory, or copy a single file to a new file.
16
- Args:
17
- sources (str): Space-separated path(s) to the file(s) to copy.
18
- For multiple sources, provide a single string with paths separated by spaces.
19
- target (str): Destination path. If copying multiple sources, this must be an existing directory.
20
- overwrite (bool, optional): Overwrite existing files. Default: False.
21
- Recommended only after reading the file to be overwritten.
22
- Returns:
23
- str: Status string for each copy operation.
24
- """
25
-
26
- permissions = ToolPermissions(read=True, write=True)
27
- tool_name = "copy_file"
28
-
29
- def run(self, sources: str, target: str, overwrite: bool = False) -> str:
30
- source_list = [expand_path(src) for src in sources.split() if src]
31
- target = expand_path(target)
32
- messages = []
33
- if len(source_list) > 1:
34
- if not os.path.isdir(target):
35
- return tr(
36
- "❗ Target must be an existing directory when copying multiple files: '{target}'",
37
- target=display_path(target),
38
- )
39
- for src in source_list:
40
- if not os.path.isfile(src):
41
- messages.append(
42
- tr(
43
- "❗ Source file does not exist: '{src}'",
44
- src=display_path(src),
45
- )
46
- )
47
- continue
48
- dst = os.path.join(target, os.path.basename(src))
49
- messages.append(self._copy_one(src, dst, overwrite=overwrite))
50
- else:
51
- src = source_list[0]
52
- if os.path.isdir(target):
53
- dst = os.path.join(target, os.path.basename(src))
54
- else:
55
- dst = target
56
- messages.append(self._copy_one(src, dst, overwrite=overwrite))
57
- return "\n".join(messages)
58
-
59
- def _copy_one(self, src, dst, overwrite=False) -> str:
60
- disp_src = display_path(src)
61
- disp_dst = display_path(dst)
62
- if not os.path.isfile(src):
63
- return tr("❗ Source file does not exist: '{src}'", src=disp_src)
64
- if os.path.exists(dst) and not overwrite:
65
- return tr(
66
- "❗ Target already exists: '{dst}'. Set overwrite=True to replace.",
67
- dst=disp_dst,
68
- )
69
- try:
70
- os.makedirs(os.path.dirname(dst), exist_ok=True)
71
- shutil.copy2(src, dst)
72
- note = (
73
- "\n⚠️ Overwrote existing file. (recommended only after reading the file to be overwritten)"
74
- if (os.path.exists(dst) and overwrite)
75
- else ""
76
- )
77
- self.report_success(
78
- tr("✅ Copied '{src}' to '{dst}'", src=disp_src, dst=disp_dst)
79
- )
80
- return tr("✅ Copied '{src}' to '{dst}'", src=disp_src, dst=disp_dst) + note
81
- except Exception as e:
82
- return tr(
83
- "❗ Copy failed from '{src}' to '{dst}': {err}",
84
- src=disp_src,
85
- dst=disp_dst,
86
- err=str(e),
87
- )
@@ -1,87 +0,0 @@
1
- """
2
- Core tools plugin implementation.
3
- """
4
-
5
- from functools import wraps
6
- from typing import Type
7
- from janito.plugin_system.base import Plugin, PluginMetadata
8
-
9
- from .ask_user import AskUser
10
- from .copy_file import CopyFile
11
- from .create_directory import CreateDirectory
12
- from .create_file import CreateFile
13
- from .delete_text_in_file import DeleteTextInFile
14
- from .fetch_url import FetchUrl
15
- from .find_files import FindFiles
16
- from .move_file import MoveFile
17
- from .open_html_in_browser import OpenHtmlInBrowser
18
- from .open_url import OpenUrl
19
- from .python_code_run import PythonCodeRun
20
- from .python_command_run import PythonCommandRun
21
- from .python_file_run import PythonFileRun
22
- from .read_chart import ReadChart
23
- from .read_files import ReadFiles
24
- from .remove_directory import RemoveDirectory
25
- from .remove_file import RemoveFile
26
- from .replace_text_in_file import ReplaceTextInFile
27
- from .run_bash_command import RunBashCommand
28
- from .run_powershell_command import RunPowershellCommand
29
- from .show_image import ShowImage
30
- from .show_image_grid import ShowImageGrid
31
- from .view_file import ViewFile
32
- from .validate_file_syntax.core import ValidateFileSyntax
33
- from .get_file_outline.core import GetFileOutline
34
- from .search_text.core import SearchText
35
- from .decorators import get_core_tools
36
-
37
- # Registry for core tools
38
- _core_tools_registry = []
39
-
40
-
41
- def register_core_tool(cls: Type):
42
- """Decorator to register a core tool."""
43
- _core_tools_registry.append(cls)
44
- return cls
45
-
46
-
47
- class CoreToolsPlugin(Plugin):
48
- """Core tools plugin providing essential janito functionality."""
49
-
50
- def get_metadata(self):
51
- return PluginMetadata(
52
- name="core_tools",
53
- version="1.0.0",
54
- description="Core tools for file operations, code execution, and system interactions",
55
- author="janito team",
56
- license="MIT",
57
- )
58
-
59
- def get_tools(self):
60
- return [
61
- AskUser,
62
- CopyFile,
63
- CreateDirectory,
64
- CreateFile,
65
- DeleteTextInFile,
66
- FetchUrl,
67
- FindFiles,
68
- MoveFile,
69
- OpenHtmlInBrowser,
70
- OpenUrl,
71
- PythonCodeRun,
72
- PythonCommandRun,
73
- PythonFileRun,
74
- ReadChart,
75
- ReadFiles,
76
- RemoveDirectory,
77
- RemoveFile,
78
- ReplaceTextInFile,
79
- RunBashCommand,
80
- RunPowershellCommand,
81
- ShowImage,
82
- ShowImageGrid,
83
- ViewFile,
84
- ValidateFileSyntax,
85
- GetFileOutline,
86
- SearchText,
87
- ] + get_core_tools()
@@ -1,70 +0,0 @@
1
- from janito.plugins.tools.decorators import register_core_tool
2
-
3
- from janito.tools.tool_utils import display_path
4
- from janito.tools.tool_base import ToolBase, ToolPermissions
5
- from janito.report_events import ReportAction
6
- from janito.i18n import tr
7
- import os
8
- from janito.tools.path_utils import expand_path
9
-
10
-
11
- @register_core_tool
12
- class CreateDirectory(ToolBase):
13
- """
14
- Create a new directory at the specified path.
15
- Args:
16
- path (str): Path for the new directory.
17
- Returns:
18
- str: Status message indicating the result. Example:
19
- - "5c5 Successfully created the directory at ..."
20
- - "5d7 Cannot create directory: ..."
21
- """
22
-
23
- permissions = ToolPermissions(write=True)
24
- tool_name = "create_directory"
25
-
26
- def run(self, path: str) -> str:
27
- path = expand_path(path)
28
- disp_path = display_path(path)
29
- self.report_action(
30
- tr("📁 Create directory '{disp_path}' ...", disp_path=disp_path),
31
- ReportAction.CREATE,
32
- )
33
- try:
34
- if os.path.exists(path):
35
- if not os.path.isdir(path):
36
- self.report_error(
37
- tr(
38
- "❌ Path '{disp_path}' exists and is not a directory.",
39
- disp_path=disp_path,
40
- )
41
- )
42
- return tr(
43
- "❌ Path '{disp_path}' exists and is not a directory.",
44
- disp_path=disp_path,
45
- )
46
- self.report_error(
47
- tr(
48
- "❗ Directory '{disp_path}' already exists.",
49
- disp_path=disp_path,
50
- )
51
- )
52
- return tr(
53
- "❗ Cannot create directory: '{disp_path}' already exists.",
54
- disp_path=disp_path,
55
- )
56
- os.makedirs(path, exist_ok=True)
57
- self.report_success(tr("✅ Directory created"))
58
- return tr(
59
- "✅ Successfully created the directory at '{disp_path}'.",
60
- disp_path=disp_path,
61
- )
62
- except Exception as e:
63
- self.report_error(
64
- tr(
65
- "❌ Error creating directory '{disp_path}': {error}",
66
- disp_path=disp_path,
67
- error=e,
68
- )
69
- )
70
- return tr("❌ Cannot create directory: {error}", error=e)
@@ -1,19 +0,0 @@
1
- """
2
- Decorators for core tools registration.
3
- """
4
-
5
- from typing import Type
6
-
7
- # Registry for core tools
8
- _core_tools_registry = []
9
-
10
-
11
- def register_core_tool(cls: Type):
12
- """Decorator to register a core tool."""
13
- _core_tools_registry.append(cls)
14
- return cls
15
-
16
-
17
- def get_core_tools():
18
- """Get all registered core tools."""
19
- return list(_core_tools_registry)
@@ -1,134 +0,0 @@
1
- from janito.tools.tool_base import ToolBase, ToolPermissions
2
- from janito.report_events import ReportAction
3
- from janito.plugins.tools.decorators import register_core_tool
4
- from janito.i18n import tr
5
- import shutil
6
- from janito.plugins.tools.validate_file_syntax.core import validate_file_syntax
7
-
8
-
9
- @register_core_tool
10
- class DeleteTextInFile(ToolBase):
11
- """
12
- Delete all occurrences of text between start_marker and end_marker (inclusive) in a file, using exact string markers.
13
-
14
- Args:
15
- path (str): Path to the file to modify.
16
- start_marker (str): The starting delimiter string.
17
- end_marker (str): The ending delimiter string.
18
-
19
- Returns:
20
- str: Status message indicating the result.
21
- """
22
-
23
- permissions = ToolPermissions(read=True, write=True)
24
- tool_name = "delete_text_in_file"
25
-
26
- def run(
27
- self,
28
- path: str,
29
- start_marker: str,
30
- end_marker: str,
31
- backup: bool = False,
32
- ) -> str:
33
- from janito.tools.tool_utils import display_path
34
-
35
- disp_path = display_path(path)
36
- info_msg = tr(
37
- "📝 Delete text in {disp_path} between markers: '{start_marker}' ... '{end_marker}'",
38
- disp_path=disp_path,
39
- start_marker=start_marker,
40
- end_marker=end_marker,
41
- )
42
- self.report_action(info_msg, ReportAction.CREATE)
43
- try:
44
- content = self._read_file_content(path)
45
- occurrences, match_lines = self._find_marker_blocks(
46
- content, start_marker, end_marker
47
- )
48
- if occurrences == 0:
49
- self.report_warning(
50
- tr(" ℹ️ No blocks found between markers."), ReportAction.CREATE
51
- )
52
- return tr(
53
- "No blocks found between markers in {path}.",
54
- path=path,
55
- )
56
-
57
- new_content, deleted_blocks = self._delete_blocks(
58
- content, start_marker, end_marker
59
- )
60
- self._write_file_content(path, new_content)
61
- validation_result = validate_file_syntax(path)
62
- self._report_success(match_lines)
63
- return tr(
64
- "Deleted {count} block(s) between markers in {path}. ",
65
- count=deleted_blocks,
66
- path=path,
67
- ) + (f"\n{validation_result}" if validation_result else "")
68
- except Exception as e:
69
- self.report_error(tr(" ❌ Error: {error}", error=e), ReportAction.REPLACE)
70
- return tr("Error deleting text: {error}", error=e)
71
-
72
- def _read_file_content(self, path):
73
- with open(path, "r", encoding="utf-8", errors="replace") as f:
74
- return f.read()
75
-
76
- def _find_marker_blocks(self, content, start_marker, end_marker):
77
- """Find all blocks between start_marker and end_marker, return count and starting line numbers."""
78
- lines = content.splitlines(keepends=True)
79
- joined = "".join(lines)
80
- match_lines = []
81
- idx = 0
82
- occurrences = 0
83
- while True:
84
- start_idx = joined.find(start_marker, idx)
85
- if start_idx == -1:
86
- break
87
- end_idx = joined.find(end_marker, start_idx + len(start_marker))
88
- if end_idx == -1:
89
- break
90
- upto = joined[:start_idx]
91
- line_no = upto.count("\n") + 1
92
- match_lines.append(line_no)
93
- idx = end_idx + len(end_marker)
94
- occurrences += 1
95
- return occurrences, match_lines
96
-
97
- def _delete_blocks(self, content, start_marker, end_marker):
98
- """Delete all blocks between start_marker and end_marker (inclusive)."""
99
- count = 0
100
- new_content = content
101
- while True:
102
- start_idx = new_content.find(start_marker)
103
- if start_idx == -1:
104
- break
105
- end_idx = new_content.find(end_marker, start_idx + len(start_marker))
106
- if end_idx == -1:
107
- break
108
- new_content = (
109
- new_content[:start_idx] + new_content[end_idx + len(end_marker) :]
110
- )
111
- count += 1
112
- return new_content, count
113
-
114
- def _backup_file(self, path, backup_path):
115
- shutil.copy2(path, backup_path)
116
-
117
- def _write_file_content(self, path, content):
118
- with open(path, "w", encoding="utf-8", errors="replace") as f:
119
- f.write(content)
120
-
121
- def _report_success(self, match_lines):
122
- if match_lines:
123
- lines_str = ", ".join(str(line_no) for line_no in match_lines)
124
- self.report_success(
125
- tr(
126
- " ✅ deleted block(s) starting at line(s): {lines_str}",
127
- lines_str=lines_str,
128
- ),
129
- ReportAction.CREATE,
130
- )
131
- else:
132
- self.report_success(
133
- tr(" ✅ deleted block(s) (lines unknown)"), ReportAction.CREATE
134
- )
@@ -1,143 +0,0 @@
1
- from janito.tools.tool_base import ToolBase, ToolPermissions
2
- from janito.report_events import ReportAction
3
- from janito.plugins.tools.decorators import register_core_tool
4
- from janito.tools.tool_utils import pluralize, display_path
5
- from janito.dir_walk_utils import walk_dir_with_gitignore
6
- from janito.i18n import tr
7
- import fnmatch
8
- import os
9
- from janito.tools.path_utils import expand_path
10
- from janito.tools.loop_protection_decorator import protect_against_loops
11
-
12
-
13
- @register_core_tool
14
- class FindFiles(ToolBase):
15
- """
16
- Find files or directories in one or more directories matching a pattern. Respects .gitignore.
17
-
18
- If a path is an existing file, it is checked against the provided pattern(s) and included in the results if it matches. This allows find_files to be used to look for a specific set of filenames in a single call, as well as searching directories.
19
-
20
- Args:
21
- paths (str): String of one or more paths (space-separated) to search in. Each path can be a directory or a file.
22
- pattern (str): File pattern(s) to match. Multiple patterns can be separated by spaces. Uses Unix shell-style wildcards (fnmatch), e.g. '*.py', 'data_??.csv', '[a-z]*.txt'.
23
- - If the pattern ends with '/' or '\', only matching directory names (with trailing slash) are returned, not the files within those directories. For example, pattern '*/' will return only directories at the specified depth.
24
- max_depth (int, optional): Maximum directory depth to search. If None, unlimited recursion. If 0, only the top-level directory. If 1, only the root directory (matches 'find . -maxdepth 1').
25
- include_gitignored (bool, optional): If True, includes files/directories ignored by .gitignore. Defaults to False.
26
- Returns:
27
- str: Newline-separated list of matching file paths. Example:
28
- "/path/to/file1.py\n/path/to/file2.py"
29
- "Warning: Empty file pattern provided. Operation skipped."
30
- """
31
-
32
- permissions = ToolPermissions(read=True)
33
- tool_name = "find_files"
34
-
35
- def _match_directories(self, root, dirs, pat):
36
- dir_output = set()
37
- dir_pat = pat.rstrip("/\\")
38
- for d in dirs:
39
- if fnmatch.fnmatch(d, dir_pat):
40
- dir_output.add(os.path.join(root, d) + os.sep)
41
- return dir_output
42
-
43
- def _match_files(self, root, files, pat):
44
- file_output = set()
45
- for filename in fnmatch.filter(files, pat):
46
- file_output.add(os.path.join(root, filename))
47
- return file_output
48
-
49
- def _match_dirs_without_slash(self, root, dirs, pat):
50
- dir_output = set()
51
- for d in fnmatch.filter(dirs, pat):
52
- dir_output.add(os.path.join(root, d))
53
- return dir_output
54
-
55
- def _handle_path(self, directory, patterns):
56
- dir_output = set()
57
- filename = os.path.basename(directory)
58
- for pat in patterns:
59
- # Only match files, not directories, for file paths
60
- if not (pat.endswith("/") or pat.endswith("\\")):
61
- if fnmatch.fnmatch(filename, pat):
62
- dir_output.add(directory)
63
- break
64
- return dir_output
65
-
66
- def _handle_directory_path(
67
- self, directory, patterns, max_depth, include_gitignored
68
- ):
69
- dir_output = set()
70
- for root, dirs, files in walk_dir_with_gitignore(
71
- directory,
72
- max_depth=max_depth,
73
- include_gitignored=include_gitignored,
74
- ):
75
- for pat in patterns:
76
- if pat.endswith("/") or pat.endswith("\\"):
77
- dir_output.update(self._match_directories(root, dirs, pat))
78
- else:
79
- dir_output.update(self._match_files(root, files, pat))
80
- dir_output.update(self._match_dirs_without_slash(root, dirs, pat))
81
- return dir_output
82
-
83
- def _report_search(self, pattern, disp_path, depth_msg):
84
- self.report_action(
85
- tr(
86
- "🔍 Search for files '{pattern}' in '{disp_path}'{depth_msg} ...",
87
- pattern=pattern,
88
- disp_path=disp_path,
89
- depth_msg=depth_msg,
90
- ),
91
- ReportAction.READ,
92
- )
93
-
94
- def _report_success(self, count):
95
- self.report_success(
96
- tr(
97
- " ✅ {count} {file_word}",
98
- count=count,
99
- file_word=pluralize("file", count),
100
- ),
101
- ReportAction.READ,
102
- )
103
-
104
- def _format_output(self, directory, dir_output):
105
- if directory.strip() == ".":
106
- dir_output = {
107
- p[2:] if (p.startswith("./") or p.startswith(".\\")) else p
108
- for p in dir_output
109
- }
110
- return sorted(dir_output)
111
-
112
- @protect_against_loops(max_calls=5, time_window=10.0, key_field="paths")
113
- def run(
114
- self,
115
- paths: str,
116
- pattern: str,
117
- max_depth: int = None,
118
- include_gitignored: bool = False,
119
- ) -> str:
120
- if not pattern:
121
- self.report_warning(tr("ℹ️ Empty file pattern provided."), ReportAction.READ)
122
- return tr("Warning: Empty file pattern provided. Operation skipped.")
123
- patterns = pattern.split()
124
- results = []
125
- for directory in [expand_path(p) for p in paths.split()]:
126
- disp_path = display_path(directory)
127
- depth_msg = (
128
- tr(" (max depth: {max_depth})", max_depth=max_depth)
129
- if max_depth is not None and max_depth > 0
130
- else ""
131
- )
132
- self._report_search(pattern, disp_path, depth_msg)
133
- dir_output = set()
134
- if os.path.isfile(directory):
135
- dir_output = self._handle_path(directory, patterns)
136
- elif os.path.isdir(directory):
137
- dir_output = self._handle_directory_path(
138
- directory, patterns, max_depth, include_gitignored
139
- )
140
- self._report_success(len(dir_output))
141
- results.extend(self._format_output(directory, dir_output))
142
- result = "\n".join(results)
143
- return result
@@ -1,7 +0,0 @@
1
- """
2
- File outline tools for janito.
3
- """
4
-
5
- from .core import GetFileOutline
6
-
7
- __all__ = ["GetFileOutline"]
@@ -1,122 +0,0 @@
1
- from janito.tools.adapters.local.adapter import register_local_tool
2
- from .python_outline import parse_python_outline
3
- from .markdown_outline import parse_markdown_outline
4
- from janito.formatting import OutlineFormatter
5
- from .java_outline import parse_java_outline
6
- import os
7
- from janito.tools.path_utils import expand_path
8
- from janito.tools.tool_base import ToolBase, ToolPermissions
9
- from janito.report_events import ReportAction
10
- from janito.tools.tool_utils import display_path, pluralize
11
- from janito.i18n import tr
12
-
13
- from janito.plugins.tools.decorators import register_core_tool
14
- from janito.tools.loop_protection_decorator import protect_against_loops
15
-
16
-
17
- @register_core_tool
18
- class GetFileOutline(ToolBase):
19
- """
20
- Get an outline of a file's structure. Supports Python and Markdown files.
21
-
22
- Args:
23
- path (str): Path to the file to outline.
24
- """
25
-
26
- permissions = ToolPermissions(read=True)
27
- tool_name = "get_file_outline"
28
-
29
- @protect_against_loops(max_calls=5, time_window=10.0, key_field="path")
30
- def run(self, path: str) -> str:
31
- try:
32
- path = expand_path(path)
33
- self.report_action(
34
- tr(
35
- "📄 Outline file '{disp_path}' ...",
36
- disp_path=display_path(path),
37
- ),
38
- ReportAction.READ,
39
- )
40
- ext = os.path.splitext(path)[1].lower()
41
- with open(path, "r", encoding="utf-8", errors="replace") as f:
42
- lines = f.readlines()
43
- return self._outline_by_extension(ext, lines)
44
- except Exception as e:
45
- self.report_error(
46
- tr("❌ Error reading file: {error}", error=e),
47
- ReportAction.READ,
48
- )
49
- return tr("Error reading file: {error}", error=e)
50
-
51
- def _outline_by_extension(self, ext, lines):
52
- if ext == ".py":
53
- outline_items = parse_python_outline(lines)
54
- outline_type = "python"
55
- table = OutlineFormatter.format_outline_table(outline_items)
56
- self.report_success(
57
- tr(
58
- "✅ Outlined {count} {item_word}",
59
- count=len(outline_items),
60
- item_word=pluralize("item", len(outline_items)),
61
- ),
62
- ReportAction.READ,
63
- )
64
- return (
65
- tr(
66
- "Outline: {count} items ({outline_type})\n",
67
- count=len(outline_items),
68
- outline_type=outline_type,
69
- )
70
- + table
71
- )
72
- elif ext == ".md":
73
- outline_items = parse_markdown_outline(lines)
74
- outline_type = "markdown"
75
- table = OutlineFormatter.format_markdown_outline_table(outline_items)
76
- self.report_success(
77
- tr(
78
- "✅ Outlined {count} {item_word}",
79
- count=len(outline_items),
80
- item_word=pluralize("item", len(outline_items)),
81
- ),
82
- ReportAction.READ,
83
- )
84
- return (
85
- tr(
86
- "Outline: {count} items ({outline_type})\n",
87
- count=len(outline_items),
88
- outline_type=outline_type,
89
- )
90
- + table
91
- )
92
- elif ext == ".java":
93
- outline_items = parse_java_outline(lines)
94
- outline_type = "java"
95
- table = OutlineFormatter.format_outline_table(outline_items)
96
- self.report_success(
97
- tr(
98
- "✅ Outlined {count} {item_word}",
99
- count=len(outline_items),
100
- item_word=pluralize("item", len(outline_items)),
101
- ),
102
- ReportAction.READ,
103
- )
104
- return (
105
- tr(
106
- "Outline: {count} items ({outline_type})\n",
107
- count=len(outline_items),
108
- outline_type=outline_type,
109
- )
110
- + table
111
- )
112
- else:
113
- outline_type = "default"
114
- self.report_success(
115
- tr("✅ Outlined {count} items", count=len(lines)),
116
- ReportAction.READ,
117
- )
118
- return tr(
119
- "Outline: {count} lines ({outline_type})\nFile has {count} lines.",
120
- count=len(lines),
121
- outline_type=outline_type,
122
- )