agentfense 0.2.1__py3-none-any.whl

agentfense/_gen/codebase_pb2.py

@@ -0,0 +1,78 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: codebase.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'codebase.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.api import annotations_pb2 as google_dot_api_dot_annotations__pb2
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+from . import common_pb2 as common__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0e\x63odebase.proto\x12\nsandbox.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x0c\x63ommon.proto\"\xb8\x01\n\x08\x43odebase\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x10\n\x08owner_id\x18\x03 \x01(\t\x12\x0c\n\x04size\x18\x04 \x01(\x03\x12\x12\n\nfile_count\x18\x05 \x01(\x05\x12.\n\ncreated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\nupdated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"7\n\x15\x43reateCodebaseRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x10\n\x08owner_id\x18\x02 \x01(\t\")\n\x12GetCodebaseRequest\x12\x13\n\x0b\x63odebase_id\x18\x01 \x01(\t\"O\n\x14ListCodebasesRequest\x12\x10\n\x08owner_id\x18\x01 \x01(\t\x12\x11\n\tpage_size\x18\x02 \x01(\x05\x12\x12\n\npage_token\x18\x03 \x01(\t\"Y\n\x15ListCodebasesResponse\x12\'\n\tcodebases\x18\x01 \x03(\x0b\x32\x14.sandbox.v1.Codebase\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\",\n\x15\x44\x65leteCodebaseRequest\x12\x13\n\x0b\x63odebase_id\x18\x01 \x01(\t\"\xa6\x01\n\x0bUploadChunk\x12\x34\n\x08metadata\x18\x01 \x01(\x0b\x32 .sandbox.v1.UploadChunk.MetadataH\x00\x12\x0e\n\x04\x64\x61ta\x18\x02 \x01(\x0cH\x00\x1a\x46\n\x08Metadata\x12\x13\n\x0b\x63odebase_id\x18\x01 \x01(\t\x12\x11\n\tfile_path\x18\x02 \x01(\t\x12\x12\n\ntotal_size\x18\x03 \x01(\x03\x42\t\n\x07\x63ontent\"V\n\x0cUploadResult\x12\x13\n\x0b\x63odebase_id\x18\x01 \x01(\t\x12\x11\n\tfile_path\x18\x02 \x01(\t\x12\x0c\n\x04size\x18\x03 \x01(\x03\x12\x10\n\x08\x63hecksum\x18\x04 \x01(\t\"=\n\x13\x44ownloadFileRequest\x12\x13\n\x0b\x63odebase_id\x18\x01 \x01(\t\x12\x11\n\tfile_path\x18\x02 \x01(\t\"\x19\n\tFileChunk\x12\x0c\n\x04\x64\x61ta\x18\x01 \x01(\x0c\"H\n\x10ListFilesRequest\x12\x13\n\x0b\x63odebase_id\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\t\x12\x11\n\trecursive\x18\x03 \x01(\x08\"8\n\x11ListFilesResponse\x12#\n\x05\x66iles\x18\x01 \x03(\x0b\x32\x14.sandbox.v1.FileInfo\"u\n\x08\x46ileInfo\x12\x0c\n\x04path\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x0e\n\x06is_dir\x18\x03 \x01(\x08\x12\x0c\n\x04size\x18\x04 \x01(\x03\x12/\n\x0bmodified_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp2\xbd\x05\n\x0f\x43odebaseService\x12\x63\n\x0e\x43reateCodebase\x12!.sandbox.v1.CreateCodebaseRequest\x1a\x14.sandbox.v1.Codebase\"\x18\x82\xd3\xe4\x93\x02\x12\"\r/v1/codebases:\x01*\x12h\n\x0bGetCodebase\x12\x1e.sandbox.v1.GetCodebaseRequest\x1a\x14.sandbox.v1.Codebase\"#\x82\xd3\xe4\x93\x02\x1d\x12\x1b/v1/codebases/{codebase_id}\x12k\n\rListCodebases\x12 .sandbox.v1.ListCodebasesRequest\x1a!.sandbox.v1.ListCodebasesResponse\"\x15\x82\xd3\xe4\x93\x02\x0f\x12\r/v1/codebases\x12k\n\x0e\x44\x65leteCodebase\x12!.sandbox.v1.DeleteCodebaseRequest\x1a\x11.sandbox.v1.Empty\"#\x82\xd3\xe4\x93\x02\x1d*\x1b/v1/codebases/{codebase_id}\x12\x42\n\x0bUploadFiles\x12\x17.sandbox.v1.UploadChunk\x1a\x18.sandbox.v1.UploadResult(\x01\x12H\n\x0c\x44ownloadFile\x12\x1f.sandbox.v1.DownloadFileRequest\x1a\x15.sandbox.v1.FileChunk0\x01\x12s\n\tListFiles\x12\x1c.sandbox.v1.ListFilesRequest\x1a\x1d.sandbox.v1.ListFilesResponse\")\x82\xd3\xe4\x93\x02#\x12!/v1/codebases/{codebase_id}/filesB=Z;github.com/AjaxZhan/AgentFense/api/gen/sandbox/v1;sandboxv1b\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'codebase_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z;github.com/AjaxZhan/AgentFense/api/gen/sandbox/v1;sandboxv1'
+  _globals['_CODEBASESERVICE'].methods_by_name['CreateCodebase']._loaded_options = None
+  _globals['_CODEBASESERVICE'].methods_by_name['CreateCodebase']._serialized_options = b'\202\323\344\223\002\022\"\r/v1/codebases:\001*'
+  _globals['_CODEBASESERVICE'].methods_by_name['GetCodebase']._loaded_options = None
+  _globals['_CODEBASESERVICE'].methods_by_name['GetCodebase']._serialized_options = b'\202\323\344\223\002\035\022\033/v1/codebases/{codebase_id}'
+  _globals['_CODEBASESERVICE'].methods_by_name['ListCodebases']._loaded_options = None
+  _globals['_CODEBASESERVICE'].methods_by_name['ListCodebases']._serialized_options = b'\202\323\344\223\002\017\022\r/v1/codebases'
+  _globals['_CODEBASESERVICE'].methods_by_name['DeleteCodebase']._loaded_options = None
+  _globals['_CODEBASESERVICE'].methods_by_name['DeleteCodebase']._serialized_options = b'\202\323\344\223\002\035*\033/v1/codebases/{codebase_id}'
+  _globals['_CODEBASESERVICE'].methods_by_name['ListFiles']._loaded_options = None
+  _globals['_CODEBASESERVICE'].methods_by_name['ListFiles']._serialized_options = b'\202\323\344\223\002#\022!/v1/codebases/{codebase_id}/files'
+  _globals['_CODEBASE']._serialized_start=108
+  _globals['_CODEBASE']._serialized_end=292
+  _globals['_CREATECODEBASEREQUEST']._serialized_start=294
+  _globals['_CREATECODEBASEREQUEST']._serialized_end=349
+  _globals['_GETCODEBASEREQUEST']._serialized_start=351
+  _globals['_GETCODEBASEREQUEST']._serialized_end=392
+  _globals['_LISTCODEBASESREQUEST']._serialized_start=394
+  _globals['_LISTCODEBASESREQUEST']._serialized_end=473
+  _globals['_LISTCODEBASESRESPONSE']._serialized_start=475
+  _globals['_LISTCODEBASESRESPONSE']._serialized_end=564
+  _globals['_DELETECODEBASEREQUEST']._serialized_start=566
+  _globals['_DELETECODEBASEREQUEST']._serialized_end=610
+  _globals['_UPLOADCHUNK']._serialized_start=613
+  _globals['_UPLOADCHUNK']._serialized_end=779
+  _globals['_UPLOADCHUNK_METADATA']._serialized_start=698
+  _globals['_UPLOADCHUNK_METADATA']._serialized_end=768
+  _globals['_UPLOADRESULT']._serialized_start=781
+  _globals['_UPLOADRESULT']._serialized_end=867
+  _globals['_DOWNLOADFILEREQUEST']._serialized_start=869
+  _globals['_DOWNLOADFILEREQUEST']._serialized_end=930
+  _globals['_FILECHUNK']._serialized_start=932
+  _globals['_FILECHUNK']._serialized_end=957
+  _globals['_LISTFILESREQUEST']._serialized_start=959
+  _globals['_LISTFILESREQUEST']._serialized_end=1031
+  _globals['_LISTFILESRESPONSE']._serialized_start=1033
+  _globals['_LISTFILESRESPONSE']._serialized_end=1089
+  _globals['_FILEINFO']._serialized_start=1091
+  _globals['_FILEINFO']._serialized_end=1208
+  _globals['_CODEBASESERVICE']._serialized_start=1211
+  _globals['_CODEBASESERVICE']._serialized_end=1912
+# @@protoc_insertion_point(module_scope)

agentfense/_gen/codebase_pb2.pyi

@@ -0,0 +1,141 @@
+import datetime
+
+from google.api import annotations_pb2 as _annotations_pb2
+from google.protobuf import timestamp_pb2 as _timestamp_pb2
+import common_pb2 as _common_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from collections.abc import Iterable as _Iterable, Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class Codebase(_message.Message):
+    __slots__ = ("id", "name", "owner_id", "size", "file_count", "created_at", "updated_at")
+    ID_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    OWNER_ID_FIELD_NUMBER: _ClassVar[int]
+    SIZE_FIELD_NUMBER: _ClassVar[int]
+    FILE_COUNT_FIELD_NUMBER: _ClassVar[int]
+    CREATED_AT_FIELD_NUMBER: _ClassVar[int]
+    UPDATED_AT_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    name: str
+    owner_id: str
+    size: int
+    file_count: int
+    created_at: _timestamp_pb2.Timestamp
+    updated_at: _timestamp_pb2.Timestamp
+    def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., owner_id: _Optional[str] = ..., size: _Optional[int] = ..., file_count: _Optional[int] = ..., created_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., updated_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
+
+class CreateCodebaseRequest(_message.Message):
+    __slots__ = ("name", "owner_id")
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    OWNER_ID_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    owner_id: str
+    def __init__(self, name: _Optional[str] = ..., owner_id: _Optional[str] = ...) -> None: ...
+
+class GetCodebaseRequest(_message.Message):
+    __slots__ = ("codebase_id",)
+    CODEBASE_ID_FIELD_NUMBER: _ClassVar[int]
+    codebase_id: str
+    def __init__(self, codebase_id: _Optional[str] = ...) -> None: ...
+
+class ListCodebasesRequest(_message.Message):
+    __slots__ = ("owner_id", "page_size", "page_token")
+    OWNER_ID_FIELD_NUMBER: _ClassVar[int]
+    PAGE_SIZE_FIELD_NUMBER: _ClassVar[int]
+    PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int]
+    owner_id: str
+    page_size: int
+    page_token: str
+    def __init__(self, owner_id: _Optional[str] = ..., page_size: _Optional[int] = ..., page_token: _Optional[str] = ...) -> None: ...
+
+class ListCodebasesResponse(_message.Message):
+    __slots__ = ("codebases", "next_page_token")
+    CODEBASES_FIELD_NUMBER: _ClassVar[int]
+    NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int]
+    codebases: _containers.RepeatedCompositeFieldContainer[Codebase]
+    next_page_token: str
+    def __init__(self, codebases: _Optional[_Iterable[_Union[Codebase, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ...
+
+class DeleteCodebaseRequest(_message.Message):
+    __slots__ = ("codebase_id",)
+    CODEBASE_ID_FIELD_NUMBER: _ClassVar[int]
+    codebase_id: str
+    def __init__(self, codebase_id: _Optional[str] = ...) -> None: ...
+
+class UploadChunk(_message.Message):
+    __slots__ = ("metadata", "data")
+    class Metadata(_message.Message):
+        __slots__ = ("codebase_id", "file_path", "total_size")
+        CODEBASE_ID_FIELD_NUMBER: _ClassVar[int]
+        FILE_PATH_FIELD_NUMBER: _ClassVar[int]
+        TOTAL_SIZE_FIELD_NUMBER: _ClassVar[int]
+        codebase_id: str
+        file_path: str
+        total_size: int
+        def __init__(self, codebase_id: _Optional[str] = ..., file_path: _Optional[str] = ..., total_size: _Optional[int] = ...) -> None: ...
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    DATA_FIELD_NUMBER: _ClassVar[int]
+    metadata: UploadChunk.Metadata
+    data: bytes
+    def __init__(self, metadata: _Optional[_Union[UploadChunk.Metadata, _Mapping]] = ..., data: _Optional[bytes] = ...) -> None: ...
+
+class UploadResult(_message.Message):
+    __slots__ = ("codebase_id", "file_path", "size", "checksum")
+    CODEBASE_ID_FIELD_NUMBER: _ClassVar[int]
+    FILE_PATH_FIELD_NUMBER: _ClassVar[int]
+    SIZE_FIELD_NUMBER: _ClassVar[int]
+    CHECKSUM_FIELD_NUMBER: _ClassVar[int]
+    codebase_id: str
+    file_path: str
+    size: int
+    checksum: str
+    def __init__(self, codebase_id: _Optional[str] = ..., file_path: _Optional[str] = ..., size: _Optional[int] = ..., checksum: _Optional[str] = ...) -> None: ...
+
+class DownloadFileRequest(_message.Message):
+    __slots__ = ("codebase_id", "file_path")
+    CODEBASE_ID_FIELD_NUMBER: _ClassVar[int]
+    FILE_PATH_FIELD_NUMBER: _ClassVar[int]
+    codebase_id: str
+    file_path: str
+    def __init__(self, codebase_id: _Optional[str] = ..., file_path: _Optional[str] = ...) -> None: ...
+
+class FileChunk(_message.Message):
+    __slots__ = ("data",)
+    DATA_FIELD_NUMBER: _ClassVar[int]
+    data: bytes
+    def __init__(self, data: _Optional[bytes] = ...) -> None: ...
+
+class ListFilesRequest(_message.Message):
+    __slots__ = ("codebase_id", "path", "recursive")
+    CODEBASE_ID_FIELD_NUMBER: _ClassVar[int]
+    PATH_FIELD_NUMBER: _ClassVar[int]
+    RECURSIVE_FIELD_NUMBER: _ClassVar[int]
+    codebase_id: str
+    path: str
+    recursive: bool
+    def __init__(self, codebase_id: _Optional[str] = ..., path: _Optional[str] = ..., recursive: bool = ...) -> None: ...
+
+class ListFilesResponse(_message.Message):
+    __slots__ = ("files",)
+    FILES_FIELD_NUMBER: _ClassVar[int]
+    files: _containers.RepeatedCompositeFieldContainer[FileInfo]
+    def __init__(self, files: _Optional[_Iterable[_Union[FileInfo, _Mapping]]] = ...) -> None: ...
+
+class FileInfo(_message.Message):
+    __slots__ = ("path", "name", "is_dir", "size", "modified_at")
+    PATH_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    IS_DIR_FIELD_NUMBER: _ClassVar[int]
+    SIZE_FIELD_NUMBER: _ClassVar[int]
+    MODIFIED_AT_FIELD_NUMBER: _ClassVar[int]
+    path: str
+    name: str
+    is_dir: bool
+    size: int
+    modified_at: _timestamp_pb2.Timestamp
+    def __init__(self, path: _Optional[str] = ..., name: _Optional[str] = ..., is_dir: bool = ..., size: _Optional[int] = ..., modified_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...

agentfense/_gen/codebase_pb2_grpc.py

@@ -0,0 +1,366 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+from . import codebase_pb2 as codebase__pb2
+from . import common_pb2 as common__pb2
+
+GRPC_GENERATED_VERSION = '1.76.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + ' but the generated code in codebase_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )
+
+
+class CodebaseServiceStub(object):
+    """CodebaseService manages user codebases (file folders)
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.CreateCodebase = channel.unary_unary(
+                '/sandbox.v1.CodebaseService/CreateCodebase',
+                request_serializer=codebase__pb2.CreateCodebaseRequest.SerializeToString,
+                response_deserializer=codebase__pb2.Codebase.FromString,
+                _registered_method=True)
+        self.GetCodebase = channel.unary_unary(
+                '/sandbox.v1.CodebaseService/GetCodebase',
+                request_serializer=codebase__pb2.GetCodebaseRequest.SerializeToString,
+                response_deserializer=codebase__pb2.Codebase.FromString,
+                _registered_method=True)
+        self.ListCodebases = channel.unary_unary(
+                '/sandbox.v1.CodebaseService/ListCodebases',
+                request_serializer=codebase__pb2.ListCodebasesRequest.SerializeToString,
+                response_deserializer=codebase__pb2.ListCodebasesResponse.FromString,
+                _registered_method=True)
+        self.DeleteCodebase = channel.unary_unary(
+                '/sandbox.v1.CodebaseService/DeleteCodebase',
+                request_serializer=codebase__pb2.DeleteCodebaseRequest.SerializeToString,
+                response_deserializer=common__pb2.Empty.FromString,
+                _registered_method=True)
+        self.UploadFiles = channel.stream_unary(
+                '/sandbox.v1.CodebaseService/UploadFiles',
+                request_serializer=codebase__pb2.UploadChunk.SerializeToString,
+                response_deserializer=codebase__pb2.UploadResult.FromString,
+                _registered_method=True)
+        self.DownloadFile = channel.unary_stream(
+                '/sandbox.v1.CodebaseService/DownloadFile',
+                request_serializer=codebase__pb2.DownloadFileRequest.SerializeToString,
+                response_deserializer=codebase__pb2.FileChunk.FromString,
+                _registered_method=True)
+        self.ListFiles = channel.unary_unary(
+                '/sandbox.v1.CodebaseService/ListFiles',
+                request_serializer=codebase__pb2.ListFilesRequest.SerializeToString,
+                response_deserializer=codebase__pb2.ListFilesResponse.FromString,
+                _registered_method=True)
+
+
+class CodebaseServiceServicer(object):
+    """CodebaseService manages user codebases (file folders)
+    """
+
+    def CreateCodebase(self, request, context):
+        """CreateCodebase creates a new codebase
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetCodebase(self, request, context):
+        """GetCodebase retrieves information about a codebase
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def ListCodebases(self, request, context):
+        """ListCodebases lists all codebases for an owner
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def DeleteCodebase(self, request, context):
+        """DeleteCodebase deletes a codebase
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def UploadFiles(self, request_iterator, context):
+        """UploadFiles uploads files to a codebase
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def DownloadFile(self, request, context):
+        """DownloadFile downloads a file from a codebase
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def ListFiles(self, request, context):
+        """ListFiles lists files in a codebase directory
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_CodebaseServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'CreateCodebase': grpc.unary_unary_rpc_method_handler(
+                    servicer.CreateCodebase,
+                    request_deserializer=codebase__pb2.CreateCodebaseRequest.FromString,
+                    response_serializer=codebase__pb2.Codebase.SerializeToString,
+            ),
+            'GetCodebase': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetCodebase,
+                    request_deserializer=codebase__pb2.GetCodebaseRequest.FromString,
+                    response_serializer=codebase__pb2.Codebase.SerializeToString,
+            ),
+            'ListCodebases': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListCodebases,
+                    request_deserializer=codebase__pb2.ListCodebasesRequest.FromString,
+                    response_serializer=codebase__pb2.ListCodebasesResponse.SerializeToString,
+            ),
+            'DeleteCodebase': grpc.unary_unary_rpc_method_handler(
+                    servicer.DeleteCodebase,
+                    request_deserializer=codebase__pb2.DeleteCodebaseRequest.FromString,
+                    response_serializer=common__pb2.Empty.SerializeToString,
+            ),
+            'UploadFiles': grpc.stream_unary_rpc_method_handler(
+                    servicer.UploadFiles,
+                    request_deserializer=codebase__pb2.UploadChunk.FromString,
+                    response_serializer=codebase__pb2.UploadResult.SerializeToString,
+            ),
+            'DownloadFile': grpc.unary_stream_rpc_method_handler(
+                    servicer.DownloadFile,
+                    request_deserializer=codebase__pb2.DownloadFileRequest.FromString,
+                    response_serializer=codebase__pb2.FileChunk.SerializeToString,
+            ),
+            'ListFiles': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListFiles,
+                    request_deserializer=codebase__pb2.ListFilesRequest.FromString,
+                    response_serializer=codebase__pb2.ListFilesResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'sandbox.v1.CodebaseService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('sandbox.v1.CodebaseService', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class CodebaseService(object):
+    """CodebaseService manages user codebases (file folders)
+    """
+
+    @staticmethod
+    def CreateCodebase(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/sandbox.v1.CodebaseService/CreateCodebase',
+            codebase__pb2.CreateCodebaseRequest.SerializeToString,
+            codebase__pb2.Codebase.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def GetCodebase(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/sandbox.v1.CodebaseService/GetCodebase',
+            codebase__pb2.GetCodebaseRequest.SerializeToString,
+            codebase__pb2.Codebase.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def ListCodebases(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/sandbox.v1.CodebaseService/ListCodebases',
+            codebase__pb2.ListCodebasesRequest.SerializeToString,
+            codebase__pb2.ListCodebasesResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def DeleteCodebase(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/sandbox.v1.CodebaseService/DeleteCodebase',
+            codebase__pb2.DeleteCodebaseRequest.SerializeToString,
+            common__pb2.Empty.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def UploadFiles(request_iterator,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.stream_unary(
+            request_iterator,
+            target,
+            '/sandbox.v1.CodebaseService/UploadFiles',
+            codebase__pb2.UploadChunk.SerializeToString,
+            codebase__pb2.UploadResult.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def DownloadFile(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_stream(
+            request,
+            target,
+            '/sandbox.v1.CodebaseService/DownloadFile',
+            codebase__pb2.DownloadFileRequest.SerializeToString,
+            codebase__pb2.FileChunk.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def ListFiles(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/sandbox.v1.CodebaseService/ListFiles',
+            codebase__pb2.ListFilesRequest.SerializeToString,
+            codebase__pb2.ListFilesResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

agentfense/_gen/common_pb2.py

@@ -0,0 +1,47 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: common.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'common.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0c\x63ommon.proto\x12\nsandbox.v1\"\x86\x01\n\x0ePermissionRule\x12\x0f\n\x07pattern\x18\x01 \x01(\t\x12%\n\x04type\x18\x02 \x01(\x0e\x32\x17.sandbox.v1.PatternType\x12*\n\npermission\x18\x03 \x01(\x0e\x32\x16.sandbox.v1.Permission\x12\x10\n\x08priority\x18\x04 \x01(\x05\"\x07\n\x05\x45mpty*}\n\nPermission\x12\x1a\n\x16PERMISSION_UNSPECIFIED\x10\x00\x12\x13\n\x0fPERMISSION_NONE\x10\x01\x12\x13\n\x0fPERMISSION_VIEW\x10\x02\x12\x13\n\x0fPERMISSION_READ\x10\x03\x12\x14\n\x10PERMISSION_WRITE\x10\x04*u\n\x0bPatternType\x12\x1c\n\x18PATTERN_TYPE_UNSPECIFIED\x10\x00\x12\x15\n\x11PATTERN_TYPE_GLOB\x10\x01\x12\x1a\n\x16PATTERN_TYPE_DIRECTORY\x10\x02\x12\x15\n\x11PATTERN_TYPE_FILE\x10\x03*\x9d\x01\n\rSandboxStatus\x12\x1e\n\x1aSANDBOX_STATUS_UNSPECIFIED\x10\x00\x12\x1a\n\x16SANDBOX_STATUS_PENDING\x10\x01\x12\x1a\n\x16SANDBOX_STATUS_RUNNING\x10\x02\x12\x1a\n\x16SANDBOX_STATUS_STOPPED\x10\x03\x12\x18\n\x14SANDBOX_STATUS_ERROR\x10\x04*e\n\rSessionStatus\x12\x1e\n\x1aSESSION_STATUS_UNSPECIFIED\x10\x00\x12\x19\n\x15SESSION_STATUS_ACTIVE\x10\x01\x12\x19\n\x15SESSION_STATUS_CLOSED\x10\x02\x42=Z;github.com/AjaxZhan/AgentFense/api/gen/sandbox/v1;sandboxv1b\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'common_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z;github.com/AjaxZhan/AgentFense/api/gen/sandbox/v1;sandboxv1'
+  _globals['_PERMISSION']._serialized_start=174
+  _globals['_PERMISSION']._serialized_end=299
+  _globals['_PATTERNTYPE']._serialized_start=301
+  _globals['_PATTERNTYPE']._serialized_end=418
+  _globals['_SANDBOXSTATUS']._serialized_start=421
+  _globals['_SANDBOXSTATUS']._serialized_end=578
+  _globals['_SESSIONSTATUS']._serialized_start=580
+  _globals['_SESSIONSTATUS']._serialized_end=681
+  _globals['_PERMISSIONRULE']._serialized_start=29
+  _globals['_PERMISSIONRULE']._serialized_end=163
+  _globals['_EMPTY']._serialized_start=165
+  _globals['_EMPTY']._serialized_end=172
+# @@protoc_insertion_point(module_scope)