whatcanGOwrong

go/pkg/mod/github.com/golang-migrate/migrate/v4@v4.17.1/source/bitbucket/.gitignore

@@ -0,0 +1 @@
+.bitbucket_test_secrets

go/pkg/mod/github.com/golang-migrate/migrate/v4@v4.17.1/source/bitbucket/README.md

@@ -0,0 +1,14 @@
+# bitbucket
+
+This driver is catered for those that want to source migrations from bitbucket cloud(https://bitbucket.com).
+
+`bitbucket://user:password@owner/repo/path#ref`
+
+| URL Query  | WithInstance Config | Description |
+|------------|---------------------|-------------|
+| user | | The username of the user connecting |
+| password | | User's password or an app password with repo read permission  |
+| owner | | the repo owner |
+| repo | | the name of the repository |
+| path | | path in repo to migrations |
+| ref | | (optional) can be a SHA, branch, or tag |

go/pkg/mod/github.com/golang-migrate/migrate/v4@v4.17.1/source/bitbucket/bitbucket.go

@@ -0,0 +1,207 @@
+package bitbucket
+
+import (
+	"fmt"
+	"io"
+	nurl "net/url"
+	"os"
+	"path"
+	"path/filepath"
+	"strings"
+
+	"github.com/golang-migrate/migrate/v4/source"
+	"github.com/ktrysmt/go-bitbucket"
+)
+
+func init() {
+	source.Register("bitbucket", &Bitbucket{})
+}
+
+var (
+	ErrNoUserInfo             = fmt.Errorf("no username:password provided")
+	ErrNoAccessToken          = fmt.Errorf("no password/app password")
+	ErrInvalidRepo            = fmt.Errorf("invalid repo")
+	ErrInvalidBitbucketClient = fmt.Errorf("expected *bitbucket.Client")
+	ErrNoDir                  = fmt.Errorf("no directory")
+)
+
+type Bitbucket struct {
+	config     *Config
+	client     *bitbucket.Client
+	migrations *source.Migrations
+}
+
+type Config struct {
+	Owner string
+	Repo  string
+	Path  string
+	Ref   string
+}
+
+func (b *Bitbucket) Open(url string) (source.Driver, error) {
+	u, err := nurl.Parse(url)
+	if err != nil {
+		return nil, err
+	}
+
+	if u.User == nil {
+		return nil, ErrNoUserInfo
+	}
+
+	password, ok := u.User.Password()
+	if !ok {
+		return nil, ErrNoAccessToken
+	}
+
+	cl := bitbucket.NewBasicAuth(u.User.Username(), password)
+
+	cfg := &Config{}
+	// set owner, repo and path in repo
+	cfg.Owner = u.Host
+	pe := strings.Split(strings.Trim(u.Path, "/"), "/")
+	if len(pe) < 1 {
+		return nil, ErrInvalidRepo
+	}
+	cfg.Repo = pe[0]
+	if len(pe) > 1 {
+		cfg.Path = strings.Join(pe[1:], "/")
+	}
+	cfg.Ref = u.Fragment
+
+	bi, err := WithInstance(cl, cfg)
+	if err != nil {
+		return nil, err
+	}
+
+	return bi, nil
+}
+
+func WithInstance(client *bitbucket.Client, config *Config) (source.Driver, error) {
+	bi := &Bitbucket{
+		client:     client,
+		config:     config,
+		migrations: source.NewMigrations(),
+	}
+
+	if err := bi.readDirectory(); err != nil {
+		return nil, err
+	}
+
+	return bi, nil
+}
+
+func (b *Bitbucket) readDirectory() error {
+	b.ensureFields()
+
+	fOpt := &bitbucket.RepositoryFilesOptions{
+		Owner:    b.config.Owner,
+		RepoSlug: b.config.Repo,
+		Ref:      b.config.Ref,
+		Path:     b.config.Path,
+	}
+
+	dirContents, err := b.client.Repositories.Repository.ListFiles(fOpt)
+
+	if err != nil {
+		return err
+	}
+
+	for _, fi := range dirContents {
+
+		m, err := source.DefaultParse(filepath.Base(fi.Path))
+		if err != nil {
+			continue // ignore files that we can't parse
+		}
+		if !b.migrations.Append(m) {
+			return fmt.Errorf("unable to parse file %v", fi.Path)
+		}
+	}
+
+	return nil
+}
+
+func (b *Bitbucket) ensureFields() {
+	if b.config == nil {
+		b.config = &Config{}
+	}
+}
+
+func (b *Bitbucket) Close() error {
+	return nil
+}
+
+func (b *Bitbucket) First() (version uint, er error) {
+	b.ensureFields()
+
+	if v, ok := b.migrations.First(); !ok {
+		return 0, &os.PathError{Op: "first", Path: b.config.Path, Err: os.ErrNotExist}
+	} else {
+		return v, nil
+	}
+}
+
+func (b *Bitbucket) Prev(version uint) (prevVersion uint, err error) {
+	b.ensureFields()
+
+	if v, ok := b.migrations.Prev(version); !ok {
+		return 0, &os.PathError{Op: fmt.Sprintf("prev for version %v", version), Path: b.config.Path, Err: os.ErrNotExist}
+	} else {
+		return v, nil
+	}
+}
+
+func (b *Bitbucket) Next(version uint) (nextVersion uint, err error) {
+	b.ensureFields()
+
+	if v, ok := b.migrations.Next(version); !ok {
+		return 0, &os.PathError{Op: fmt.Sprintf("next for version %v", version), Path: b.config.Path, Err: os.ErrNotExist}
+	} else {
+		return v, nil
+	}
+}
+
+func (b *Bitbucket) ReadUp(version uint) (r io.ReadCloser, identifier string, err error) {
+	b.ensureFields()
+
+	if m, ok := b.migrations.Up(version); ok {
+		fBlobOpt := &bitbucket.RepositoryBlobOptions{
+			Owner:    b.config.Owner,
+			RepoSlug: b.config.Repo,
+			Ref:      b.config.Ref,
+			Path:     path.Join(b.config.Path, m.Raw),
+		}
+		file, err := b.client.Repositories.Repository.GetFileBlob(fBlobOpt)
+		if err != nil {
+			return nil, "", err
+		}
+		if file != nil {
+			r := file.Content
+			return io.NopCloser(strings.NewReader(string(r))), m.Identifier, nil
+		}
+	}
+	return nil, "", &os.PathError{Op: fmt.Sprintf("read version %v", version), Path: b.config.Path, Err: os.ErrNotExist}
+}
+
+func (b *Bitbucket) ReadDown(version uint) (r io.ReadCloser, identifier string, err error) {
+	b.ensureFields()
+
+	if m, ok := b.migrations.Down(version); ok {
+		fBlobOpt := &bitbucket.RepositoryBlobOptions{
+			Owner:    b.config.Owner,
+			RepoSlug: b.config.Repo,
+			Ref:      b.config.Ref,
+			Path:     path.Join(b.config.Path, m.Raw),
+		}
+		file, err := b.client.Repositories.Repository.GetFileBlob(fBlobOpt)
+
+		if err != nil {
+			return nil, "", err
+		}
+		if file != nil {
+			r := file.Content
+
+			return io.NopCloser(strings.NewReader(string(r))), m.Identifier, nil
+		}
+	}
+	return nil, "", &os.PathError{Op: fmt.Sprintf("read version %v", version), Path: b.config.Path, Err: os.ErrNotExist}
+}

go/pkg/mod/github.com/golang-migrate/migrate/v4@v4.17.1/source/bitbucket/bitbucket_test.go

@@ -0,0 +1,33 @@
+package bitbucket
+
+import (
+	"bytes"
+	"os"
+	"testing"
+
+	st "github.com/golang-migrate/migrate/v4/source/testing"
+)
+
+var BitbucketTestSecret = "" // username:password
+
+func init() {
+	secrets, err := os.ReadFile(".bitbucket_test_secrets")
+	if err == nil {
+		BitbucketTestSecret = string(bytes.TrimSpace(secrets)[:])
+	}
+}
+
+func Test(t *testing.T) {
+	if len(BitbucketTestSecret) == 0 {
+		t.Skip("test requires .bitbucket_test_secrets")
+	}
+
+	b := &Bitbucket{}
+
+	d, err := b.Open("bitbucket://" + BitbucketTestSecret + "@abhishekbipp/test-migration/migrations/test#master")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	st.Test(t, d)
+}