117 lines
3.8 KiB
Go
117 lines
3.8 KiB
Go
package publish
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"time"
|
|
|
|
"gitea.maximumdirect.net/eric/distributor/internal/config"
|
|
"gitea.maximumdirect.net/eric/distributor/internal/state"
|
|
"gitea.maximumdirect.net/eric/distributor/internal/storage"
|
|
)
|
|
|
|
func Execute(ctx context.Context, req Request, plan Plan) error {
|
|
switch plan.Action {
|
|
case ActionSkipSame, ActionSkipDestinationNewer:
|
|
return nil
|
|
case ActionPublishNew, ActionReplaceOlder, ActionForceReplace:
|
|
default:
|
|
return fmt.Errorf("cannot execute action %s: %s", plan.Action, plan.Reason)
|
|
}
|
|
|
|
if plan.Action == ActionReplaceOlder {
|
|
if plan.ExistingState == nil {
|
|
return fmt.Errorf("replace requires existing destination state")
|
|
}
|
|
if err := req.DestinationBackend.DeleteManagedBundle(ctx, req.DestinationBundlePath, stateOutputManagedPaths(plan.ExistingState.Outputs), storage.DeleteOptions{IgnoreMissing: true, PruneEmptyDirs: true}); err != nil {
|
|
return err
|
|
}
|
|
if err := ensureDestinationEmpty(ctx, req.DestinationBackend, req.DestinationBundlePath); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if plan.Action == ActionForceReplace {
|
|
if err := req.DestinationBackend.DeletePrefix(ctx, req.DestinationBundlePath, storage.DeleteOptions{IgnoreMissing: true, PruneEmptyDirs: true}); err != nil {
|
|
return err
|
|
}
|
|
if err := ensureDestinationEmpty(ctx, req.DestinationBackend, req.DestinationBundlePath); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
writtenOutputs := make([]Output, 0, len(plan.Outputs))
|
|
cleanup := func() {
|
|
_ = req.DestinationBackend.DeleteManagedBundle(ctx, req.DestinationBundlePath, ManagedOutputPaths(writtenOutputs), storage.DeleteOptions{IgnoreMissing: true, PruneEmptyDirs: true})
|
|
}
|
|
for _, output := range plan.Outputs {
|
|
destinationPath, err := storage.Join(req.DestinationBundlePath, output.DestinationPath)
|
|
if err != nil {
|
|
cleanup()
|
|
return err
|
|
}
|
|
data := output.Data
|
|
if output.Kind == state.OutputKindSource {
|
|
sourcePath, err := storage.Join(req.SourceBundle.RootRelativePath, output.SourcePath)
|
|
if err != nil {
|
|
cleanup()
|
|
return err
|
|
}
|
|
data, err = req.SourceBackend.ReadFile(ctx, sourcePath)
|
|
if err != nil {
|
|
cleanup()
|
|
return err
|
|
}
|
|
}
|
|
if _, err := req.DestinationBackend.WriteFile(ctx, destinationPath, data, storage.WriteOptions{Overwrite: false, PreferAtomic: true}); err != nil {
|
|
cleanup()
|
|
return err
|
|
}
|
|
writtenOutputs = append(writtenOutputs, output)
|
|
}
|
|
|
|
now := time.Now().UTC()
|
|
createdAt := now
|
|
existingOutputs := []state.OutputFile(nil)
|
|
if plan.ExistingState != nil {
|
|
createdAt = plan.ExistingState.CreatedAt
|
|
existingOutputs = plan.ExistingState.Outputs
|
|
}
|
|
destinationState := state.DistributorState{
|
|
SchemaVersion: state.SchemaVersion,
|
|
DistributorVersion: req.DistributorVersion,
|
|
PipelineID: req.PipelineID,
|
|
DestinationID: req.DestinationID,
|
|
PublishedAt: now,
|
|
CreatedAt: createdAt,
|
|
UpdatedAt: now,
|
|
State: state.StatePolicy{Mode: state.StateModeSingleOwner},
|
|
Reconciliation: state.ReconciliationPolicy{Mode: config.ReconciliationModeReplace},
|
|
Source: state.SourceState{Manifest: req.SourceBundle.Manifest},
|
|
Outputs: state.ProjectOutputs(StateOutputProjections(plan.Outputs), existingOutputs, now),
|
|
}
|
|
if plan.PrimaryURL != "" {
|
|
destinationState.Links = &state.LinkState{PrimaryURL: plan.PrimaryURL}
|
|
}
|
|
if err := state.Validate(destinationState); err != nil {
|
|
cleanup()
|
|
return err
|
|
}
|
|
data, err := json.MarshalIndent(destinationState, "", " ")
|
|
if err != nil {
|
|
cleanup()
|
|
return err
|
|
}
|
|
data = append(data, '\n')
|
|
statePath, err := storage.StatePath(req.DestinationBundlePath)
|
|
if err != nil {
|
|
cleanup()
|
|
return err
|
|
}
|
|
if _, err := req.DestinationBackend.WriteFile(ctx, statePath, data, storage.WriteOptions{Overwrite: false, PreferAtomic: true}); err != nil {
|
|
cleanup()
|
|
return err
|
|
}
|
|
return nil
|
|
}
|