Add stats command

Author: JojiiOfficial

go.mod

@@ -3,7 +3,7 @@ module github.com/DataManager-Go/DataManagerServer
 go 1.14
 
 require (
-	github.com/DataManager-Go/libdatamanager v1.3.6
+	github.com/DataManager-Go/libdatamanager v1.3.7
 	github.com/JojiiOfficial/configService v0.0.0-20200219132202-6e71512e2e28
 	github.com/JojiiOfficial/gaw v1.2.8
 	github.com/JojiiOfficial/shred v1.2.1

go.sum

@@ -1,7 +1,7 @@
 github.com/BurntSushi/toml v0.3.1 h1:WXkYYl6Yr3qBf1K79EBnL4mak0OimBfB0XUf9Vl28OQ=
 github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
-github.com/DataManager-Go/libdatamanager v1.3.6 h1:+hOGZecSTpdskGgo045hfFWqe97hSkHX4EVzN1EyuVI=
-github.com/DataManager-Go/libdatamanager v1.3.6/go.mod h1:DjxzXo2LkUjUWk6ihkRpdnGkm3jmKC31oNO0KqmgrTc=
+github.com/DataManager-Go/libdatamanager v1.3.7 h1:EzS4jlomalvEE8nNZFLFEdUuwZCm1GmlhhY7l0FupQ4=
+github.com/DataManager-Go/libdatamanager v1.3.7/go.mod h1:DjxzXo2LkUjUWk6ihkRpdnGkm3jmKC31oNO0KqmgrTc=
 github.com/JojiiOfficial/configService v0.0.0-20200219132202-6e71512e2e28 h1:nYoIExG+Z/gSLS9Jbpu6lnrh+m6e9gTxQfGhanTsExE=
 github.com/JojiiOfficial/configService v0.0.0-20200219132202-6e71512e2e28/go.mod h1:j1kHFoYWAbLRPE5nyAAtODwUc0xwd2+ifPZ3uCAgv/g=
 github.com/JojiiOfficial/gaw v1.2.8 h1:crLd2hrRvTlCClZDtwqnr8AoVKs3uQAk8B2AdOfUCsg=

handlers/FileHandler.go

@@ -641,8 +641,3 @@ func FileHandler(handlerData web.HandlerData, w http.ResponseWriter, r *http.Req
 
 	return nil
 }
-
-// Stats for user
-func Stats(handlerData web.HandlerData, w http.ResponseWriter, r *http.Request) error {
-	return nil
-}

handlers/Router.go

@@ -80,7 +80,7 @@ var (
 		Route{
 			Name:        "stats",
 			Pattern:     "/user/stats",
-			Method:      GetMethod,
+			Method:      POSTMethod,
 			HandlerFunc: Stats,
 			HandlerType: sessionRequest,
 		},

handlers/UserHandler.go

@@ -77,3 +77,56 @@ func Register(handlerData web.HandlerData, w http.ResponseWriter, r *http.Reques
 
 	return nil
 }
+
+// Stats for a user
+func Stats(handlerData web.HandlerData, w http.ResponseWriter, r *http.Request) error {
+	var request libdm.StatsRequestStruct
+	if !readRequestLimited(w, r, &request, handlerData.Config.Webserver.MaxRequestBodyLength) {
+		return nil
+	}
+
+	// With deleted
+	totalFileCount, err := handlerData.User.GetTotalFileCount(handlerData.Db)
+	if err != nil {
+		return err
+	}
+
+	// Without deleted
+	fileCount, err := handlerData.User.GetFileCount(handlerData.Db)
+	if err != nil {
+		return err
+	}
+
+	totalFileSize, err := handlerData.User.GetTotalFilesize(handlerData.Db)
+	if err != nil {
+		return err
+	}
+
+	namespaceCount, err := handlerData.User.GetNamespaceCount(handlerData.Db)
+	if err != nil {
+		return err
+	}
+
+	tagCount, err := handlerData.User.GetTagCount(handlerData.Db)
+	if err != nil {
+		return err
+	}
+
+	groupCount, err := handlerData.User.GetGroupCount(handlerData.Db)
+	if err != nil {
+		return err
+	}
+
+	respones := libdm.StatsResponse{
+		FilesUploaded:  totalFileCount,
+		FileCount:      fileCount,
+		DeletedFiles:   (totalFileCount - fileCount),
+		NamespaceCount: namespaceCount,
+		GroupCount:     groupCount,
+		TagCount:       tagCount,
+		TotalFileSize:  totalFileSize,
+	}
+
+	sendResponse(w, libdm.ResponseSuccess, "", respones)
+	return nil
+}

models/User.go

@@ -158,3 +158,52 @@ func (user *User) GetAllGroups(db *gorm.DB) ([]Group, error) {
 
 	return groups, err
 }
+
+// GetNamespaceCount return count of users namespaces
+func (user *User) GetNamespaceCount(db *gorm.DB) (int64, error) {
+	var c int64
+	return c, db.Model(Namespace{}).Where(&Namespace{UserID: user.ID}).Count(&c).Error
+}
+
+// GetTotalFileCount return total count of users files (with deletet files)
+func (user *User) GetTotalFileCount(db *gorm.DB) (int64, error) {
+	var c int64
+	return c, db.Unscoped().Model(File{}).Where(&File{UserID: user.ID}).Count(&c).Error
+}
+
+// GetFileCount return count of users files (without deleted files)
+func (user *User) GetFileCount(db *gorm.DB) (int64, error) {
+	var c int64
+	return c, db.Model(File{}).Where(&File{UserID: user.ID}).Count(&c).Error
+}
+
+// GetTotalFilesize return size of all users files
+func (user *User) GetTotalFilesize(db *gorm.DB) (int64, error) {
+	var c int64
+
+	rows, err := db.Table("files").Select("sum(file_size)").Where(&File{UserID: user.ID}).Rows()
+	if err != nil {
+		return 0, err
+	}
+
+	for rows.Next() {
+		err := rows.Scan(&c)
+		if err != nil {
+			return 0, err
+		}
+	}
+
+	return c, nil
+}
+
+// GetTagCount for user
+func (user *User) GetTagCount(db *gorm.DB) (int64, error) {
+	var c int64
+	return c, db.Model(Tag{}).Where(&Tag{UserID: user.ID}).Distinct("Name").Count(&c).Error
+}
+
+// GetGroupCount for user
+func (user *User) GetGroupCount(db *gorm.DB) (int64, error) {
+	var c int64
+	return c, db.Model(Group{}).Where(&Group{UserID: user.ID}).Distinct("Name").Count(&c).Error
+}