package routers import ( "encoding/json" "fmt" "github.com/pkg/errors" "net/http" "os" "path" "recipe-manager/contracts" "recipe-manager/data" "recipe-manager/helpers" "recipe-manager/models" "recipe-manager/services/logger" "recipe-manager/services/recipe" "recipe-manager/services/sheet" "strconv" "strings" "github.com/go-chi/chi/v5" "go.uber.org/zap" ) type RecipeRouter struct { data *data.Data sheetService sheet.SheetService recipeService recipe.RecipeService taoLogger *logger.TaoLogger } func NewRecipeRouter(data *data.Data, recipeService recipe.RecipeService, sheetService sheet.SheetService, taoLogger *logger.TaoLogger) *RecipeRouter { return &RecipeRouter{ data, sheetService, recipeService, taoLogger, } } func (rr *RecipeRouter) Route(r chi.Router) { r.Route("/recipes", func(r chi.Router) { r.Get("/dashboard", rr.dashBoard) r.Get("/overview", rr.overview) r.Get("/{product_code}", rr.getRecipeByProductCode) r.Get("/{product_code}/mat", rr.getRecipeMatByProductCode) r.Get("/{country}/{filename}/json", rr.getRecipeJson) r.Post("/edit/{country}/{filename}", rr.updateRecipe) r.Get("/countries", func(w http.ResponseWriter, r *http.Request) { w.Header().Add("Content-Type", "application/json") // get key from map var keys []string for k := range rr.data.AllRecipeFiles { countryName, err := rr.data.GetCountryNameByID(k) if err != nil { continue } keys = append(keys, countryName) } if err := json.NewEncoder(w).Encode(keys); err != nil { rr.taoLogger.Log.Error("RecipeRouter.GetCountryRecipe", zap.Error(err)) http.Error(w, "Internal Error", http.StatusInternalServerError) return } }) r.Get("/{country}/versions", func(w http.ResponseWriter, r *http.Request) { w.Header().Add("Content-Type", "application/json") countryName := chi.URLParam(r, "country") countryID, err := rr.data.GetCountryIDByName(countryName) if err != nil { http.Error(w, fmt.Sprintf("Country Name: %s not found!!!", countryName), http.StatusNotFound) return } var files []string for _, v := range rr.data.AllRecipeFiles[countryID] { files = append(files, v.Name) } if err := json.NewEncoder(w).Encode(files); err != nil { rr.taoLogger.Log.Error("RecipeRouter.GetVersionsCountryRecipe", zap.Error(err)) http.Error(w, "Internal Error", http.StatusInternalServerError) return } }) r.Get("/test/sheet", func(w http.ResponseWriter, r *http.Request) { result := rr.sheetService.GetSheet(r.Context(), "1rSUKcc5POR1KeZFGoeAZIoVoI7LPGztBhPw5Z_ConDE") var mapResult []map[string]string for _, v := range result { mapResult = append(mapResult, map[string]string{ "productCode": v[0].(string), "name": v[1].(string), "otherName": v[2].(string), "description": v[3].(string), "otherDescription": v[4].(string), "picture": v[5].(string), }) } if err := json.NewEncoder(w).Encode(mapResult); err != nil { rr.taoLogger.Log.Error("RecipeRouter.TestSheet", zap.Error(err)) http.Error(w, "Internal Error", http.StatusInternalServerError) return } }) }) } // ====================== Handler ================================= func (rr *RecipeRouter) dashBoard(w http.ResponseWriter, r *http.Request) { w.Header().Add("Content-Type", "application/json") country := r.URL.Query().Get("country") filename := r.URL.Query().Get("filename") result, err := rr.recipeService.GetRecipeDashboard(&contracts.RecipeDashboardRequest{ Country: country, Filename: filename, }) if err != nil { http.Error(w, err.Error(), http.StatusNotFound) return } if err := json.NewEncoder(w).Encode(result); err != nil { rr.taoLogger.Log.Error("RecipeRouter.Dashboard", zap.Error(err)) http.Error(w, "Internal Error", http.StatusInternalServerError) return } } func (rr *RecipeRouter) overview(w http.ResponseWriter, r *http.Request) { w.Header().Add("Content-Type", "application/json") var take, offset uint64 = 10, 0 if newOffset, err := strconv.ParseUint(r.URL.Query().Get("offset"), 10, 64); err == nil { offset = newOffset } if newTake, err := strconv.ParseUint(r.URL.Query().Get("take"), 10, 64); err == nil { take = newTake } country := r.URL.Query().Get("country") filename := r.URL.Query().Get("filename") materialIds := r.URL.Query().Get("materialIds") var materialIdsUint []int for _, v := range strings.Split(materialIds, ",") { materialIdUint, err := strconv.ParseUint(v, 10, 64) if err != nil || materialIdUint == 0 { continue } materialIdsUint = append(materialIdsUint, int(materialIdUint)) } result, err := rr.recipeService.GetRecipeOverview(&contracts.RecipeOverviewRequest{ Take: int(take), Skip: int(offset), Search: r.URL.Query().Get("search"), Country: country, Filename: filename, MatIds: materialIdsUint, }) if err != nil { http.Error(w, err.Error(), http.StatusNotFound) return } if err := json.NewEncoder(w).Encode(result); err != nil { rr.taoLogger.Log.Error("RecipeRouter.Overview", zap.Error(err)) http.Error(w, "Internal Error", http.StatusInternalServerError) return } } func (rr *RecipeRouter) getRecipeByProductCode(w http.ResponseWriter, r *http.Request) { w.Header().Add("Content-Type", "application/json") productCode := chi.URLParam(r, "product_code") fileName := r.URL.Query().Get("filename") country := r.URL.Query().Get("country") // recipe := rr.data.GetRecipe01() // recipeMetaData := rr.sheetService.GetSheet(r.Context(), "1rSUKcc5POR1KeZFGoeAZIoVoI7LPGztBhPw5Z_ConDE") // var recipeResult *models.Recipe01 // recipeMetaDataResult := map[string]string{} // for _, v := range recipe { // if v.ProductCode == productCode { // recipeResult = &v // break // } // } // for _, v := range recipeMetaData { // if v[0].(string) == productCode { // recipeMetaDataResult = map[string]string{ // "productCode": v[0].(string), // "name": v[1].(string), // "otherName": v[2].(string), // "description": v[3].(string), // "otherDescription": v[4].(string), // "picture": v[5].(string), // } // break // } // } // if recipeResult == nil { // http.Error(w, "Not Found", http.StatusNotFound) // return // } // json.NewEncoder(w).Encode(map[string]interface{}{ // "recipe": recipeResult, // "recipeMetaData": recipeMetaDataResult, // }) result, err := rr.recipeService.GetRecipeDetail(&contracts.RecipeDetailRequest{ Filename: fileName, Country: country, ProductCode: productCode, }) if err != nil { rr.taoLogger.Log.Error("RecipeRouter.GetRecipeByProductCode", zap.Error(err)) http.Error(w, fmt.Sprintf("Recipe file: %s with productCode: %s not found", fileName, productCode), http.StatusNotFound) return } if err := json.NewEncoder(w).Encode(result); err != nil { rr.taoLogger.Log.Error("RecipeRouter.GetRecipeByProductCode", zap.Error(err)) http.Error(w, "Internal Error", http.StatusInternalServerError) return } } func (rr *RecipeRouter) getRecipeMatByProductCode(w http.ResponseWriter, r *http.Request) { w.Header().Add("Content-Type", "application/json") productCode := chi.URLParam(r, "product_code") fileName := r.URL.Query().Get("filename") country := r.URL.Query().Get("country") result, err := rr.recipeService.GetRecipeDetailMat(&contracts.RecipeDetailRequest{ Filename: fileName, Country: country, ProductCode: productCode, }) if err != nil { rr.taoLogger.Log.Error("RecipeRouter.GetRecipeMatByProductCode", zap.Error(err)) http.Error(w, fmt.Sprintf("Material Recipe file: %s with productCode: %s not found", fileName, productCode), http.StatusNotFound) return } if err := json.NewEncoder(w).Encode(result); err != nil { rr.taoLogger.Log.Error("RecipeRouter.GetRecipeMatByProductCode", zap.Error(err)) http.Error(w, "Internal Error", http.StatusInternalServerError) return } } func (rr *RecipeRouter) getRecipeJson(w http.ResponseWriter, r *http.Request) { country := chi.URLParam(r, "country") filename := chi.URLParam(r, "filename") w.Header().Add("Content-Type", "application/json") countryID, err := rr.data.GetCountryIDByName(country) if err != nil { rr.taoLogger.Log.Error("RecipeRouter.GetRecipeJson", zap.Error(err)) http.Error(w, fmt.Sprintf("Country Name: %s not found!!!", country), http.StatusNotFound) return } if err := json.NewEncoder(w).Encode(rr.data.GetRecipe(countryID, filename)); err != nil { rr.taoLogger.Log.Error("RecipeRouter.GetRecipeJson", zap.Error(err)) http.Error(w, "Internal Error", http.StatusInternalServerError) return } } func (rr *RecipeRouter) updateRecipe(w http.ResponseWriter, r *http.Request) { rr.taoLogger.Log.Debug("Edit: ", zap.String("path", r.RequestURI)) filename := chi.URLParam(r, "filename") country := chi.URLParam(r, "country") countryID, err := rr.data.GetCountryIDByName(country) if err != nil { rr.taoLogger.Log.Error("RecipeRouter.UpdateRecipe", zap.Error(err)) http.Error(w, fmt.Sprintf("Country Name: %s not found!!!", country), http.StatusNotFound) return } targetRecipe := rr.data.GetRecipe(countryID, filename) rr.taoLogger.Log.Debug("Target => ", zap.Any("target", targetRecipe.MachineSetting.ConfigNumber)) // check request structure // FIXME: Request structure bug. Case-sensitive, likely bug at client // uncomment the below code to view the bug // var change_request map[string]interface{} // err = json.NewDecoder(r.Body).Decode(&change_request) // if err != nil { // rr.taoLogger.Log.Error("Decode in request failed: ", zap.Error(err)) // } // Log.Debug("Request => ", zap.Any("request", change_request)) // Body var changes models.Recipe01 err = json.NewDecoder(r.Body).Decode(&changes) if err != nil { rr.taoLogger.Log.Error("RecipeRouter.UpdateRecipe", zap.Error(errors.WithMessage(err, "Decode in request failed"))) http.Error(w, "Can't Decode Recipe request body.", http.StatusBadRequest) return } rr.taoLogger.Log.Debug("Changes: ", zap.Any("changes", changes)) // TODO: find the matched pd targetMenu, err := rr.data.GetRecipe01ByProductCode(filename, countryID, changes.ProductCode) if err != nil { rr.taoLogger.Log.Error("RecipeRouter.UpdateRecipe", zap.Error(errors.WithMessage(err, "Error when get recipe by product code"))) http.Error(w, fmt.Sprintf("Recipe country: %s file: %s productCode: %s not found.", country, filename, changes.ProductCode), http.StatusNotFound) return } menuMap := targetMenu.ToMap() changeMap := changes.ToMap() // Find changes for key, val := range menuMap { testBool, err := helpers.DynamicCompare(val, changeMap[key]) if err != nil { rr.taoLogger.Log.Error("RecipeRouter.UpdateRecipe", zap.Error(errors.WithMessage(err, "DynamicCompare in request failed"))) http.Error(w, "Internal Error", http.StatusInternalServerError) return } if !testBool { menuMap[key] = changeMap[key] } } // Apply changes tempRecipe := models.Recipe01{} tempRecipe = tempRecipe.FromMap(menuMap) rr.data.SetValuesToRecipe(tempRecipe) rr.taoLogger.Log.Debug("ApplyChange", zap.Any("status", "passed")) // check if changed // Log.Debug("Check if changed", zap.Any("result", rr.data.GetRecipe01ByProductCode(changes.ProductCode))) file, _ := os.Create(path.Join("./cofffeemachineConfig", countryID, filename)) if err != nil { rr.taoLogger.Log.Error("RecipeRouter.UpdateRecipe", zap.Error(errors.WithMessage(err, "Error when tried to create file"))) http.Error(w, "Internal Error", http.StatusInternalServerError) return } encoder := json.NewEncoder(file) encoder.SetIndent("", " ") err = encoder.Encode(rr.data.GetRecipe(countryID, filename)) if err != nil { rr.taoLogger.Log.Error("RecipeRouter.UpdateRecipe", zap.Error(errors.WithMessage(err, "Error when write file"))) http.Error(w, "Internal Error", http.StatusInternalServerError) return } w.Header().Add("Content-Type", "application/json") if err := json.NewEncoder(w).Encode(&contracts.ResponseDefault{ Status: http.StatusText(http.StatusOK), Message: "Recipe Updated", }); err != nil { rr.taoLogger.Log.Error("RecipeRouter.UpdateRecipe", zap.Error(err)) http.Error(w, "Internal Error", http.StatusInternalServerError) return } }