Diffstat (limited to 'internal/storage/b2.go')
| -rw-r--r-- | internal/storage/b2.go | 142 |
1 files changed, 115 insertions, 27 deletions
diff --git a/internal/storage/b2.go b/internal/storage/b2.go index 6697023..b02261f 100644 --- a/internal/storage/b2.go +++ b/internal/storage/b2.go @@ -3,6 +3,7 @@ package storage import ( "context" "fmt" + "io" "strconv" "strings" "time" @@ -15,12 +16,13 @@ import ( var _ StorageClient = &B2StorageClient{} type B2StorageClient struct { - prefix string - bucket *b2.Bucket - b2cl *b2.Client + prefix string + bucket *b2.Bucket + b2cl *b2.Client + draftModeCfg *config.DraftModeConfig } -func NewB2StorageClient(cfg *config.StorageConfig) (StorageClient, error) { +func NewB2StorageClient(cfg *config.StorageConfig, draftModeCfg *config.DraftModeConfig) (StorageClient, error) { if cfg.Type != "b2" { return nil, fmt.Errorf("invalid storage type for B2InputClient") } @@ -36,25 +38,73 @@ func NewB2StorageClient(cfg *config.StorageConfig) (StorageClient, error) { return nil, err } - return &B2StorageClient{b2cl: b2cl, bucket: bucket, prefix: b2cfg.Prefix}, nil + draftModeCfgCopy := *draftModeCfg + + return &B2StorageClient{b2cl: b2cl, bucket: bucket, prefix: b2cfg.Prefix, draftModeCfg: &draftModeCfgCopy}, nil } -func (sc *B2StorageClient) GetMetadata(key string) (map[string]string, error) { - obj := sc.bucket.Object(sc.prefix + key) +func (sc *B2StorageClient) Scan() ([]string, error) { + filePaths := []string{} + + iter := sc.bucket.List(context.Background(), b2.ListPrefix(sc.prefix)) + + for iter.Next() { + obj := iter.Object() + if obj == nil { + return nil, fmt.Errorf("failed to reference object in B2 bucket") + } + + attrs, err := obj.Attrs(context.Background()) + if err != nil { + return nil, fmt.Errorf("get attributes for object: %w", err) + } + + if attrs.Status != b2.Uploaded { + continue + } + + name := obj.Name() + if !strings.HasSuffix(name, ".md") { + continue + } + + if sc.draftModeCfg.Enabled && !strings.HasSuffix(name, sc.draftModeCfg.DraftSuffix) { + continue + } + + filePaths = append(filePaths, strings.TrimPrefix(name, sc.prefix)) + } + + if err := iter.Err(); err != nil { + return nil, fmt.Errorf("iterate over B2 objects: %w", err) + } + + return filePaths, nil +} + +func (sc *B2StorageClient) GetReader(path string) (io.ReadCloser, int64, error) { + obj := sc.bucket.Object(sc.prefix + path) if obj == nil { - return nil, fmt.Errorf("failed to reference object in B2 bucket") + return nil, 0, fmt.Errorf("failed to reference object in B2 bucket") } attrs, err := obj.Attrs(context.Background()) if err != nil { - if b2.IsNotExist(err) { - return map[string]string{}, nil - } - return nil, fmt.Errorf("get attributes of B2 object %q: %w", sc.prefix+key, err) + return nil, 0, fmt.Errorf("error getting attributes of an object: %w", err) } - return attrs.Info, nil + + return obj.NewReader(context.Background()), attrs.Size, nil } -func (sc *B2StorageClient) Put(key string, content []byte, metadata *frontmatter.Metadata) error { +func (sc *B2StorageClient) WriteMetadata(path string, metadata *frontmatter.Metadata) error { + draft := sc.bucket.Object(sc.prefix + path) + if draft == nil { + return fmt.Errorf("failed to reference draft object in B2 bucket") + } + draftAttrs, err := draft.Attrs(context.Background()) + if err != nil { + return fmt.Errorf("error getting attributes of a draft object: %w", err) + } + geolocationParts := strings.Split(metadata.Geolocation, " ") if (len(geolocationParts) == 1 && geolocationParts[0] != "") || len(geolocationParts) >= 4 { return fmt.Errorf("invalid geolocation format, expecting '{x} {y} [areaError]' or an empty string") @@ -81,19 +131,32 @@ func (sc *B2StorageClient) Put(key string, content []byte, metadata *frontmatter attrs := &b2.Attrs{ ContentType: "text/markdown; charset=utf-8", Info: map[string]string{ - "title": metadata.Title, - "short-description": metadata.ShortDescription, - "action-date": metadata.ActionDate, - "published-time": metadata.PublishedTime.Format(time.RFC3339), - "thumbnail": metadata.Thumbnail, - "tags": strings.Join(metadata.Tags, ","), - "geolocation": metadata.Geolocation, - "medley": medley, + "title": metadata.Title, + "short-description": metadata.ShortDescription, + "action-date": metadata.ActionDate, + "published-time": metadata.PublishedTime.Format(time.RFC3339), + "thumbnail": metadata.Thumbnail, + "tags": strings.Join(metadata.Tags, ","), + "geolocation": metadata.Geolocation, + "medley": medley, + "metadata-last-update-sha1": draftAttrs.SHA1, }} - prod := sc.bucket.Object(sc.prefix + key) - if prod == nil { - return fmt.Errorf("failed to reference prod object in B2 bucket") + reader := draft.NewReader(context.Background()) + content := make([]byte, draftAttrs.Size) + if _, err = reader.Read(content); err != nil { + return fmt.Errorf("failed to read a draft object back for writing (required for attribute setting): %w", err) + } + + var prod *b2.Object + if sc.draftModeCfg.Enabled { + prodPath := strings.TrimSuffix(path, sc.draftModeCfg.DraftSuffix) + sc.draftModeCfg.ProdSuffix + prod = sc.bucket.Object(sc.prefix + prodPath) + if prod == nil { + return fmt.Errorf("failed to reference prod object in B2 bucket") + } + } else { + prod = draft } writer := prod.NewWriter(context.Background(), b2.WithAttrsOption(attrs)) @@ -105,6 +168,31 @@ func (sc *B2StorageClient) Put(key string, content []byte, metadata *frontmatter return nil } -func (sc *B2StorageClient) BuildIndex() error { - return fmt.Errorf("BuildIndex not implemented for B2 storage") +func (sc *B2StorageClient) CompareDraftAndProd(path string) (changed bool) { + draft := sc.bucket.Object(sc.prefix + path) + if draft == nil { + return false + } + + prodPath := strings.TrimSuffix(path, sc.draftModeCfg.DraftSuffix) + sc.draftModeCfg.ProdSuffix + prod := sc.bucket.Object(sc.prefix + prodPath) + if prod == nil { + return true + } + + draftAttrs, err := draft.Attrs(context.Background()) + if err != nil { + return false + } + prodAttrs, err := prod.Attrs(context.Background()) + if err != nil { + return true + } + + lastUpdateSha1, ok := prodAttrs.Info["metadata-last-update-sha1"] + if !ok { + return true + } + + return draftAttrs.SHA1 != lastUpdateSha1 } |