Add library size to main stats page (#427)

go.mod

@@ -7,6 +7,7 @@ require (
 	github.com/antchfx/xpath v1.1.2 // indirect
 	github.com/bmatcuk/doublestar v1.1.5
 	github.com/disintegration/imaging v1.6.0
+	github.com/dustin/go-humanize v1.0.0
 	github.com/go-chi/chi v4.0.2+incompatible
 	github.com/gobuffalo/packr/v2 v2.0.2
 	github.com/golang-migrate/migrate/v4 v4.3.1

go.sum

@@ -81,6 +81,7 @@ github.com/docker/docker v0.7.3-0.20190108045446-77df18c24acf/go.mod h1:eEKB0N0r
 github.com/docker/go-connections v0.4.0/go.mod h1:Gbd7IOopHjR8Iph03tsViu4nIes5XhDvyHbTtUxmeec=
 github.com/docker/go-units v0.3.3/go.mod h1:fgPhTUdO+D/Jk86RDLlptpiXQzgHJF7gydDDbaIK4Dk=
 github.com/dustin/go-humanize v0.0.0-20180713052910-9f541cc9db5d/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=
+github.com/dustin/go-humanize v1.0.0 h1:VSnTsYCnlFHaM2/igO1h6X3HA71jcobQuxemgkq4zYo=
 github.com/dustin/go-humanize v1.0.0/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=
 github.com/eapache/go-resiliency v1.1.0/go.mod h1:kFI+JgMyC7bLPUVY133qvEBtVayf5mFgVsvEsIPBvNs=
 github.com/eapache/go-xerial-snappy v0.0.0-20180814174437-776d5712da21/go.mod h1:+020luEh2TKB4/GOp8oxxtq0Daoen/Cii55CzbTV6DU=

graphql/documents/queries/misc.graphql

@@ -52,6 +52,7 @@ query ValidGalleriesForScene($scene_id: ID!) {
 query Stats {
   stats {
     scene_count,
+    scene_size_count,
     gallery_count,
     performer_count,
     studio_count,

graphql/schema/types/stats.graphql

@@ -1,8 +1,9 @@
 type StatsResultType {
   scene_count: Int!
+  scene_size_count: String!
   gallery_count: Int!
   performer_count: Int!
   studio_count: Int!
   movie_count: Int!
   tag_count: Int!
-}
+}

pkg/api/resolver.go

@@ -93,6 +93,7 @@ func (r *queryResolver) ValidGalleriesForScene(ctx context.Context, scene_id *st
 func (r *queryResolver) Stats(ctx context.Context) (*models.StatsResultType, error) {
 	scenesQB := models.NewSceneQueryBuilder()
 	scenesCount, _ := scenesQB.Count()
+	scenesSizeCount, _ := scenesQB.SizeCount()
 	galleryQB := models.NewGalleryQueryBuilder()
 	galleryCount, _ := galleryQB.Count()
 	performersQB := models.NewPerformerQueryBuilder()
@@ -105,6 +106,7 @@ func (r *queryResolver) Stats(ctx context.Context) (*models.StatsResultType, err
 	tagsCount, _ := tagsQB.Count()
 	return &models.StatsResultType{
 		SceneCount:     scenesCount,
+		SceneSizeCount: scenesSizeCount,
 		GalleryCount:   galleryCount,
 		PerformerCount: performersCount,
 		StudioCount:    studiosCount,

pkg/models/querybuilder_scene.go

@@ -5,6 +5,7 @@ import (
 	"strconv"
 	"strings"
 
+	"github.com/dustin/go-humanize"
 	"github.com/jmoiron/sqlx"
 	"github.com/stashapp/stash/pkg/database"
 )
@@ -195,6 +196,14 @@ func (qb *SceneQueryBuilder) Count() (int, error) {
 	return runCountQuery(buildCountQuery("SELECT scenes.id FROM scenes"), nil)
 }
 
+func (qb *SceneQueryBuilder) SizeCount() (string, error) {
+	sum, err := runSumQuery("SELECT SUM(size) as sum FROM scenes", nil)
+	if err != nil {
+		return "0", err
+	}
+	return humanize.Bytes(sum), err
+}
+
 func (qb *SceneQueryBuilder) CountByStudioID(studioID int) (int, error) {
 	args := []interface{}{studioID}
 	return runCountQuery(buildCountQuery(scenesForStudioQuery), args)

pkg/models/querybuilder_sql.go

@@ -258,6 +258,18 @@ func runCountQuery(query string, args []interface{}) (int, error) {
 	return result.Int, nil
 }
 
+func runSumQuery(query string, args []interface{}) (uint64, error) {
+	// Perform query and fetch result
+	result := struct {
+		Uint uint64 `db:"sum"`
+	}{0}
+	if err := database.DB.Get(&result, query, args...); err != nil && err != sql.ErrNoRows {
+		return 0, err
+	}
+
+	return result.Uint, nil
+}
+
 func executeFindQuery(tableName string, body string, args []interface{}, sortAndPagination string, whereClauses []string, havingClauses []string) ([]int, int) {
 	if len(whereClauses) > 0 {
 		body = body + " WHERE " + strings.Join(whereClauses, " AND ") // TODO handle AND or OR

ui/v2.5/src/components/Stats.tsx

@@ -13,6 +13,14 @@ export const Stats: React.FC = () => {
   return (
     <div className="mt-5">
       <div className="col col-sm-8 m-sm-auto row stats">
+        <div className="stats-element">
+          <p className="title">
+            <FormattedMessage id="size" defaultMessage={data.stats.scene_size_count} />
+          </p>
+          <p className="heading">
+            <FormattedMessage id="library" defaultMessage="Library size" />
+          </p>
+        </div>
         <div className="stats-element">
           <p className="title">
             <FormattedNumber value={data.stats.scene_count} />

ui/v2/src/components/Stats.tsx

@@ -9,6 +9,12 @@ export const Stats: FunctionComponent = () => {
     if (!data || !data.stats) { return; }
     return (
       <nav id="details-container" className="level stats">
+        <div className="level-item has-text-centered">
+          <div>
+            <p className="title">{data.stats.scene_size_count}</p>
+            <p className="heading">Library size</p>
+          </div>
+        </div>
         <div className="level-item has-text-centered">
           <div>
             <p className="title">{data.stats.scene_count}</p>