feat: add dynamic time limit conf from api

README.md

@@ -46,6 +46,9 @@ Create the file `/etc/gomonitor.json`. Yes, it's a json file. Yes GoMonitor does
     "screenTimeLimitMinutes": {
       "USER": 120,
     },
+    "screenTimeLimitApiMinutes": {
+      "OTHER_USER": "https://myapi/gomonitor",
+    },
     "logIntervalMinutes": 10,
     "retainPeriodDays": 10
   }
@@ -63,6 +66,14 @@ GoMonitor keeps data and logs for `retainPeriodDays` days
 
 Replace `USER` with the user you want to monitor.
 
+You can set a static limit for `screenTimeLimitMinutes` or use an API endpoint for `screenTimeLimitApiMinutes`, in such case you must return the following json response:
+
+``` json
+{
+  "limitMin": 120
+}
+```
+
 ### Usage
 
 GoMonitor behaves differently when running as root or as normal user. Fact.

core/monitor/root.go

@@ -1,6 +1,7 @@
 package monitor
 
 import (
+	"fmt"
 	"math"
 	"os"
 	"path/filepath"
@@ -21,6 +22,7 @@ type ProcessLog struct {
 }
 
 func RunAsRoot() {
+	fmt.Println("Retrieving current user processes")
 	// now
 	startTime := time.Now()
 	// every logIntervalMinutes minutes we log
@@ -41,6 +43,7 @@ func RunAsRoot() {
 		}
 
 		timeScreenLimit, err := utils.GetScreenTimeLimitMinutes(currentUser)
+		logger.ZapLog.Debug("Current screen time limit ", timeScreenLimit)
 
 		logger.ZapLog.Debug("Retrieving current user processes")
 		logProcesses := make([]ProcessLog, 0)

core/monitor/user.go

@@ -155,6 +155,8 @@ func notificationsThread() {
 	}
 
 	for {
+		// if taken from api, we need to refresh it
+		timeScreenLimit, err := utils.GetScreenTimeLimitMinutes(currentUser)
 		// get sum of partial_time_minutes for current day and user
 		totalMinutes, err := utils.GetTotalTodayTimeMinutes(currentUser)
 		if err == nil {

core/utils/settings.go

@@ -1,15 +1,47 @@
 package utils
 
-import "github.com/spf13/viper"
+import (
+	"encoding/json"
+	"io"
+	"net/http"
+
+	"github.com/spf13/viper"
+)
+
+type ApiLimitBody struct {
+	LimitMin int `json:"limitMin"`
+}
 
 func GetScreenTimeLimitMinutes(user string) (int, error) {
+	apiLimits := make(map[string]string)
+	err := viper.UnmarshalKey("app.screenTimeLimitApiMinutes", &apiLimits)
+	timeScreenLimitApi, okTimeScreenLimitApi := apiLimits[user]
+
+	if err == nil && okTimeScreenLimitApi {
+		res, err := http.Get(timeScreenLimitApi)
+		if err == nil {
+			defer res.Body.Close()
+			body, err := io.ReadAll(res.Body)
+			if err == nil {
+				var apiLimitBody ApiLimitBody
+				err = json.Unmarshal(body, &apiLimitBody)
+				if err == nil {
+					return apiLimitBody.LimitMin * 100, err
+				}
+			}
+		}
+	}
+
 	limits := make(map[string]int)
-	err := viper.UnmarshalKey("app.screenTimeLimitMinutes", &limits)
+	err = viper.UnmarshalKey("app.screenTimeLimitMinutes", &limits)
 	timeScreenLimit, okTimeScreenLimit := limits[user]
 
 	if err == nil && okTimeScreenLimit {
 		return timeScreenLimit, nil
 	} else {
+		// try api limimts
+		apiLimits := make(map[string]string)
+		err := viper.UnmarshalKey("app.screenTimeLimitMinutes", &apiLimits)
 		return 0, err
 	}
 }