summaryrefslogtreecommitdiff
path: root/backend
diff options
context:
space:
mode:
Diffstat (limited to 'backend')
-rw-r--r--backend/main.go67
-rw-r--r--backend/recipe.go74
2 files changed, 139 insertions, 2 deletions
diff --git a/backend/main.go b/backend/main.go
index 7376bd0..c39f123 100644
--- a/backend/main.go
+++ b/backend/main.go
@@ -1,6 +1,8 @@
package main
import "fmt"
+import "io"
+import "io/ioutil"
import "net/http"
import "os"
import "strconv"
@@ -56,7 +58,70 @@ func RecipeList(w http.ResponseWriter, r *http.Request) {
if err := json.NewEncoder(w).Encode(resp); err != nil {
panic(err)
}
+ } else if r.Method == "POST" {
+ var recipe *Recipe
+
+ body, err := ioutil.ReadAll(io.LimitReader(r.Body, 1048576))
+ if err != nil {
+ panic(err)
+ }
+
+ err = r.Body.Close()
+ if err != nil {
+ panic(err)
+ }
+
+ err = json.Unmarshal(body, &recipe)
+ if err != nil {
+ w.WriteHeader(http.StatusUnprocessableEntity)
+ w.Header().Set("Content-Type", "application/json; charset=UTF-8")
+ resp := APIResponseItem{
+ Status: APIError{
+ Code: http.StatusUnprocessableEntity,
+ Msg: "Invalid Recipe"},
+ Data: make([]APIDataRecipe, 0),
+ }
+ if err := json.NewEncoder(w).Encode(resp); err != nil {
+ panic(err)
+ }
+ return
+ }
+
+ err = AddRecipeDB(recipe, db)
+ if err != nil {
+ resp := APIResponseItem{
+ Status: APIError{Code: http.StatusBadRequest,
+ Msg: "Recipe could not be added"},
+ Data: make([]APIDataRecipe, 0),
+ }
+
+ resp.Data = append(resp.Data, APIDataRecipe{recipe})
+
+ w.Header().Set("Content-Type",
+ "application/json; charset=UTF-8")
+ w.WriteHeader(http.StatusBadRequest)
+ if err := json.NewEncoder(w).Encode(resp); err != nil {
+ panic(err)
+ }
+ return
+ }
+
+ resp := APIResponseItem{
+ Status: APIError{Code: http.StatusCreated,
+ Msg: "Recipe added successfully"},
+ Data: make([]APIDataRecipe, 0),
+ }
+
+ resp.Data = append(resp.Data, APIDataRecipe{recipe})
+
+ w.Header().Set("Content-Type", "application/json; charset=UTF-8")
+ w.WriteHeader(http.StatusCreated)
+ if err := json.NewEncoder(w).Encode(resp); err != nil {
+ panic(err)
+ }
+
}
+
}
func SingleRecipe(w http.ResponseWriter, r *http.Request) {
@@ -77,7 +142,6 @@ func SingleRecipe(w http.ResponseWriter, r *http.Request) {
status = http.StatusOK
msg = "Successful"
}
- fmt.Println(status, msg, recipe)
resp := APIResponseItem{
Status: APIError{Code: status, Msg: msg},
@@ -119,7 +183,6 @@ func main() {
dbinfo := fmt.Sprintf("user=%s password=%s dbname=%s sslmode=disable",
DB_USER, DB_PASSWORD, DB_NAME)
- fmt.Println(dbinfo)
db, err = sql.Open("postgres", dbinfo)
if err != nil || db.Ping() != nil {
fmt.Println("Error connecting to database")
diff --git a/backend/recipe.go b/backend/recipe.go
index ceae402..e14c79b 100644
--- a/backend/recipe.go
+++ b/backend/recipe.go
@@ -23,6 +23,7 @@ type Recipe struct {
Serving_size int
Cook_time int
Rating int
+ Num_cooked int
Keywords []string
Ingredients []Ingredient
Steps []Step
@@ -38,6 +39,7 @@ func MakeRecipe() *Recipe {
Serving_size: 0,
Cook_time: 0,
Rating: 5,
+ Num_cooked: 0,
Keywords: make([]string, 0),
Ingredients: make([]Ingredient, 0),
Steps: make([]Step, 0)}
@@ -73,6 +75,7 @@ func RecipeFromId(id int, db *sql.DB) *Recipe {
Serving_size: servings,
Cook_time: cook_time,
Rating: rating,
+ Num_cooked: num_cooked,
Keywords: make([]string, 0),
Ingredients: make([]Ingredient, 0),
Steps: make([]Step, 0)}
@@ -126,3 +129,74 @@ func RecipeFromId(id int, db *sql.DB) *Recipe {
return &rec
}
+
+func AddRecipeDB(r *Recipe, db *sql.DB) error {
+ var photo_urls, keywords strings.Builder
+ var id int
+
+ for _, u := range r.Photos {
+ photo_urls.WriteString(u)
+ photo_urls.WriteRune('|')
+ }
+
+ for _, k := range r.Keywords {
+ keywords.WriteString(k)
+ keywords.WriteRune('|')
+ }
+
+ err := db.QueryRow(`INSERT INTO recipes (title, photo_urls,
+ keywords, description, serving_size, cook_time,
+ rating, num_cooked)
+ VALUES ($1, $2, $3, $4, $5, $6, $7, $8)
+ RETURNING id`,
+ r.Title, //1
+ photo_urls.String(), //2
+ keywords.String(), //3
+ r.Desc, //4
+ r.Serving_size, //5
+ r.Cook_time, //6
+ r.Rating, //7
+ r.Num_cooked, //8
+ ).Scan(&id)
+
+ if err != nil {
+ return err
+ }
+
+ var ingr_id int
+ for i, ingr := range r.Ingredients {
+ err := db.QueryRow(`INSERT INTO ingredients
+ (name, amount, unit, recipe_id)
+ VALUES ($1, $2, $3, $4)
+ RETURNING id`,
+ ingr.Name,
+ ingr.Amount,
+ ingr.Unit,
+ id,
+ ).Scan(&ingr_id)
+ if err != nil {
+ r.Ingredients = append(r.Ingredients[:i],
+ r.Ingredients[i+1:]...)
+ }
+ }
+
+ var step_id int
+ for i, step := range r.Steps {
+ err := db.QueryRow(`INSERT INTO steps
+ (step, description, timer, recipe_id)
+ VALUES ($1, $2, $3, $4)
+ RETURNING id`,
+ step.Num,
+ step.Desc,
+ step.Time,
+ id,
+ ).Scan(&step_id)
+ if err != nil {
+ r.Steps = append(r.Steps[:i],
+ r.Steps[i+1:]...)
+ }
+ }
+
+ r.Id = id
+ return nil
+}