summaryrefslogtreecommitdiff
path: root/src/vfs.py
blob: 2bbfdc35838bc0c71766ebaa1a7ca4fc2848f32c (plain)
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
# vfs.py
import os
import copy

class VirtualFileSystem:
    def __init__(self, root_path=None, fs_data=None):
        if fs_data is not None:
            self.fs = fs_data
        elif root_path and os.path.exists(root_path):
            self.fs = self._load_from_path(root_path)
        else:
            self.fs = {"/": {}}

    def _load_from_path(self, root_path):
        fs = {}
        for dirpath, dirnames, filenames in os.walk(root_path):
            # Get the relative path from the root
            rel_path = os.path.relpath(dirpath, root_path)
            if rel_path == ".":
                rel_path = ""

            # Navigate to the correct node in the fs
            parts = rel_path.split(os.sep)
            node = fs
            for part in parts:
                if part:
                    node = node.setdefault(part, {})
            
            # Add files
            for filename in filenames:
                try:
                    with open(os.path.join(dirpath, filename), 'r') as f:
                        node[filename] = f.read()
                except (IOError, UnicodeDecodeError) as e:
                    node[filename] = f"Error reading file: {e}"

        return {"/": fs}

    def _get_path(self, path):
        parts = path.strip("/").split("/")
        node = self.fs["/"]
        for part in parts:
            if part == "":
                continue
            if isinstance(node, dict) and part in node:
                node = node[part]
            else:
                return None
        return node

    def list_files(self, path="."):
        if path == ".":
            path = "/"
        node = self._get_path(path)
        if node is None or not isinstance(node, dict):
            return "Path not found or not a directory."
        return list(node.keys())

    def create_file(self, file_path, content):
        parts = file_path.strip("/").split("/")
        filename = parts.pop()
        if not filename:
            return "Invalid file path."
            
        dir_path = "/" + "/".join(parts)
        node = self._get_path(dir_path)
        
        if node is None or not isinstance(node, dict):
            # Automatically create the directory if it doesn't exist.
            node = self.fs["/"]
            for part in parts:
                if part == "":
                    continue
                if part not in node:
                    node[part] = {}
                node = node[part]

        node[filename] = content
        return f"File '{file_path}' created successfully."

    def read_file(self, file_path):
        node = self._get_path(file_path)
        if node is None or isinstance(node, dict):
            return "File not found or is a directory."
        return node

    def delete_file(self, file_path):
        parts = file_path.strip("/").split("/")
        filename = parts.pop()
        if not filename:
            return "Invalid file path."
        
        dir_path = "/" + "/".join(parts)
        node = self._get_path(dir_path)

        if node is None or not isinstance(node, dict) or filename not in node:
            return "File not found."
        
        del node[filename]
        return f"File '{file_path}' deleted successfully."

    def print_fs(self, path="/", indent=0):
        node = self._get_path(path)
        if node is None:
            return

        for name, content in list(node.items()):
            print("  " * indent + f"- {name}")
            if isinstance(content, dict):
                self.print_fs(f"{path}/{name}".replace("//", "/"), indent + 1)

class VFS:
    _instance = None

    @classmethod
    def get_instance(cls, root_path=None, fs_data=None):
        if cls._instance is None:
            cls._instance = VirtualFileSystem(root_path, fs_data)
        return cls._instance