[add] tests, [add] refactor epub feat

api/api.go

@@ -13,8 +13,6 @@ import (
 	log "github.com/sirupsen/logrus"
 	"reichard.io/bbank/config"
 	"reichard.io/bbank/database"
-	"reichard.io/bbank/graph"
-	"reichard.io/bbank/utils"
 )
 
 type API struct {
@@ -75,9 +73,9 @@ func (api *API) registerWebAppRoutes() {
 	// Define Templates & Helper Functions
 	render := multitemplate.NewRenderer()
 	helperFuncs := template.FuncMap{
-		"GetSVGGraphData": graph.GetSVGGraphData,
-		"GetUTCOffsets":   utils.GetUTCOffsets,
-		"NiceSeconds":     utils.NiceSeconds,
+		"GetSVGGraphData": getSVGGraphData,
+		"GetUTCOffsets":   getUTCOffsets,
+		"NiceSeconds":     niceSeconds,
 	}
 
 	render.AddFromFilesFuncs("login", helperFuncs, "templates/login.html")

api/app-routes.go

@@ -19,7 +19,6 @@ import (
 	"reichard.io/bbank/database"
 	"reichard.io/bbank/metadata"
 	"reichard.io/bbank/search"
-	"reichard.io/bbank/utils"
 )
 
 type queryParams struct {
@@ -587,7 +586,7 @@ func (api *API) saveNewDocument(c *gin.Context) {
 	}
 
 	// Calculate Partial MD5 ID
-	partialMD5, err := utils.CalculatePartialMD5(tempFilePath)
+	partialMD5, err := calculatePartialMD5(tempFilePath)
 	if err != nil {
 		log.Warn("[saveNewDocument] Partial MD5 Error: ", err)
 		c.AbortWithStatus(http.StatusBadRequest)

api/utils.go

@@ -0,0 +1,136 @@
+package api
+
+import (
+	"bytes"
+	"crypto/md5"
+	"fmt"
+	"io"
+	"math"
+	"os"
+
+	"reichard.io/bbank/database"
+	"reichard.io/bbank/graph"
+)
+
+type UTCOffset struct {
+	Name  string
+	Value string
+}
+
+var UTC_OFFSETS = []UTCOffset{
+	{Value: "-12 hours", Name: "UTC−12:00"},
+	{Value: "-11 hours", Name: "UTC−11:00"},
+	{Value: "-10 hours", Name: "UTC−10:00"},
+	{Value: "-9.5 hours", Name: "UTC−09:30"},
+	{Value: "-9 hours", Name: "UTC−09:00"},
+	{Value: "-8 hours", Name: "UTC−08:00"},
+	{Value: "-7 hours", Name: "UTC−07:00"},
+	{Value: "-6 hours", Name: "UTC−06:00"},
+	{Value: "-5 hours", Name: "UTC−05:00"},
+	{Value: "-4 hours", Name: "UTC−04:00"},
+	{Value: "-3.5 hours", Name: "UTC−03:30"},
+	{Value: "-3 hours", Name: "UTC−03:00"},
+	{Value: "-2 hours", Name: "UTC−02:00"},
+	{Value: "-1 hours", Name: "UTC−01:00"},
+	{Value: "0 hours", Name: "UTC±00:00"},
+	{Value: "+1 hours", Name: "UTC+01:00"},
+	{Value: "+2 hours", Name: "UTC+02:00"},
+	{Value: "+3 hours", Name: "UTC+03:00"},
+	{Value: "+3.5 hours", Name: "UTC+03:30"},
+	{Value: "+4 hours", Name: "UTC+04:00"},
+	{Value: "+4.5 hours", Name: "UTC+04:30"},
+	{Value: "+5 hours", Name: "UTC+05:00"},
+	{Value: "+5.5 hours", Name: "UTC+05:30"},
+	{Value: "+5.75 hours", Name: "UTC+05:45"},
+	{Value: "+6 hours", Name: "UTC+06:00"},
+	{Value: "+6.5 hours", Name: "UTC+06:30"},
+	{Value: "+7 hours", Name: "UTC+07:00"},
+	{Value: "+8 hours", Name: "UTC+08:00"},
+	{Value: "+8.75 hours", Name: "UTC+08:45"},
+	{Value: "+9 hours", Name: "UTC+09:00"},
+	{Value: "+9.5 hours", Name: "UTC+09:30"},
+	{Value: "+10 hours", Name: "UTC+10:00"},
+	{Value: "+10.5 hours", Name: "UTC+10:30"},
+	{Value: "+11 hours", Name: "UTC+11:00"},
+	{Value: "+12 hours", Name: "UTC+12:00"},
+	{Value: "+12.75 hours", Name: "UTC+12:45"},
+	{Value: "+13 hours", Name: "UTC+13:00"},
+	{Value: "+14 hours", Name: "UTC+14:00"},
+}
+
+func getUTCOffsets() []UTCOffset {
+	return UTC_OFFSETS
+}
+
+func niceSeconds(input int64) (result string) {
+	if input == 0 {
+		return "N/A"
+	}
+
+	days := math.Floor(float64(input) / 60 / 60 / 24)
+	seconds := input % (60 * 60 * 24)
+	hours := math.Floor(float64(seconds) / 60 / 60)
+	seconds = input % (60 * 60)
+	minutes := math.Floor(float64(seconds) / 60)
+	seconds = input % 60
+
+	if days > 0 {
+		result += fmt.Sprintf("%dd ", int(days))
+	}
+	if hours > 0 {
+		result += fmt.Sprintf("%dh ", int(hours))
+	}
+	if minutes > 0 {
+		result += fmt.Sprintf("%dm ", int(minutes))
+	}
+	if seconds > 0 {
+		result += fmt.Sprintf("%ds", int(seconds))
+	}
+
+	return
+}
+
+// Reimplemented KOReader Partial MD5 Calculation
+func calculatePartialMD5(filePath string) (string, error) {
+	file, err := os.Open(filePath)
+	if err != nil {
+		return "", err
+	}
+
+	defer file.Close()
+
+	var step int64 = 1024
+	var size int64 = 1024
+	var buf bytes.Buffer
+
+	for i := -1; i <= 10; i++ {
+		byteStep := make([]byte, size)
+
+		var newShift int64 = int64(i * 2)
+		var newOffset int64
+		if i == -1 {
+			newOffset = 0
+		} else {
+			newOffset = step << newShift
+		}
+
+		_, err := file.ReadAt(byteStep, newOffset)
+		if err == io.EOF {
+			break
+		}
+		buf.Write(byteStep)
+	}
+
+	allBytes := buf.Bytes()
+	return fmt.Sprintf("%x", md5.Sum(allBytes)), nil
+}
+
+// Convert Database Array -> Int64 Array
+func getSVGGraphData(inputData []database.GetDailyReadStatsRow, svgWidth int, svgHeight int) graph.SVGGraphData {
+	var intData []int64
+	for _, item := range inputData {
+		intData = append(intData, item.MinutesRead)
+	}
+
+	return graph.GetSVGGraphData(intData, svgWidth, svgHeight)
+}