package api
import (
"encoding/base64"
"github.com/gin-gonic/gin"
)
type Route struct {
Method string
Handlers []gin.HandlerFunc
Path string
}
type Group struct {
Root string
Middlewares []Middleware
Routes []Route
}
var Groups = []Group{
{
"",
nil,
[]Route{
{"POST", []gin.HandlerFunc{Signup}, "/signup"},
{"POST", []gin.HandlerFunc{Signin}, "/signin"},
{"POST", []gin.HandlerFunc{PasswordReset}, "/passwordreset"},
{"POST", []gin.HandlerFunc{ChangePassword}, "/changepassword"},
{"POST", []gin.HandlerFunc{ConfirmEmail}, "/confirmemail"},
},
},
{
"/otp",
[]Middleware{JwtAuth, UserConfirmed},
[]Route{
{"GET", []gin.HandlerFunc{OtpEnrollment}, "/enroll"},
{"POST", []gin.HandlerFunc{OtpValidate}, "/validate"},
},
},
{
"/market",
[]Middleware{JwtAuth, UserConfirmed, OtpAuth},
[]Route{
{"GET", []gin.HandlerFunc{GetMarketConfig}, "/:name"},
{"POST", []gin.HandlerFunc{UpdateMarketConfig}, "/:name/update"},
{"GET", []gin.HandlerFunc{TestMarketCredentials}, "/:name/test-credentials"},
{"GET", []gin.HandlerFunc{GetPortfolio}, "/:name/portfolio"},
},
},
{
"/user",
[]Middleware{JwtAuth, UserConfirmed, OtpAuth},
[]Route{
{"GET", []gin.HandlerFunc{UserAccount}, "/account"},
},
},
{
"/admin",
[]Middleware{JwtAuth, UserConfirmed, UserIsAdmin, OtpAuth},
[]Route{
{"GET", []gin.HandlerFunc{AdminGetAllPortfolios}, "/portfolios"},
},
},
}
func Signup(c *gin.Context) {
query := &SignupQuery{}
query.In.Email = c.PostForm("email")
query.In.Password = c.PostForm("password")
RunQuery(query, c)
}
func Signin(c *gin.Context) {
query := &SigninQuery{}
query.In.Email = c.PostForm("email")
query.In.Password = c.PostForm("password")
RunQuery(query, c)
}
func OtpValidate(c *gin.Context) {
query := &OtpValidateQuery{}
query.In.Pass = c.PostForm("pass")
query.In.User = GetUser(c)
query.In.Claims = GetClaims(c)
RunQuery(query, c)
}
func OtpEnrollment(c *gin.Context) {
query := &OtpEnrollmentQuery{}
query.In.User = GetUser(c)
qrPng, secret, err := query.Run()
if err != nil {
WriteJsonResponse(ErrorResponse(err.Flag(), err.UserMessage), c)
c.Error(err)
return
}
if c.Query("format") == "png" {
c.Header("X-OTP-Secret", secret)
WriteBinary("image/png", qrPng.Bytes(), c)
} else {
response := struct {
Base64img string `json:"base64img"`
OtpSecret string `json:"secret"`
}{
base64.StdEncoding.EncodeToString(qrPng.Bytes()),
secret,
}
WriteJsonResponse(SuccessResponse(response), c)
}
}
func GetMarketConfig(c *gin.Context) {
query := &MarketConfigQuery{}
query.In.User = GetUser(c)
query.In.Market = c.Param("name")
RunQuery(query, c)
}
func TestMarketCredentials(c *gin.Context) {
query := &TestMarketCredentialsQuery{}
query.In.User = GetUser(c)
query.In.Market = c.Param("name")
RunQuery(query, c)
}
func GetPortfolio(c *gin.Context) {
query := &GetPortfolioQuery{}
query.In.User = GetUser(c)
query.In.Market = c.Param("name")
RunQuery(query, c)
}
func UpdateMarketConfig(c *gin.Context) {
query := &UpdateMarketConfigQuery{}
query.In.User = GetUser(c)
query.In.Market = c.Param("name")
query.In.Key = c.PostForm("key")
query.In.Secret = c.PostForm("secret")
RunQuery(query, c)
}
func PasswordReset(c *gin.Context) {
query := &PasswordResetQuery{}
query.In.Email = c.PostForm("email")
RunQuery(query, c)
}
func ChangePassword(c *gin.Context) {
query := &ChangePasswordQuery{}
query.In.Token = c.PostForm("token")
query.In.Password = c.PostForm("password")
RunQuery(query, c)
}
func ConfirmEmail(c *gin.Context) {
query := &ConfirmEmailQuery{}
query.In.Token = c.PostForm("token")
RunQuery(query, c)
}
func UserAccount(c *gin.Context) {
query := &UserAccountQuery{}
query.In.User = GetUser(c)
RunQuery(query, c)
}
func AdminGetAllPortfolios(c *gin.Context) {
query := &GetAllPortfoliosQuery{}
query.In.Market = "poloniex"
RunQuery(query, c)
}
