Small cleanings

TrueBlocks · Jan 25, 2025 · 7ccb692 · 7ccb692
1 parent 05f9894
commit 7ccb692
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/services/service_control.go b/services/service_control.go
@@ -44,6 +44,7 @@ func (s *ControlService) Initialize() error {
 	mux.HandleFunc("/isPaused", s.handleIsPaused)
 	mux.HandleFunc("/pause", s.handlePause)
 	mux.HandleFunc("/unpause", s.handleUnpause)
+	mux.HandleFunc("/", s.handleDefault)
 
 	s.server = &http.Server{
 		Addr:    s.listenAddr,
@@ -75,6 +76,16 @@ func (s *ControlService) Logger() *slog.Logger {
 	return s.logger
 }
 
+func (s *ControlService) handleDefault(w http.ResponseWriter, r *http.Request) {
+	results := map[string]string{
+		"/status":   "[name]",
+		"/isPaused": "name",
+		"/pause":    "name",
+		"/unpause":  "name",
+	}
+	writeJSONResponse(w, results)
+}
+
 func (s *ControlService) handleIsPaused(w http.ResponseWriter, r *http.Request) {
 	name := r.URL.Query().Get("name")
 	results, err := s.manager.IsPaused(name)

diff --git a/services/service_scraper.go b/services/service_scraper.go
@@ -140,10 +140,10 @@ func (s *ScrapeService) Logger() *slog.Logger {
 func (s *ScrapeService) initOneChain(chain string) (*scraperReport, error) {
 	defer func() {
 		logger.SetLoggerWriter(io.Discard)
-		os.Setenv("TB_NODE_HEADLESS", "")
+		os.Setenv("TB_SCRAPE_HEADLESS", "")
 	}()
 	logger.SetLoggerWriter(os.Stderr)
-	os.Setenv("TB_NODE_HEADLESS", "true")
+	os.Setenv("TB_SCRAPE_HEADLESS", "true")
 
 	opts := sdk.InitOptions{
 		Globals: sdk.Globals{
@@ -175,10 +175,10 @@ func (s *ScrapeService) initOneChain(chain string) (*scraperReport, error) {
 func (s *ScrapeService) scrapeOneChain(chain string) (*scraperReport, error) {
 	defer func() {
 		logger.SetLoggerWriter(io.Discard)
-		os.Setenv("TB_NODE_HEADLESS", "")
+		os.Setenv("TB_SCRAPE_HEADLESS", "")
 	}()
 	logger.SetLoggerWriter(os.Stderr)
-	os.Setenv("TB_NODE_HEADLESS", "true")
+	os.Setenv("TB_SCRAPE_HEADLESS", "true")
 
 	if s.IsPaused() {
 		s.logger.Debug("Scraper is paused, skipping scraping step")