feat: add local cache

3 files changed, 11 insertions, 1 deletions

diff --git a/internal/client/input/b2.go b/internal/client/input/b2.go
index db68268..fe94333 100644
--- a/internal/client/input/b2.go
+++ b/internal/client/input/b2.go
@@ -16,6 +16,7 @@ var _ InputClient = (*B2InputClient)(nil)
 type B2InputClient struct {
 	prefix          string
 	bucket          *b2.Bucket
+	bucketName      string
 	b2cl            *b2.Client
 	knownExtensions []string
 }
@@ -36,7 +37,7 @@ func NewB2InputClient(cfg *config.InputConfig) (InputClient, error) {
 		return nil, err
 	}
 
-	return &B2InputClient{b2cl: b2cl, bucket: bucket, prefix: b2cfg.Prefix, knownExtensions: cfg.KnownExtensions}, nil
+	return &B2InputClient{b2cl: b2cl, bucket: bucket, bucketName: b2cfg.BucketName, prefix: b2cfg.Prefix, knownExtensions: cfg.KnownExtensions}, nil
 }
 
 func (c *B2InputClient) Scan() ([]string, error) {
@@ -120,6 +121,10 @@ func (c *B2InputClient) ReadMetadata(path string) (*MetadataStruct, error) {
 	return &metadata, nil
 }
 
+func (c *B2InputClient) ID(path string) string {
+	return fmt.Sprintf("b2://%s/%s%s", c.bucketName, c.prefix, path)
+}
+
 func (c *B2InputClient) GetReader(path string) (io.ReadCloser, error) {
 	obj := c.bucket.Object(c.prefix + path)
 	if obj == nil {
diff --git a/internal/client/input/input_client_interface.go b/internal/client/input/input_client_interface.go
index 4930aa0..a2cf84b 100644
--- a/internal/client/input/input_client_interface.go
+++ b/internal/client/input/input_client_interface.go
@@ -11,6 +11,7 @@ type InputClient interface {
 	Scan() ([]string, error)
 	ReadMetadata(string) (*MetadataStruct, error)
 	GetReader(string) (io.ReadCloser, error)
+	ID(path string) string
 }
 
 type MetadataStruct struct {
diff --git a/internal/client/input/local_unix.go b/internal/client/input/local_unix.go
index e509c8b..21b08a9 100644
--- a/internal/client/input/local_unix.go
+++ b/internal/client/input/local_unix.go
@@ -99,6 +99,10 @@ func (c *LocalUnixInputClient) ReadMetadata(path string) (*MetadataStruct, error
 	}, nil
 }
 
+func (c *LocalUnixInputClient) ID(path string) string {
+	return fmt.Sprintf("local-unix://%s%s", c.path, path)
+}
+
 func (c *LocalUnixInputClient) GetReader(path string) (io.ReadCloser, error) {
 	return os.Open(c.path + path)
 }