Skip to content

Commit

Permalink
Add CertificatesService.
Browse files Browse the repository at this point in the history
  • Loading branch information
viola committed Feb 15, 2017
1 parent 7679760 commit 68caed3
Show file tree
Hide file tree
Showing 3 changed files with 291 additions and 0 deletions.
119 changes: 119 additions & 0 deletions certificates.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,119 @@
package godo

import (
"fmt"
)

const certificatesBasePath = "/v2/certificates"

// CertificatesService is an interface for managing certificates with the DigitalOcean API.
// See: https://developers.digitalocean.com/documentation/v2/#certificates
type CertificatesService interface {
Get(cID string) (*Certificate, *Response, error)
List(opt *ListOptions) ([]Certificate, *Response, error)
Create(cr *CertificateRequest) (*Certificate, *Response, error)
Delete(cID string) (*Response, error)
}

// Certificate represents a DigitalOcean certificate configuration.
type Certificate struct {
ID string `json:"id,omitempty"`
Name string `json:"name,omitempty"`
NotAfter string `json:"not_after,omitempty"`
SHA1Fingerprint string `json:"sha1_fingerprint,omitempty"`
Created string `json:"created_at,omitempty"`
}

// CertificateRequest represents configuration for a new certificate.
type CertificateRequest struct {
Name string `json:"name,omitempty"`
PrivateKey string `json:"private_key,omitempty"`
LeafCertificate string `json:"leaf_certificate,omitempty"`
CertificateChain string `json:"certificate_chain,omitempty"`
}

type certificateRoot struct {
Certificate *Certificate `json:"certificate"`
}

type certificatesRoot struct {
Certificates []Certificate `json:"certificates"`
Links *Links `json:"links"`
}

// CertificatesServiceOp handles communication with certificates methods of the DigitalOcean API.
type CertificatesServiceOp struct {
client *Client
}

var _ CertificatesService = &CertificatesServiceOp{}

// Get an existing certificate by its identifier.
func (c *CertificatesServiceOp) Get(cID string) (*Certificate, *Response, error) {
path := fmt.Sprintf("%s/%s", certificatesBasePath, cID)

req, err := c.client.NewRequest("GET", path, nil)
if err != nil {
return nil, nil, err
}

root := new(certificateRoot)
resp, err := c.client.Do(req, root)
if err != nil {
return nil, resp, err
}

return root.Certificate, resp, nil
}

// List all certificates.
func (c *CertificatesServiceOp) List(opt *ListOptions) ([]Certificate, *Response, error) {
path, err := addOptions(certificatesBasePath, opt)
if err != nil {
return nil, nil, err
}

req, err := c.client.NewRequest("GET", path, nil)
if err != nil {
return nil, nil, err
}

root := new(certificatesRoot)
resp, err := c.client.Do(req, root)
if err != nil {
return nil, resp, err
}
if l := root.Links; l != nil {
resp.Links = l
}

return root.Certificates, resp, err
}

// Create a new certificate with provided configuration.
func (c *CertificatesServiceOp) Create(cr *CertificateRequest) (*Certificate, *Response, error) {
req, err := c.client.NewRequest("POST", certificatesBasePath, cr)
if err != nil {
return nil, nil, err
}

root := new(certificateRoot)
resp, err := c.client.Do(req, root)
if err != nil {
return nil, resp, err
}

return root.Certificate, resp, nil
}

// Delete a certificate by its identifier.
func (c *CertificatesServiceOp) Delete(cID string) (*Response, error) {
path := fmt.Sprintf("%s/%s", certificatesBasePath, cID)

req, err := c.client.NewRequest("DELETE", path, nil)
if err != nil {
return nil, err
}

return c.client.Do(req, nil)
}
170 changes: 170 additions & 0 deletions certificates_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,170 @@
package godo

import (
"encoding/json"
"fmt"
"net/http"
"testing"

"github.com/stretchr/testify/assert"
)

var certificateJSONResponse = `
{
"certificate": {
"id": "892071a0-bb95-49bc-8021-3afd67a210bf",
"name": "web-cert-01",
"not_after": "2017-02-22T00:23:00Z",
"sha1_fingerprint": "dfcc9f57d86bf58e321c2c6c31c7a971be244ac7",
"created_at": "2017-02-08T16:02:37Z"
}
}
`

var cListJSONResponse = `
{
"certificates": [
{
"id": "892071a0-bb95-49bc-8021-3afd67a210bf",
"name": "web-cert-01",
"not_after": "2017-02-22T00:23:00Z",
"sha1_fingerprint": "dfcc9f57d86bf58e321c2c6c31c7a971be244ac7",
"created_at": "2017-02-08T16:02:37Z"
},
{
"id": "992071a0-bb95-49bc-8021-3afd67a210bf",
"name": "web-cert-02",
"not_after": "2017-02-22T00:23:00Z",
"sha1_fingerprint": "cfcc9f57d86bf58e321c2c6c31c7a971be244ac7",
"created_at": "2017-02-08T16:02:37Z"
}
],
"links": {},
"meta": {
"total": 1
}
}
`

