From 8725cd2012b36601f79d5e5eb57ed98cde3d11f4 Mon Sep 17 00:00:00 2001
From: Matt Baer <matt@baer.works>
Date: Wed, 5 Oct 2016 18:12:54 -0400
Subject: [PATCH] Support creating a collection

---
 collection.go | 66 +++++++++++++++++++++++++++++++++++++++++----------
 1 file changed, 53 insertions(+), 13 deletions(-)

diff --git a/collection.go b/collection.go
index 2f28dbc..e9102b0 100644
--- a/collection.go
+++ b/collection.go
@@ -5,20 +5,28 @@ import (
 	"net/http"
 )
 
-// Collection represents a collection of posts. Blogs are a type of collection
-// on Write.as.
-type Collection struct {
-	Alias       string `json:"alias"`
-	Title       string `json:"title"`
-	Description string `json:"description"`
-	StyleSheet  string `json:"style_sheet"`
-	Private     bool   `json:"private"`
-	Views       int64  `json:"views"`
-	Domain      string `json:"domain,omitempty"`
-	Email       string `json:"email,omitempty"`
+type (
+	// Collection represents a collection of posts. Blogs are a type of collection
+	// on Write.as.
+	Collection struct {
+		Alias       string `json:"alias"`
+		Title       string `json:"title"`
+		Description string `json:"description"`
+		StyleSheet  string `json:"style_sheet"`
+		Private     bool   `json:"private"`
+		Views       int64  `json:"views"`
+		Domain      string `json:"domain,omitempty"`
+		Email       string `json:"email,omitempty"`
 
-	TotalPosts int `json:"total_posts"`
-}
+		TotalPosts int `json:"total_posts"`
+	}
+
+	// CollectionParams holds values for creating a collection.
+	CollectionParams struct {
+		Alias string `json:"alias"`
+		Title string `json:"title"`
+	}
+)
 
 // GetCollection retrieves a collection, returning the Collection and any error
 // (in user-friendly form) that occurs. See
@@ -45,3 +53,35 @@ func (c *Client) GetCollection(alias string) (*Collection, error) {
 	}
 	return coll, nil
 }
+
+// CreateCollection creates a new collection, returning a user-friendly error
+// if one comes up. Requires a Write.as subscription. See
+// https://writeas.github.io/docs/#create-a-collection
+func (c *Client) CreateCollection(sp *CollectionParams) (*Collection, error) {
+	p := &Collection{}
+	env, err := c.post("/collections", sp, p)
+	if err != nil {
+		return nil, err
+	}
+
+	var ok bool
+	if p, ok = env.Data.(*Collection); !ok {
+		return nil, fmt.Errorf("Wrong data returned from API.")
+	}
+
+	status := env.Code
+	if status == http.StatusCreated {
+		return p, nil
+	} else if status == http.StatusBadRequest {
+		return nil, fmt.Errorf("Bad request: %s", env.ErrorMessage)
+	} else if status == http.StatusForbidden {
+		return nil, fmt.Errorf("Casual or Pro user required.")
+	} else if status == http.StatusConflict {
+		return nil, fmt.Errorf("Collection name is already taken.")
+	} else if status == http.StatusPreconditionFailed {
+		return nil, fmt.Errorf("Reached max collection quota.")
+	} else {
+		return nil, fmt.Errorf("Problem getting post: %s. %v\n", status, err)
+	}
+	return p, nil
+}