api/routes.go

   1 package api
   2
   3 import (
   4         "encoding/base64"
   5
   6         "github.com/gin-gonic/gin"
   7 )
   8
   9 type Route struct {
  10         Method   string
  11         Handlers []gin.HandlerFunc
  12         Path     string
  13 }
  14
  15 type Group struct {
  16         Root        string
  17         Middlewares []Middleware
  18         Routes      []Route
  19 }
  20
  21 var Groups = []Group{
  22         {
  23                 "",
  24                 nil,
  25                 []Route{
  26                         {"POST", []gin.HandlerFunc{Signup}, "/signup"},
  27                         {"POST", []gin.HandlerFunc{Signin}, "/signin"},
  28                         {"POST", []gin.HandlerFunc{PasswordReset}, "/passwordreset"},
  29                         {"POST", []gin.HandlerFunc{ChangePassword}, "/changepassword"},
  30                 },
  31         },
  32         {
  33                 "/otp",
  34                 []Middleware{JwtAuth, UserConfirmed},
  35                 []Route{
  36                         {"GET", []gin.HandlerFunc{OtpEnrollment}, "/enroll"},
  37                         {"POST", []gin.HandlerFunc{OtpValidate}, "/validate"},
  38                 },
  39         },
  40         {
  41                 "/market",
  42                 []Middleware{JwtAuth, UserConfirmed, OtpAuth},
  43                 []Route{
  44                         {"GET", []gin.HandlerFunc{GetMarketConfig}, "/:name"},
  45                         {"POST", []gin.HandlerFunc{UpdateMarketConfig}, "/:name/update"},
  46                         {"GET", []gin.HandlerFunc{GetMarketBalance}, "/:name/balance"},
  47                 },
  48         },
  49 }
  50
  51 func Signup(c *gin.Context) {
  52         query := &SignupQuery{}
  53
  54         query.In.Email = c.PostForm("email")
  55         query.In.Password = c.PostForm("password")
  56
  57         RunQuery(query, c)
  58 }
  59
  60 func Signin(c *gin.Context) {
  61         query := &SigninQuery{}
  62
  63         query.In.Email = c.PostForm("email")
  64         query.In.Password = c.PostForm("password")
  65
  66         RunQuery(query, c)
  67 }
  68
  69 func OtpValidate(c *gin.Context) {
  70         query := &OtpValidateQuery{}
  71
  72         query.In.Pass = c.PostForm("pass")
  73         query.In.User = GetUser(c)
  74         query.In.Claims = GetClaims(c)
  75
  76         RunQuery(query, c)
  77 }
  78
  79 func OtpEnrollment(c *gin.Context) {
  80         query := &OtpEnrollmentQuery{}
  81
  82         query.In.User = GetUser(c)
  83
  84         qrPng, secret, err := query.Run()
  85         if err != nil {
  86                 WriteJsonResponse(ErrorResponse(err.Code, err.UserMessage), c)
  87                 c.Error(err)
  88                 return
  89         }
  90
  91         if c.Query("format") == "png" {
  92                 c.Header("X-OTP-Secret", secret)
  93                 WriteBinary("image/png", qrPng.Bytes(), c)
  94         } else {
  95                 response := struct {
  96                         Base64img string `json:"base64img"`
  97                         OtpSecret string `json:"secret"`
  98                 }{
  99                         base64.StdEncoding.EncodeToString(qrPng.Bytes()),
 100                         secret,
 101                 }
 102
 103                 WriteJsonResponse(SuccessResponse(response), c)
 104         }
 105
 106 }
 107
 108 func GetMarketConfig(c *gin.Context) {
 109         query := &MarketConfigQuery{}
 110
 111         query.In.User = GetUser(c)
 112         query.In.Market = c.Param("name")
 113
 114         RunQuery(query, c)
 115 }
 116
 117 func GetMarketBalance(c *gin.Context) {
 118         query := &MarketBalanceQuery{}
 119
 120         query.In.User = GetUser(c)
 121         query.In.Market = c.Param("name")
 122         query.In.Currency = c.Query("currency")
 123
 124         RunQuery(query, c)
 125 }
 126
 127 func UpdateMarketConfig(c *gin.Context) {
 128         query := &UpdateMarketConfigQuery{}
 129
 130         query.In.User = GetUser(c)
 131         query.In.Market = c.Param("name")
 132         query.In.Key = c.PostForm("key")
 133         query.In.Secret = c.PostForm("secret")
 134
 135         RunQuery(query, c)
 136 }
 137
 138 func PasswordReset(c *gin.Context) {
 139         query := &PasswordResetQuery{}
 140
 141         query.In.Email = c.PostForm("email")
 142
 143         RunQuery(query, c)
 144 }
 145
 146 func ChangePassword(c *gin.Context) {
 147         query := &ChangePasswordQuery{}
 148
 149         query.In.Token = c.PostForm("token")
 150         query.In.Password = c.PostForm("password")
 151
 152         RunQuery(query, c)
 153 }