diff --git a/go.mod b/go.mod index f4dd485..a73bd7a 100644 --- a/go.mod +++ b/go.mod @@ -1,3 +1,15 @@ module github.com/vasyansk/imap-copier -go 1.22 +go 1.22.0 + +require github.com/jackc/pgx/v5 v5.7.0 + +require ( + github.com/jackc/pgpassfile v1.0.0 // indirect + github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 // indirect + github.com/jackc/puddle/v2 v2.2.1 // indirect + github.com/stretchr/testify v1.9.0 // indirect + golang.org/x/crypto v0.31.0 // indirect + golang.org/x/sync v0.10.0 // indirect + golang.org/x/text v0.21.0 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..7bacfd8 --- /dev/null +++ b/go.sum @@ -0,0 +1,28 @@ +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM= +github.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg= +github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 h1:iCEnooe7UlwOQYpKFhBabPMi4aNAfoODPEFNiAnClxo= +github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761/go.mod h1:5TJZWKEWniPve33vlWYSoGYefn3gLQRzjfDlhSJ9ZKM= +github.com/jackc/pgx/v5 v5.7.0 h1:FG6VLIdzvAPhnYqP14sQ2xhFLkiUQHCs6ySqO91kF4g= +github.com/jackc/pgx/v5 v5.7.0/go.mod h1:awP1KNnjylvpxHuHP63gzjhnGkI1iw+PMoIwvoleN/8= +github.com/jackc/puddle/v2 v2.2.1 h1:RhxXJtFG022u4ibrCSMSiu5aOq1i77R3OHKNJj77OAk= +github.com/jackc/puddle/v2 v2.2.1/go.mod h1:vriiEXHvEE654aYKXXjOvZM39qJ0q+azkZFrfEOc3H4= +github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg= +github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= +golang.org/x/crypto v0.31.0 h1:ihbySMvVjLAeSH1IbfcRTkD/iNscyz8rGzjF/E5hV6U= +golang.org/x/crypto v0.31.0/go.mod h1:kDsLvtWBEx7MV9tJOj9bnXsPbxwJQ6csT/x4KIN4Ssk= +golang.org/x/sync v0.10.0 h1:3NQrjDixjgGwUOCaF8w2+VYHv0Ve/vGYSbdkTa98gmQ= +golang.org/x/sync v0.10.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk= +golang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo= +golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= diff --git a/internal/store/endpoints.go b/internal/store/endpoints.go new file mode 100644 index 0000000..e43a418 --- /dev/null +++ b/internal/store/endpoints.go @@ -0,0 +1,46 @@ +package store + +import "context" + +type Endpoint struct { + ID int64 + RoleLabel string + Host string + Port int + TLSMode string +} + +func (s *Store) CreateEndpoint(ctx context.Context, e Endpoint) (int64, error) { + var id int64 + err := s.Pool.QueryRow(ctx, + `INSERT INTO endpoints (role_label, host, port, tls_mode) + VALUES ($1,$2,$3,$4) RETURNING id`, + e.RoleLabel, e.Host, e.Port, e.TLSMode).Scan(&id) + return id, err +} + +func (s *Store) GetEndpoint(ctx context.Context, id int64) (Endpoint, error) { + var e Endpoint + err := s.Pool.QueryRow(ctx, + `SELECT id, role_label, host, port, tls_mode FROM endpoints WHERE id=$1`, id). + Scan(&e.ID, &e.RoleLabel, &e.Host, &e.Port, &e.TLSMode) + return e, err +} + +func (s *Store) ListEndpoints(ctx context.Context) ([]Endpoint, error) { + rows, err := s.Pool.Query(ctx, + `SELECT id, role_label, host, port, tls_mode FROM endpoints ORDER BY id`) + if err != nil { + return nil, err + } + defer rows.Close() + var out []Endpoint + for rows.Next() { + var e Endpoint + if err := rows.Scan(&e.ID, &e.RoleLabel, &e.Host, &e.Port, &e.TLSMode); err != nil { + return nil, err + } + out = append(out, e) + } + return out, rows.Err() +} diff --git a/internal/store/store.go b/internal/store/store.go new file mode 100644 index 0000000..da10279 --- /dev/null +++ b/internal/store/store.go @@ -0,0 +1,22 @@ +package store + +import ( + "context" + + "github.com/jackc/pgx/v5/pgxpool" +) + +type Store struct { + Pool *pgxpool.Pool +} + +func New(ctx context.Context, dsn string) (*Store, error) { + pool, err := pgxpool.New(ctx, dsn) + if err != nil { + return nil, err + } + if err := pool.Ping(ctx); err != nil { + return nil, err + } + return &Store{Pool: pool}, nil +} diff --git a/internal/store/store_test.go b/internal/store/store_test.go new file mode 100644 index 0000000..ef9d556 --- /dev/null +++ b/internal/store/store_test.go @@ -0,0 +1,40 @@ +package store + +import ( + "context" + "os" + "testing" +) + +func testStore(t *testing.T) *Store { + dsn := os.Getenv("TEST_DATABASE_URL") + if dsn == "" { + t.Skip("TEST_DATABASE_URL not set") + } + s, err := New(context.Background(), dsn) + if err != nil { + t.Fatalf("New: %v", err) + } + t.Cleanup(func() { + s.Pool.Exec(context.Background(), + `TRUNCATE endpoints, tasks, accounts, runs, migrated_messages RESTART IDENTITY CASCADE`) + s.Pool.Close() + }) + return s +} + +func TestCreateAndGetEndpoint(t *testing.T) { + s := testStore(t) + ctx := context.Background() + id, err := s.CreateEndpoint(ctx, Endpoint{RoleLabel: "src", Host: "imap.a.com", Port: 993, TLSMode: "ssl"}) + if err != nil { + t.Fatalf("create: %v", err) + } + got, err := s.GetEndpoint(ctx, id) + if err != nil { + t.Fatalf("get: %v", err) + } + if got.Host != "imap.a.com" || got.Port != 993 { + t.Fatalf("got %+v", got) + } +}