json.go

package main

import (
	"fmt"
	"net/http"

	"github.com/gorilla/websocket"
	log "github.com/sirupsen/logrus"
)

const contentTypeJSON = "application/json; charset=utf-8"

func WriteError(w http.ResponseWriter, errStr string, secretErr error) error {
	return WriteErrorStatus(w, errStr, secretErr, http.StatusInternalServerError)
}

func WriteErrorStatus(w http.ResponseWriter, errStr string, secretErr error, status int) error {
	log.Debugf("Real error: %v", secretErr)
	log.Debugf("Returning HTTP %d w/error: %q", status, errStr)

	w.Header().Set("Content-Type", contentTypeJSON)
	w.WriteHeader(status)
	_, err := fmt.Fprintf(w, `{"error":%q}`, errStr)
	return err
}

// WebSockets

func WSWriteError(wsConn *websocket.Conn, errStr string, secretErr error) error {
	log.Debugf("WebSocket error: " + secretErr.Error())

	wsErr := fmt.Sprintf(`{"error":%q}`, errStr)
	err := wsConn.WriteMessage(websocket.TextMessage, []byte(wsErr))
	wsConn.Close() // TODO: Will this panic?
	return err
}