azure storage

README.md

@@ -1,2 +1,8 @@
 # webapp-go
 The sample app from Writing Web Applications on golang.org.
+
+# Get Azure SDK package: 
+
+```sh
+go get -d github.com/Azure/azure-sdk-for-go/management
+```

main.go

@@ -5,29 +5,85 @@
 package main
 
 import (
+	"bytes"
+	"errors"
+	"github.com/Azure/azure-sdk-for-go/storage"
 	"html/template"
 	"io/ioutil"
 	"net/http"
+	"os"
 	"regexp"
 )
 
+var blobService storage.BlobStorageClient
+var containerName = os.Getenv("AZURE_STORAGE_CONTAINER_NAME")
+
 type Page struct {
 	Title string
 	Body  []byte
 }
 
+func initAzureStorage() {
+	var storageKey = os.Getenv("AZURE_STORAGE_ACCESS_KEY")
+	var storageName = os.Getenv("AZURE_STORAGE_ACCOUNT")
+	client, err := storage.NewBasicClient(storageName, storageKey)
+	if err != nil {
+		panic(err)
+	}
+
+	blobService = client.GetBlobService()
+	ok, err := blobService.ContainerExists(containerName)
+	if err != nil {
+		panic(err)
+	} else if !ok {
+		err = blobService.CreateContainer(containerName, storage.ContainerAccessTypePrivate)
+		if err != nil {
+			panic(err)
+		}
+	}
+}
+
 func (p *Page) save() error {
-	filename := p.Title + ".txt"
-	return ioutil.WriteFile(filename, p.Body, 0600)
+	ok, err := blobService.BlobExists(containerName, p.Title)
+	if err != nil {
+		return err
+	}
+
+	if ok {
+		err = blobService.DeleteBlob(containerName, p.Title)
+		if err != nil {
+			return err
+		}
+	}
+
+	err = blobService.CreateBlockBlobFromReader(containerName, p.Title, uint64(len(p.Body)), bytes.NewReader(p.Body))
+	if err != nil {
+		return err
+	}
+
+	return nil
 }
 
 func loadPage(title string) (*Page, error) {
-	filename := title + ".txt"
-	body, err := ioutil.ReadFile(filename)
-	if err != nil {
+	ok, err := blobService.BlobExists(containerName, title)
+	if ok {
+		var body []byte
+		data, err := blobService.GetBlob(containerName, title)
+		if err == nil {
+			body, err = ioutil.ReadAll(data)
+		}
+
+		if data != nil {
+			data.Close()
+		}
+
+		return &Page{Title: title, Body: body}, err
+	} else {
+		if err == nil {
+			err = errors.New("Page not found.")
+		}
 		return nil, err
 	}
-	return &Page{Title: title, Body: body}, nil
 }
 
 func viewHandler(w http.ResponseWriter, r *http.Request, title string) {
@@ -81,6 +137,8 @@ func makeHandler(fn func(http.ResponseWriter, *http.Request, string)) http.Handl
 }
 
 func main() {
+	initAzureStorage()
+
 	http.HandleFunc("/view/", makeHandler(viewHandler))
 	http.HandleFunc("/edit/", makeHandler(editHandler))
 	http.HandleFunc("/save/", makeHandler(saveHandler))