af83/edwig

package api

import (
    "encoding/json"
    "fmt"
    "net/http"
    "net/url"

    "bitbucket.org/enroute-mobi/ara/core"
    "bitbucket.org/enroute-mobi/ara/logger"
    "bitbucket.org/enroute-mobi/ara/model"
)

type StopAreaController struct {
    referential *core.Referential
}

func NewStopAreaController(referential *core.Referential) ControllerInterface {
    return &Controller{
        restfulResource: &StopAreaController{
            referential: referential,
        },
    }
}

func (controller *StopAreaController) findStopArea(identifier string) (*model.StopArea, bool) {
    foundStrings := idPattern.FindStringSubmatch(identifier)
    if foundStrings != nil {
        code := model.NewCode(foundStrings[1], foundStrings[2])
        return controller.referential.Model().StopAreas().FindByCode(code)
    }
    return controller.referential.Model().StopAreas().Find(model.StopAreaId(identifier))
}

func (controller *StopAreaController) Index(response http.ResponseWriter, filters url.Values) {
    logger.Log.Debugf("StopAreas Index")

    stime := controller.referential.Clock().Now()
    sas := controller.referential.Model().StopAreas().FindAll()
    logger.Log.Debugf("StopAreaController FindAll time : %v", controller.referential.Clock().Since(stime))
    stime = controller.referential.Clock().Now()
    jsonBytes, _ := json.Marshal(sas)
    logger.Log.Debugf("StopAreaController Json Marshal time : %v ", controller.referential.Clock().Since(stime))
    response.Write(jsonBytes)
}

func (controller *StopAreaController) Show(response http.ResponseWriter, identifier string) {
    stopArea, ok := controller.findStopArea(identifier)
    if !ok {
        http.Error(response, fmt.Sprintf("Stop area not found: %s", identifier), http.StatusNotFound)
        return
    }
    logger.Log.Debugf("Get stopArea %s", identifier)

    jsonBytes, _ := stopArea.MarshalJSON()
    response.Write(jsonBytes)
}

func (controller *StopAreaController) Delete(response http.ResponseWriter, identifier string) {
    stopArea, ok := controller.findStopArea(identifier)
    if !ok {
        http.Error(response, fmt.Sprintf("Stop area not found: %s", identifier), http.StatusNotFound)
        return
    }
    logger.Log.Debugf("Delete stopArea %s", identifier)

    jsonBytes, _ := stopArea.MarshalJSON()
    controller.referential.Model().StopAreas().Delete(stopArea)
    response.Write(jsonBytes)
}

func (controller *StopAreaController) Update(response http.ResponseWriter, identifier string, body []byte) {
    stopArea, ok := controller.findStopArea(identifier)
    if !ok {
        http.Error(response, fmt.Sprintf("Stop area not found: %s", identifier), http.StatusNotFound)
        return
    }

    logger.Log.Debugf("Update stopArea %s: %s", identifier, string(body))

    err := json.Unmarshal(body, &stopArea)
    if err != nil {
        http.Error(response, fmt.Sprintf("Invalid request: can't parse request body: %v", err), http.StatusBadRequest)
        return
    }

    for _, obj := range stopArea.Codes() {
        sa, ok := controller.referential.Model().StopAreas().FindByCode(obj)
        if ok && sa.Id() != stopArea.Id() {
            http.Error(response, fmt.Sprintf("Invalid request: stopArea %v already have a code %v", sa.Id(), obj.String()), http.StatusBadRequest)
            return
        }
    }

    controller.referential.Model().StopAreas().Save(stopArea)
    jsonBytes, _ := stopArea.MarshalJSON()
    response.Write(jsonBytes)
}

func (controller *StopAreaController) Create(response http.ResponseWriter, body []byte) {
    logger.Log.Debugf("Create stopArea: %s", string(body))

    stopArea := controller.referential.Model().StopAreas().New()

    err := json.Unmarshal(body, &stopArea)
    if err != nil {
        http.Error(response, fmt.Sprintf("Invalid request: can't parse request body: %v", err), http.StatusBadRequest)
        return
    }

    if stopArea.Id() != "" {
        http.Error(response, "Invalid request", http.StatusBadRequest)
        return
    }

    for _, obj := range stopArea.Codes() {
        sa, ok := controller.referential.Model().StopAreas().FindByCode(obj)
        if ok {
            http.Error(response, fmt.Sprintf("Invalid request: stopArea %v already have a code %v", sa.Id(), obj.String()), http.StatusBadRequest)
            return
        }
    }

    controller.referential.Model().StopAreas().Save(stopArea)
    jsonBytes, _ := stopArea.MarshalJSON()
    response.Write(jsonBytes)
}