func TestCertificates_Get(t *testing.T) {
setup()
defer teardown()

path := "/v2/certificates"
cID := "892071a0-bb95-49bc-8021-3afd67a210bf"
path = fmt.Sprintf("%s/%s", path, cID)
mux.HandleFunc(path, func(w http.ResponseWriter, r *http.Request) {
testMethod(t, r, "GET")
fmt.Fprint(w, certificateJSONResponse)
})

certificate, _, err := client.Certificates.Get(cID)
if err != nil {
t.Errorf("Certificates.Get returned error: %v", err)
}

expected := &Certificate{
ID: "892071a0-bb95-49bc-8021-3afd67a210bf",
Name: "web-cert-01",
NotAfter: "2017-02-22T00:23:00Z",
SHA1Fingerprint: "dfcc9f57d86bf58e321c2c6c31c7a971be244ac7",
Created: "2017-02-08T16:02:37Z",
}

assert.Equal(t, expected, certificate)
}

func TestCertificates_List(t *testing.T) {
setup()
defer teardown()

path := "/v2/certificates"
mux.HandleFunc(path, func(w http.ResponseWriter, r *http.Request) {
testMethod(t, r, "GET")
fmt.Fprint(w, cListJSONResponse)
})

certificates, _, err := client.Certificates.List(nil)

if err != nil {
t.Errorf("Certificates.List returned error: %v", err)
}

expected := []Certificate{
{
ID: "892071a0-bb95-49bc-8021-3afd67a210bf",
Name: "web-cert-01",
NotAfter: "2017-02-22T00:23:00Z",
SHA1Fingerprint: "dfcc9f57d86bf58e321c2c6c31c7a971be244ac7",
Created: "2017-02-08T16:02:37Z",
},
{
ID: "992071a0-bb95-49bc-8021-3afd67a210bf",
Name: "web-cert-02",
NotAfter: "2017-02-22T00:23:00Z",
SHA1Fingerprint: "cfcc9f57d86bf58e321c2c6c31c7a971be244ac7",
Created: "2017-02-08T16:02:37Z",
},
}

assert.Equal(t, expected, certificates)
}

func TestCertificates_Create(t *testing.T) {
setup()
defer teardown()

createRequest := &CertificateRequest{
Name: "web-cert-01",
PrivateKey: "-----BEGIN PRIVATE KEY-----",
LeafCertificate: "-----BEGIN CERTIFICATE-----",
CertificateChain: "-----BEGIN CERTIFICATE-----",
}

path := "/v2/certificates"
mux.HandleFunc(path, func(w http.ResponseWriter, r *http.Request) {
v := new(CertificateRequest)
err := json.NewDecoder(r.Body).Decode(v)
if err != nil {
t.Fatal(err)
}

testMethod(t, r, "POST")
assert.Equal(t, createRequest, v)

fmt.Fprint(w, certificateJSONResponse)
})

certificate, _, err := client.Certificates.Create(createRequest)
if err != nil {
t.Errorf("Certificates.Create returned error: %v", err)
}

expected := &Certificate{
ID: "892071a0-bb95-49bc-8021-3afd67a210bf",
Name: "web-cert-01",
NotAfter: "2017-02-22T00:23:00Z",
SHA1Fingerprint: "dfcc9f57d86bf58e321c2c6c31c7a971be244ac7",
Created: "2017-02-08T16:02:37Z",
}

assert.Equal(t, expected, certificate)
}

func TestCertificates_Delete(t *testing.T) {
setup()
defer teardown()

cID := "892071a0-bb95-49bc-8021-3afd67a210bf"
path := "/v2/certificates"
path = fmt.Sprintf("%s/%s", path, cID)
mux.HandleFunc(path, func(w http.ResponseWriter, r *http.Request) {
testMethod(t, r, "DELETE")
})

_, err := client.Certificates.Delete(cID)

if err != nil {
t.Errorf("Certificates.Delete returned error: %v", err)
}
}
2 changes: 2 additions & 0 deletions godo.go
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,7 @@ type Client struct {
StorageActions StorageActionsService
Tags TagsService
LoadBalancers LoadBalancersService
Certificates CertificatesService

// Optional function called after every successful request made to the DO APIs
onRequestCompleted RequestCompletionCallback
Expand Down Expand Up @@ -169,6 +170,7 @@ func NewClient(httpClient *http.Client) *Client {
c.StorageActions = &StorageActionsServiceOp{client: c}
c.Tags = &TagsServiceOp{client: c}
c.LoadBalancers = &LoadBalancersServiceOp{client: c}
c.Certificates = &CertificatesServiceOp{client: c}

return c
}
Expand Down

0 comments on commit 68caed3

Please sign in to comment.