minimal tracking only that can be disabled

Author: prasunanand

app.go

@@ -68,6 +68,7 @@ func main() {
 	cwd := flag.String("cwd", ".", "base directory of project")
 	port := flag.String("port", ":8048", "port to start the server on")
 	protected := flag.Bool("protected", false, "enable protected mode")
+	tracking := flag.Bool("tracking", true, "enable usage tracking")
 
 	flag.Parse()
 
@@ -174,28 +175,26 @@ func main() {
 
 	router.PathPrefix("/").Handler(getSpaHandler())
 
-	analytics.TrackEvent("server_started", map[string]interface{}{"source": "web"})
+	// Track server start and stop events if tracking is enabled
+	// Note that this helps me understand if the users are actually using Zasper
+	// and keeps me motivated to maintain and improve the product
+
+	if *tracking {
+		analytics.TrackServerStartStopEvent("server_started", map[string]interface{}{"source": "web"})
+	}
 
 	// Channel for graceful shutdown
 	stop := make(chan os.Signal, 1)
 	signal.Notify(stop, syscall.SIGINT, syscall.SIGTERM)
 
-	printBanner(*port, core.ServerAccessToken, version, *protected)
+	printBanner(*port, core.ServerAccessToken, version, *protected, *tracking)
 
 	go func() {
 		if err := http.ListenAndServe(*port, corsOpts.Handler(router)); err != nil && err != http.ErrServerClosed {
 			fmt.Printf("ListenAndServe(): %s\n", err)
 		}
 	}()
 
-	go func() {
-		ticker := time.NewTicker(10 * time.Minute)
-		defer ticker.Stop()
-		for range ticker.C {
-			analytics.SendStatsToPostHog()
-		}
-	}()
-
 	<-stop
 	fmt.Println("Shutting down server...")
 
@@ -208,7 +207,7 @@ func main() {
 	fmt.Println("Server exiting")
 }
 
-func printBanner(port string, accessToken string, version string, protected bool) {
+func printBanner(port string, accessToken string, version string, protected bool, tracking bool) {
 	fmt.Println("==========================================================")
 	fmt.Println("     ███████╗ █████╗ ███████╗██████╗ ███████╗██████╗ ")
 	fmt.Println("     ╚══███╔╝██╔══██╗██╔════╝██╔══██╗██╔════╝██╔══██╗")
@@ -229,6 +228,7 @@ func printBanner(port string, accessToken string, version string, protected bool
 	} else {
 		fmt.Println(" 🔒 Protected Mode:      disabled")
 	}
+	fmt.Println(" 🔄 Server start/shutdown tracking enabled:", tracking)
 	fmt.Println("==========================================================")
 }
 

internal/analytics/tracking.go

@@ -2,7 +2,6 @@ package analytics
 
 import (
 	"strings"
-	"time"
 
 	"github.com/google/uuid"
 	"github.com/posthog/posthog-go"
@@ -54,51 +53,6 @@ func IncrementUsageStat(eventType EventType) {
 	}
 }
 
-func SendStatsToPostHog() {
-	if client == nil {
-		log.Info().Msg("PostHog client not initialized.")
-		return
-	}
-
-	props := posthog.NewProperties().
-		Set("notebooks_opened", stats.NotebooksOpened).
-		Set("terminals_opened", stats.TerminalsOpened).
-		Set("code_cells_executed", stats.CodeCellsExecuted).
-		Set("files_opened", stats.FilesOpened).
-		Set("timestamp", time.Now().Format(time.RFC3339)).
-		Set("OS", core.Zasper.OSName)
-
-	config, _ := core.ReadConfig()
-
-	err := client.Enqueue(posthog.Capture{
-		DistinctId: config.TrackingID,
-		Event:      "session usage summary",
-		Properties: props,
-	})
-
-	if err != nil {
-		log.Debug().Msgf("PostHog enqueue failed: %v. Retrying once...", err)
-
-		// Retry once after a short delay
-		time.Sleep(2 * time.Second)
-		err = client.Enqueue(posthog.Capture{
-			DistinctId: config.TrackingID,
-			Event:      "session usage summary",
-			Properties: props,
-		})
-
-		if err != nil {
-			log.Printf("PostHog enqueue retry also failed: %v", err)
-		} else {
-			log.Debug().Msg("PostHog retry succeeded.")
-			stats = UsageStats{}
-		}
-	} else {
-		log.Debug().Msg("PostHog session usage summary sent successfully.")
-		stats = UsageStats{}
-	}
-}
-
 // Function to generate or retrieve the tracking ID from the config
 func GetAnonymousTrackingId() (string, error) {
 	config, err := core.ReadConfig()
@@ -143,7 +97,7 @@ func SetUpPostHogClient() error {
 	return nil
 }
 
-func TrackEvent(eventName string, properties map[string]interface{}) {
+func TrackServerStartStopEvent(eventName string, properties map[string]interface{}) {
 	if phAPIKey == "" {
 		log.Warn().Msg("API key is missing. Event tracking skipped.")
 		return
@@ -167,8 +121,7 @@ func TrackEvent(eventName string, properties map[string]interface{}) {
 }
 
 func CloseClient() {
-	SendStatsToPostHog()
-	TrackEvent("server_shutdown", map[string]interface{}{"source": "web"})
+	TrackServerStartStopEvent("server_shutdown", map[string]interface{}{"source": "web"})
 	// Flush analytics queue
 	if client != nil {
 		client.Close() // This will block until the queue is flushed

internal/content/content_manager.go

@@ -370,12 +370,6 @@ func GetKernelPath(path string) int {
 	return 1
 }
 
-func dirExists(path string) bool {
-	path = filepath.Clean(path)
-	os_path := GetSafePath(path)
-	return IsDir(os_path)
-}
-
 func IsDir(path string) bool {
 	info, err := os.Lstat(path)