Merge pull request #34 from alcounit/develop

Fix issue #33
alcounit · Apr 13, 2021 · a283926 · a283926
2 parents 2aac7a8 + b19725b
commit a283926
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/handlers.go b/handlers.go
@@ -375,11 +375,17 @@ func (app *App) HandleStatus(w http.ResponseWriter, r *http.Request) {
 }
 
 func parseImage(image string) (container string) {
-	pref, err := regexp.Compile("[^a-zA-Z0-9]+")
-	if err != nil {
-		return "selenoid-browser"
+	browser := "browser"
+	if len(image) > 0 {
+		pref, err := regexp.Compile("[^a-zA-Z0-9]+")
+		if err != nil {
+			return browser
+		}
+		fragments := strings.Split(image, "/")
+		image = fragments[len(fragments)-1]
+		return pref.ReplaceAllString(image, "-")
 	}
-	return pref.ReplaceAllString(image, "-")
+	return browser
 }
 
 func getSessionStats(sessions []platform.Service) (active []platform.Service, pending []platform.Service) {

diff --git a/selenosis.go b/selenosis.go
@@ -63,7 +63,7 @@ func New(logger *log.Logger, client platform.Platform, browsers *config.Browsers
 	if quota, err = client.Quota().Get(); err != nil {
 		quota, err = client.Quota().Create(currentTotal())
 		if err != nil {
-			logger.Errorf("failed to create quota resource: %v", err)
+			logger.Fatalf("failed to create quota resource: %v", err)
 		}
 	}