From 8e8b3cc5238693151978f5eefa2b6fac63adc50a Mon Sep 17 00:00:00 2001
From: SayaAndy <saya.andy@posteo.com>
Date: Sun, 24 Aug 2025 21:03:10 +0700
Subject: feat: add local cache

---
 internal/client/input/b2.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'internal/client/input/b2.go')

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 {
-- 
cgit v1.3.1+13