66 lines
1.6 KiB
Go
66 lines
1.6 KiB
Go
package plugin
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/cenkalti/backoff/v4"
|
|
"github.com/rs/zerolog/log"
|
|
plugin_exec "github.com/thegeeklab/wp-plugin-go/v3/exec"
|
|
)
|
|
|
|
const (
|
|
netrcFile = `machine %s
|
|
login %s
|
|
password %s
|
|
`
|
|
strictFilePerm = 0o600
|
|
)
|
|
|
|
// shouldRetry returns true if the command should be re-executed. Currently
|
|
// this only returns true if the remote ref does not exist.
|
|
func shouldRetry(s string) bool {
|
|
return strings.Contains(s, "find remote ref")
|
|
}
|
|
|
|
func newBackoff(maxRetries uint64) backoff.BackOff {
|
|
b := backoff.NewExponentialBackOff()
|
|
b.InitialInterval = daemonBackoffInitialInterval
|
|
b.Multiplier = daemonBackoffMultiplier
|
|
|
|
return backoff.WithMaxRetries(b, maxRetries)
|
|
}
|
|
|
|
func retryCmd(cmd *plugin_exec.Cmd) error {
|
|
backoffOps := func() error {
|
|
// copy the original command
|
|
retry := plugin_exec.Command(cmd.Cmd.Path, cmd.Cmd.Args...)
|
|
retry.Env = cmd.Env
|
|
retry.Stdout = cmd.Stdout
|
|
retry.Stderr = cmd.Stderr
|
|
retry.Dir = cmd.Dir
|
|
|
|
return retry.Run()
|
|
}
|
|
backoffLog := func(err error, delay time.Duration) {
|
|
log.Error().Msgf("failed to find remote ref: %v: retry in %s", err, delay.Truncate(time.Second))
|
|
}
|
|
|
|
return backoff.RetryNotify(backoffOps, newBackoff(daemonBackoffMaxRetries), backoffLog)
|
|
}
|
|
|
|
// WriteNetrc writes the netrc file.
|
|
func WriteNetrc(path, machine, login, password string) error {
|
|
netrcPath := filepath.Join(path, ".netrc")
|
|
netrcContent := fmt.Sprintf(netrcFile, machine, login, password)
|
|
|
|
if err := os.WriteFile(netrcPath, []byte(netrcContent), strictFilePerm); err != nil {
|
|
return fmt.Errorf("failed to create .netrc file: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|