1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
import asyncio
from textwrap import dedent
from typing import List, Union
from ..models.filesystem import RangeInFile
from ..models.main import Range
from ..core.main import Step
from ..core.sdk import ContinueSDK
import os
import re
# Already have some code for this somewhere
IGNORE_DIRS = ["env", "venv", ".venv"]
IGNORE_FILES = [".env"]
def find_all_matches_in_dir(pattern: str, dirpath: str) -> List[RangeInFile]:
range_in_files = []
for root, dirs, files in os.walk(dirpath):
dirname = os.path.basename(root)
if dirname.startswith(".") or dirname in IGNORE_DIRS:
continue
for file in files:
if file in IGNORE_FILES:
continue
with open(os.path.join(root, file), "r") as f:
# Find the index of all occurences of the pattern in the file. Use re.
file_content = f.read()
results = re.finditer(pattern, file_content)
range_in_files += [
RangeInFile(filepath=os.path.join(root, file), range=Range.from_indices(
file_content, result.start(), result.end()))
for result in results
]
return range_in_files
class WriteRegexPatternStep(Step):
user_request: str
async def run(self, sdk: ContinueSDK):
# Ask the user for a regex pattern
pattern = await sdk.models.gpt35.complete(dedent(f"""\
This is the user request:
{self.user_request}
Please write either a regex pattern or just a string that be used with python's re module to find all matches requested by the user. It will be used as `re.findall(<PATTERN_YOU_WILL_WRITE>, file_content)`. Your output should be only the regex or string, nothing else:"""))
return pattern
class EditAllMatchesStep(Step):
pattern: str
user_request: str
directory: Union[str, None] = None
async def run(self, sdk: ContinueSDK):
# Search all files for a given string
range_in_files = find_all_matches_in_dir(self.pattern, self.directory or await sdk.ide.getWorkspaceDirectory())
tasks = [asyncio.create_task(sdk.edit_file(
range=range_in_file.range,
filename=range_in_file.filepath,
prompt=self.user_request
)) for range_in_file in range_in_files]
await asyncio.gather(*tasks)
|
