feat: Add Institutions

plugin/plugin.go

@@ -16,6 +16,7 @@ func Plugin() *source.Plugin {
 		"plaid",
 		Version,
 		schema.Tables{
+			resources.Institutions(),
 			resources.Wallets(),
 		},
 		client.New,

resources/institutions.go

@@ -0,0 +1,64 @@
+package resources
+
+import (
+	"context"
+
+	"github.com/cloudquery/cq-source-plaid/client"
+	"github.com/cloudquery/plugin-sdk/schema"
+	"github.com/cloudquery/plugin-sdk/transformers"
+	"github.com/plaid/plaid-go/v10/plaid"
+)
+
+const count = 500
+
+func Institutions() *schema.Table {
+	return &schema.Table{
+		Name:      "plaid_institutions",
+		Resolver:  fetchInstitutions,
+		Transform: transformers.TransformWithStruct(plaid.Institution{}, transformers.WithPrimaryKeys("InstitutionId")),
+	}
+}
+
+func fetchInstitutions(ctx context.Context, meta schema.ClientMeta, parent *schema.Resource, res chan<- interface{}) error {
+	client := meta.(*client.Client)
+
+	var total int32
+	for {
+		request := plaid.NewInstitutionsGetRequest(
+			count,
+			total,
+			[]plaid.CountryCode{
+				plaid.COUNTRYCODE_US,
+				plaid.COUNTRYCODE_GB,
+				plaid.COUNTRYCODE_ES,
+				plaid.COUNTRYCODE_NL,
+				plaid.COUNTRYCODE_FR,
+				plaid.COUNTRYCODE_IE,
+				plaid.COUNTRYCODE_CA,
+				plaid.COUNTRYCODE_DE,
+				plaid.COUNTRYCODE_IT,
+				plaid.COUNTRYCODE_PL,
+				plaid.COUNTRYCODE_DK,
+				plaid.COUNTRYCODE_NO,
+				plaid.COUNTRYCODE_SE,
+				plaid.COUNTRYCODE_EE,
+				plaid.COUNTRYCODE_LT,
+				plaid.COUNTRYCODE_LV,
+			},
+		)
+		resp, _, err := client.Services.PlaidApi.InstitutionsGet(ctx).InstitutionsGetRequest(*request).Execute()
+		if err != nil {
+			return err
+		}
+
+		institutions := resp.GetInstitutions()
+		total += int32(len(institutions))
+		res <- institutions
+
+		if total >= resp.GetTotal() || len(institutions) == 0 {
+			break
+		}
+	}
+
+	return nil
+}

resources/institutions_test.go

@@ -0,0 +1,34 @@
+package resources
+
+import (
+	"testing"
+
+	_ "embed"
+
+	"github.com/cloudquery/cq-source-plaid/client"
+	"github.com/cloudquery/plugin-sdk/faker"
+	"github.com/plaid/plaid-go/v10/plaid"
+)
+
+type institutionsResponse struct {
+	Institutions []plaid.Institution `json:"institutions"`
+}
+
+func TestInstitutions(t *testing.T) {
+	var res institutionsResponse
+	if err := faker.FakeObject(&res); err != nil {
+		t.Fatal(err)
+	}
+	testString := "test"
+	res.Institutions[0].Url.Set(&testString)
+	res.Institutions[0].Logo.Set(&testString)
+	res.Institutions[0].PrimaryColor.Set(&testString)
+	res.Institutions[0].Status.Set(&plaid.InstitutionStatus{})
+	res.Institutions[0].PaymentInitiationMetadata.Set(&plaid.PaymentInitiationMetadata{})
+	res.Institutions[0].AuthMetadata.Set(&plaid.AuthMetadata{})
+
+	ts := testServer(t, res)
+
+	defer ts.Close()
+	client.TestHelper(t, Institutions(), ts)
+}

resources/wallets_test.go

@@ -10,12 +10,12 @@ import (
 	"github.com/plaid/plaid-go/v10/plaid"
 )
 
-type response struct {
+type walletsResponse struct {
 	Wallets []plaid.Wallet `json:"wallets"`
 }
 
 func TestWallets(t *testing.T) {
-	var res response
+	var res walletsResponse
 	if err := faker.FakeObject(&res); err != nil {
 		t.Fatal(err)
 	}