mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 15:09:42 +00:00
129 lines
2.5 KiB
Go
129 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/thrasher-/gocryptotrader/exchanges"
|
|
)
|
|
|
|
// RESTLogger logs the requests internally
|
|
func RESTLogger(inner http.Handler, name string) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
start := time.Now()
|
|
|
|
inner.ServeHTTP(w, r)
|
|
|
|
log.Printf(
|
|
"%s\t%s\t%s\t%s",
|
|
r.Method,
|
|
r.RequestURI,
|
|
name,
|
|
time.Since(start),
|
|
)
|
|
})
|
|
}
|
|
|
|
// Route is a sub type that holds the request routes
|
|
type Route struct {
|
|
Name string
|
|
Method string
|
|
Pattern string
|
|
HandlerFunc http.HandlerFunc
|
|
}
|
|
|
|
// Routes is an array of all the registered routes
|
|
type Routes []Route
|
|
|
|
var routes = Routes{}
|
|
|
|
// NewRouter takes in the exchange interfaces and returns a new multiplexor
|
|
// router
|
|
func NewRouter(exchanges []exchange.IBotExchange) *mux.Router {
|
|
router := mux.NewRouter().StrictSlash(true)
|
|
|
|
routes = Routes{
|
|
Route{
|
|
"",
|
|
"GET",
|
|
"/",
|
|
getIndex,
|
|
},
|
|
Route{
|
|
"GetAllSettings",
|
|
"GET",
|
|
"/config/all",
|
|
RESTGetAllSettings,
|
|
},
|
|
Route{
|
|
"SaveAllSettings",
|
|
"POST",
|
|
"/config/all/save",
|
|
RESTSaveAllSettings,
|
|
},
|
|
Route{
|
|
"AllEnabledAccountInfo",
|
|
"GET",
|
|
"/exchanges/enabled/accounts/all",
|
|
RESTGetAllEnabledAccountInfo,
|
|
},
|
|
Route{
|
|
"AllActiveExchangesAndCurrencies",
|
|
"GET",
|
|
"/exchanges/enabled/latest/all",
|
|
RESTGetAllActiveTickers,
|
|
},
|
|
Route{
|
|
"IndividualExchangeAndCurrency",
|
|
"GET",
|
|
"/exchanges/{exchangeName}/latest/{currency}",
|
|
RESTGetTicker,
|
|
},
|
|
Route{
|
|
"GetPortfolio",
|
|
"GET",
|
|
"/portfolio/all",
|
|
RESTGetPortfolio,
|
|
},
|
|
Route{
|
|
"AllActiveExchangesAndOrderbooks",
|
|
"GET",
|
|
"/exchanges/orderbook/latest/all",
|
|
RESTGetAllActiveOrderbooks,
|
|
},
|
|
Route{
|
|
"IndividualExchangeOrderbook",
|
|
"GET",
|
|
"/exchanges/{exchangeName}/orderbook/latest/{currency}",
|
|
RESTGetOrderbook,
|
|
},
|
|
Route{
|
|
"ws",
|
|
"GET",
|
|
"/ws",
|
|
WebsocketClientHandler,
|
|
},
|
|
}
|
|
|
|
for _, route := range routes {
|
|
var handler http.Handler
|
|
handler = route.HandlerFunc
|
|
handler = RESTLogger(handler, route.Name)
|
|
|
|
router.
|
|
Methods(route.Method).
|
|
Path(route.Pattern).
|
|
Name(route.Name).
|
|
Handler(handler)
|
|
}
|
|
return router
|
|
}
|
|
|
|
func getIndex(w http.ResponseWriter, r *http.Request) {
|
|
fmt.Fprint(w, "<html>GoCryptoTrader RESTful interface. For the web GUI, please visit the <a href=https://github.com/thrasher-/gocryptotrader/blob/master/web/README.md>web GUI readme.</a></html>")
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|