filesystem.h

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#pragma once
 
#include <entropic/mcp/server_base.h>
#include <entropic/mcp/servers/ignore_matcher.h>
#include <entropic/types/config.h>
 
#include <filesystem>
#include <memory>
#include <string>
#include <unordered_map>
#include <vector>
 
namespace entropic {
 
class FileAccessTracker {
public:
    void record_read(const std::string& path, size_t hash);
 
 
    bool was_read(const std::string& path) const;
 
private:
    std::unordered_map<std::string, size_t> reads_; 
};
 
// Forward declarations for tool classes
class ReadFileTool;
class WriteFileTool;
class EditFileTool;
class GlobTool;
class GrepTool;
class ListDirectoryTool;
 
class FilesystemServer : public MCPServerBase {
public:
    FilesystemServer(const std::filesystem::path& root_dir,
                     const FilesystemConfig& config,
                     const std::string& data_dir,
                     int model_context_bytes = 0);
 
    ~FilesystemServer() override;
 
    bool skip_duplicate_check(const std::string& tool_name) const override;
 
    bool set_working_dir(const std::string& path) override;
 
    const std::filesystem::path& root_dir() const;
 
    FileAccessTracker& tracker();
 
    const FilesystemConfig& config() const;
 
    int max_read_bytes() const;
 
    std::filesystem::path resolve_path(const std::string& requested) const;
 
    const IgnoreMatcher& ignore() const;
 
private:
    void create_fs_tools(const std::string& data_dir);
 
    void register_fs_tools();
 
    std::filesystem::path root_dir_;  
    FilesystemConfig config_;         
    int max_read_bytes_ = 0;          
    FileAccessTracker tracker_;       
    IgnoreMatcher ignore_;            
 
    // Owned tool instances
    std::unique_ptr<ReadFileTool> read_file_;
    std::unique_ptr<WriteFileTool> write_file_;
    std::unique_ptr<EditFileTool> edit_file_;
    std::unique_ptr<GlobTool> glob_;
    std::unique_ptr<GrepTool> grep_;
    std::unique_ptr<ListDirectoryTool> list_dir_;
};
 
} // namespace entropic