added sms response format

briandoesdev · Jun 23, 2024 · 1b09731 · 1b09731
1 parent cc1e7a0
commit 1b09731
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/routes/lookup/lookup_route.go b/routes/lookup/lookup_route.go
@@ -29,7 +29,7 @@ func Route(e *echo.Echo) {
 
 func getNumberSummary(c echo.Context) error {
 	var sum string
-	f := c.Request().Header.Get("X-Response-Format")
+	f := c.Request().Header.Get("X-Custom-Format")
 	number := c.FormValue("number")
 
 	if number == "" {
@@ -42,6 +42,8 @@ func getNumberSummary(c echo.Context) error {
 		switch f {
 		case "api":
 			return c.JSON(500, &routes.ErrorPayload{Message: err.Error()})
+		case "sms":
+			return c.String(200, err.Error())
 		default:
 			return c.Render(200, "home.html", map[string]interface{}{"sum": err.Error()})
 		}
@@ -53,6 +55,8 @@ func getNumberSummary(c echo.Context) error {
 		switch f {
 		case "api":
 			return c.JSON(500, &routes.ErrorPayload{Message: err.Error()})
+		case "sms":
+			return c.String(200, sum)
 		default:
 			return c.Render(200, "home.html", map[string]interface{}{"sum": err.Error()})
 		}
@@ -61,6 +65,8 @@ func getNumberSummary(c echo.Context) error {
 	switch f {
 	case "api":
 		return c.JSON(200, &Data{PhoneNumber: number, Summary: sum})
+	case "sms":
+		return c.String(200, sum)
 	default:
 		return c.Render(200, "home.html", map[string]interface{}{"sum": sum})
 	}