Working field fetch

client.go

@@ -1,12 +1,9 @@
 package airtable
 
 import (
-	"bytes"
-	"context"
-	"encoding/json"
 	"fmt"
 	"net/http"
-	"net/url"
+	"os"
 	"time"
 )
 
@@ -34,122 +31,19 @@ func New(apiKey string) *Client {
 	return c
 }
 
-func (at *Client) SetRateLimit(rateLimit int) {
-	at.rateLimiter = time.Tick(time.Second / time.Duration(rateLimit))
-}
-
-func (at *Client) waitForRateLimit() {
-	<-at.rateLimiter
-}
-
-func (at *Client) get(ctx context.Context, db, table, recordID string, params url.Values, target any) error {
-	return at.do(ctx, "GET", db, table, recordID, params, nil, target)
-}
-
-func (at *Client) post(ctx context.Context, db, table string, data, target any) error {
-	return at.do(ctx, "POST", db, table, "", nil, data, target)
-}
-
-func (at *Client) delete(ctx context.Context, db, table string, recordIDs []string, target any) error {
-	params := url.Values{}
-
-	for _, recordID := range recordIDs {
-		params.Add("records[]", recordID)
+func NewFromEnv() (*Client, error) {
+	apiKey := os.Getenv("AIRTABLE_TOKEN")
+	if apiKey == "" {
+		return nil, fmt.Errorf("please set $AIRTABLE_TOKEN")
 	}
 
-	return at.do(ctx, "DELETE", db, table, "", params, nil, target)
+	return New(apiKey), nil
 }
 
-func (at *Client) patch(ctx context.Context, db, table string, data, target any) error {
-	return at.do(ctx, "PATCH", db, table, "", nil, data, target)
+func (c *Client) SetRateLimit(rateLimit int) {
+	c.rateLimiter = time.Tick(time.Second / time.Duration(rateLimit))
 }
 
-func (at *Client) put(ctx context.Context, db, table string, data, target any) error {
-	return at.do(ctx, "PUT", db, table, "", nil, data, target)
-}
-
-func (at *Client) do(ctx context.Context, method, db, table, recordID string, params url.Values, data, target any) error {
-	var err error
-
-	at.waitForRateLimit()
-
-	url := fmt.Sprintf("%s/%s/%s", at.BaseURL, db, table)
-
-	if recordID != "" {
-		url += fmt.Sprintf("/%s", recordID)
-	}
-
-	body := []byte{}
-
-	if data != nil {
-		body, err = json.Marshal(data)
-		if err != nil {
-			return fmt.Errorf("marshalling message body: %w", err)
-		}
-	}
-
-	req, err := http.NewRequestWithContext(ctx, method, url, bytes.NewReader(body))
-	if err != nil {
-		return fmt.Errorf("cannot create request: %w", err)
-	}
-
-	req.URL.RawQuery = params.Encode()
-
-	req.Header.Set("Content-Type", "application/json")
-	req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", at.apiKey))
-
-	resp, err := at.Client.Do(req)
-	if err != nil {
-		return fmt.Errorf("http request failed: %w", err)
-	}
-
-	defer resp.Body.Close()
-
-	if resp.StatusCode < 200 || resp.StatusCode > 299 {
-		return makeHTTPClientError(url, resp)
-	}
-
-	dec := json.NewDecoder(resp.Body)
-
-	err = dec.Decode(target)
-	if err != nil {
-		return fmt.Errorf("json decode failed: %w", err)
-	}
-
-	return nil
-}
-
-func listAll[T any](ctx context.Context, c *Client, db, table string, params url.Values, key string) ([]*T, error) {
-	ret := []*T{}
-
-	for {
-		resp := map[string]any{}
-
-		err := c.get(ctx, db, table, "", params, &resp)
-		if err != nil {
-			return nil, err
-		}
-
-		subresp, err := json.Marshal(resp[key])
-		if err != nil {
-			return nil, err
-		}
-
-		obj := []*T{}
-
-		err = json.Unmarshal(subresp, &obj)
-		if err != nil {
-			return nil, err
-		}
-
-		ret = append(ret, obj...)
-
-		off, found := resp["offset"]
-		if !found {
-			return ret, nil
-		}
-
-		params.Set("offset", off.(string))
-
-	}
+func (c *Client) waitForRateLimit() {
+	<-c.rateLimiter
 }