Browse Source

Draw processed string in share PNG when ISP info is not available

Maddie Zhan 5 years ago
parent
commit
f7ccda23e1
1 changed files with 9 additions and 5 deletions
  1. 9 5
      results/telemetry.go

+ 9 - 5
results/telemetry.go

@@ -321,12 +321,16 @@ func DrawPNG(w http.ResponseWriter, r *http.Request) {
 	drawer.Face = orgFace
 	drawer.Src = colorISP
 	drawer.Dot = freetype.Pt(6, canvasHeight-ctx.PointToFixed(14).Round()-15)
-	removeRegexp := regexp.MustCompile(`AS\d+\s`)
-	org := removeRegexp.ReplaceAllString(result.RawISPInfo.Organization, "")
-	if result.RawISPInfo.Country != "" {
-		org += ", " + result.RawISPInfo.Country
+	if result.RawISPInfo.Organization != "" {
+		removeRegexp := regexp.MustCompile(`AS\d+\s`)
+		org := removeRegexp.ReplaceAllString(result.RawISPInfo.Organization, "")
+		if result.RawISPInfo.Country != "" {
+			org += ", " + result.RawISPInfo.Country
+		}
+		drawer.DrawString(org)
+	} else {
+		drawer.DrawString(result.ProcessedString)
 	}
-	drawer.DrawString(org)
 
 	w.Header().Set("Content-Disposition", "inline; filename="+uuid+".png")
 	w.Header().Set("Content-Type", "image/png")