order.go

package store

import (
	"errors"
	"log"
	"time"

	"github.com/incrypt0/cokut-server/models"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/bson/primitive"
)

// CreateOrder creates a new order
func (s *Store) CreateOrder(o *models.Order) (po *models.Order, err error) {
	c := s.orders // Basic Validation

	o.Time = primitive.NewDateTimeFromTime(time.Now())

	if err = o.Validate(); err != nil {
		return po, err
	}

	if err = s.processOrder(o); err != nil {
		return po, err
	}

	if id, err := s.w.Add(c, o); err != nil {
		return nil, err
	} else if o.ID, err = primitive.ObjectIDFromHex(id); err != nil {
		return nil, err
	}

	return o, err
}

func (s *Store) processOrder(o *models.Order) error {
	mealCollection := s.mc
	ids := make([]string, 0, len(o.Items))

	for key := range o.Items {
		log.Println(key, "_____", len(o.Items))
		ids = append(ids, key)
	}

	rid, err := primitive.ObjectIDFromHex(o.RID)
	if err != nil {
		return err
	}

	if r, err := s.w.FindOne(s.rc, models.Restaurant{ID: rid}); err != nil {
		return err
	} else if r.(*models.Restaurant).Closed {
		return errors.New("RESTAURANT CLOSED")
	}

	l, err := s.w.GetMultipleByID(mealCollection, models.Meal{}, ids)

	s.calculateOrderPrice(o, l)
	s.calculateDeliveryCharge(o)
	s.calculateTotal(o)

	if err != nil {
		return err
	}

	return err
}

func (s *Store) calculateOrderPrice(o *models.Order, l []interface{}) {
	for _, item := range l {
		meal := item.(*models.Meal)
		count := o.Items[meal.ID.Hex()]
		price := meal.Price * float64(count)

		o.Summary = append(o.Summary, models.Summary{Meal: *meal, Count: count, Price: price})
		o.Price += price
	}
}

func (s *Store) calculateDeliveryCharge(o *models.Order) {
	o.DeliveryCharge = 20
}

func (s *Store) calculateTotal(o *models.Order) {
	o.Total = o.Price + o.DeliveryCharge
}

// GetAllOrders Admin only function
func (s *Store) GetAllOrders() (l []interface{}, err error) {
	return s.w.Get(s.orders, models.Order{})
}

// GetOrdersByUser user orders are returned
func (s *Store) GetOrdersByUser(uid string) (l []interface{}, err error) {
	return s.w.Get(s.orders, bson.M{"uid": uid})
}