af83/edwig

package api

import (
    "net/http"

    "bitbucket.org/enroute-mobi/ara/api/rah"
    "bitbucket.org/enroute-mobi/ara/audit"
    "bitbucket.org/enroute-mobi/ara/core"
)

type SIRILiteRequestHandler interface {
    ConnectorType() string
    Respond(core.Connector, http.ResponseWriter, *audit.BigQueryMessage)
}

type SIRILiteHandler struct {
    rah.RemoteAddressHandler

    referential *core.Referential
    token       string
}

func NewSIRILiteHandler(referential *core.Referential, token string) *SIRILiteHandler {
    return &SIRILiteHandler{
        referential: referential,
        token:       token,
    }
}

func (handler *SIRILiteHandler) requestHandler(requestData *SIRIRequestData) SIRILiteRequestHandler {
    switch requestData.Request {
    case "vehicle-monitoring":
        return &SIRILiteVehicleMonitoringRequestHandler{
            requestUrl:  requestData.Url,
            filters:     requestData.Filters,
            referential: handler.referential,
        }
    }
    return nil
}

func (handler *SIRILiteHandler) serve(response http.ResponseWriter, request *http.Request, requestData *SIRIRequestData) {
    if handler.token == "" {
        http.Error(response, "No Authorization Token", http.StatusUnauthorized)
        return
    }

    if handler.referential == nil {
        http.Error(response, "Referential not found", http.StatusNotFound)
        return
    }

    // Find Partner by authorization Key
    partner, ok := handler.referential.Partners().FindByCredential(handler.token)
    if !ok {
        http.Error(response, "Invalid Authorization Token", http.StatusForbidden)
        return
    }

    requestHandler := handler.requestHandler(requestData)
    if requestHandler == nil {
        http.Error(response, "The SIRI Lite request doesn’t match a defined broadcast", http.StatusNotFound)
        return
    }

    connector, ok := partner.Connector(requestHandler.ConnectorType())
    if !ok {
        http.Error(response, "The Partner don't support this SIRI Lite request", http.StatusNotFound)
        return
    }

    response.Header().Set("Content-Type", "application/json")

    m := &audit.BigQueryMessage{
        Protocol:  "siri-lite",
        Direction: "received",
        Partner:   string(partner.Slug()),
        IPAddress: handler.HandleRemoteAddress(request),
        Status:    "OK",
    }

    requestHandler.Respond(connector, response, m)
}