From 963993d520db4b64d28bab6bc47b7df40e1cef28 Mon Sep 17 00:00:00 2001
From: Matthew R <matthew@staff.libraryofcode.org>
Date: Wed, 18 Dec 2019 23:47:00 -0500
Subject: [PATCH] dirsize and storage go files

---
 src/intervals/dirsize.go | 25 +++++++++++
 src/intervals/storage.go | 89 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 114 insertions(+)
 create mode 100644 src/intervals/dirsize.go
 create mode 100644 src/intervals/storage.go

diff --git a/src/intervals/dirsize.go b/src/intervals/dirsize.go
new file mode 100644
index 0000000..f54802c
--- /dev/null
+++ b/src/intervals/dirsize.go
@@ -0,0 +1,25 @@
+package main
+
+import (
+	"os"
+	"path/filepath"
+)
+
+// DirSize This function returns the total size of a directory in bytes.
+func DirSize(path string) float64 {
+	var dirSize int64 = 0
+
+	readSize := func(path string, file os.FileInfo, err error) error {
+		if !file.IsDir() {
+			dirSize += file.Size()
+		}
+
+		return nil
+	}
+
+	filepath.Walk(path, readSize)
+
+	size := float64(dirSize)
+
+	return size
+}
diff --git a/src/intervals/storage.go b/src/intervals/storage.go
new file mode 100644
index 0000000..5583a76
--- /dev/null
+++ b/src/intervals/storage.go
@@ -0,0 +1,89 @@
+package main
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"github.com/go-redis/redis/v7"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+	"io/ioutil"
+	"time"
+)
+
+// Collection the MongoDB Account collection
+var Collection *mongo.Collection
+
+// RedisClient the Redis client
+var RedisClient *redis.Client
+
+// Account represents an user's account.
+type Account struct {
+	Username     string    `json:"username"`
+	UserID       string    `json:"userID"`
+	EmailAddress string    `json:"emailAddress"`
+	CreatedBy    string    `json:"createdBy"`
+	CreatedAt    time.Time `json:"createdAt"`
+	Locked       bool      `json:"locked"`
+}
+
+// HandleError This function panics if an error exists.
+func HandleError(e error, serv int) {
+	if e != nil && serv == 1 {
+		panic(e)
+	} else if e != nil && serv == 0 {
+		fmt.Println(e)
+	}
+}
+
+func main() {
+	type Config struct {
+		MongoDB string `json:"mongoURL"`
+	}
+	config := &Config{}
+	file, err := ioutil.ReadFile("../config.json")
+	HandleError(err, 1)
+	err = json.Unmarshal(file, &config)
+
+	client, err := mongo.NewClient(options.Client().ApplyURI(config.MongoDB))
+	HandleError(err, 1)
+	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
+	err = client.Connect(ctx)
+	HandleError(err, 1)
+	err = client.Ping(context.TODO(), nil)
+	HandleError(err, 1)
+
+	RedisClient = redis.NewClient(&redis.Options{
+		Addr:     "localhost:6379",
+		Password: "",
+		DB:       0,
+	})
+
+	_, err = RedisClient.Ping().Result()
+	HandleError(err, 1)
+
+	for {
+    handler()
+    time.Sleep(1000000 * time.Second)
+  }
+}
+
+func handler() {
+	cur, err := Collection.Find(context.TODO(), bson.M{})
+	HandleError(err, 0)
+
+	for cur.Next(context.TODO()) {
+		go checkAccountSizeAndUpdate(cur.Current.Lookup("username").String(), cur.Current.Lookup("id").String())
+		time.Sleep(600000 * time.Second)
+	}
+}
+
+func checkAccountSizeAndUpdate(username string, id string) {
+  var bytes float64 = 0
+  sizeHome := DirSize("/home/" + username)
+  bytes += sizeHome
+  sizeMail := DirSize("/var/mail/" + username)
+  bytes += sizeMail
+	RedisClient.Set("storage"+"-"+id, bytes, 0)
+}