Merge pull request #3 from rubiojr/api-keys

Add ApiKeys support

README.md

@@ -8,7 +8,7 @@ Supported queries:
 
 - [x] Search
 - [x] NewsletterEmails
-- [ ] ApiKeys
+- [x] ApiKeys
 - [ ] Users
 - [ ] Labels
 - [ ] Integrations

api.go

@@ -88,6 +88,16 @@ type SearchOpts struct {
 	Format         string
 }
 
+type ApiKey struct {
+	ID        graphql.ID
+	Key       string
+	CreatedAt time.Time
+	ExpiresAt string
+	Name      string
+	Scopes    []string
+	UsedAt    time.Time
+}
+
 func NewClient(opts Opts) *Omnivore {
 	return NewClientFor("https://api-prod.omnivore.app/api/graphql", opts)
 }
@@ -215,6 +225,33 @@ func (c *Omnivore) Subscriptions() ([]Subscription, error) {
 			Url:              sub.Url,
 		})
 	}
+	return a, nil
+}
+
+func (c *Omnivore) ApiKeys() ([]ApiKey, error) {
+	a := []ApiKey{}
+
+	err := c.graphql.Query(context.Background(), &queries.ApiKeys, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	result := queries.ApiKeys.ApiKeys.ApiKeysSuccess
+	for _, apiKey := range result.ApiKeys {
+		a = append(a, ApiKey{
+			ID:        apiKey.ID,
+			Key:       apiKey.Key,
+			CreatedAt: apiKey.CreatedAt,
+			ExpiresAt: apiKey.ExpiresAt,
+			Name:      apiKey.Name,
+			Scopes:    apiKey.Scopes,
+			UsedAt:    apiKey.UsedAt,
+		})
+	}
 
 	return a, nil
 }
+
+func (a *ApiKey) HasExpiry() bool {
+	return a.ExpiresAt != "+275760-09-13T00:00:00.000Z"
+}

api_test.go

@@ -29,3 +29,13 @@ func TestSubcriptions(t *testing.T) {
 	assert.Equal(t, len(subscriptions), 1)
 	assert.Equal(t, subscriptions[0].Name, "Omnivore Blog")
 }
+
+func TestApiKeys(t *testing.T) {
+	client := omnivore.NewClient(omnivore.Opts{Token: os.Getenv("OMNIVORE_API_TOKEN")})
+	keys, err := client.ApiKeys()
+	assert.NoError(t, err, "Failed to get api keys")
+	assert.Equal(t, len(keys), 1)
+	assert.Equal(t, keys[0].Name, "omnivore-go-github")
+	assert.Equal(t, keys[0].ExpiresAt, "+275760-09-13T00:00:00.000Z")
+	assert.False(t, keys[0].HasExpiry())
+}

queries/apikeys.go

@@ -0,0 +1,23 @@
+package queries
+
+import (
+	"time"
+
+	"github.com/shurcooL/graphql"
+)
+
+var ApiKeys struct {
+	ApiKeys struct {
+		ApiKeysSuccess struct {
+			ApiKeys []struct {
+				ID        graphql.ID
+				Key       string
+				CreatedAt time.Time
+				UsedAt    time.Time
+				ExpiresAt string
+				Scopes    []string
+				Name      string
+			}
+		} `graphql:"... on ApiKeysSuccess"`
+	}
+}