rest.go 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190
  1. package rest_http
  2. import (
  3. "bytes"
  4. "encoding/json"
  5. "log"
  6. "net/http"
  7. "strings"
  8. mjson "git.ali33.ru/fcg-xvii/go-tools/json"
  9. "git.ali33.ru/fcg-xvii/rest"
  10. )
  11. func New(app rest.IApplication, core any, responseHeaders map[string]string) *Rest {
  12. return &Rest{
  13. app: app,
  14. core: core,
  15. responseHeaders: responseHeaders,
  16. }
  17. }
  18. type Rest struct {
  19. server rest.IServer
  20. app rest.IApplication
  21. core any
  22. responseHeaders map[string]string
  23. Debug bool
  24. }
  25. func (s *Rest) App() rest.IApplication {
  26. return s.app
  27. }
  28. func (s *Rest) Prepare(server rest.IServer, httpPrefix string) {
  29. server.HandleFunc(httpPrefix, s.handle)
  30. s.server = server
  31. }
  32. func responseNotFound(w http.ResponseWriter) {
  33. w.WriteHeader(404)
  34. }
  35. func responseError(w http.ResponseWriter, err rest.IErrorArgs, code int) {
  36. w.WriteHeader(code)
  37. w.Write([]byte(err.Error()))
  38. }
  39. // handle
  40. func (s *Rest) handle(w http.ResponseWriter, r *http.Request) {
  41. // установка заголовков ответа
  42. for k, v := range s.responseHeaders {
  43. w.Header().Set(k, v)
  44. }
  45. // запросы prelight
  46. if r.Method == "OPTIONS" {
  47. return
  48. }
  49. // Инициализация rest Request
  50. rr := &RequestIn{
  51. RequestIn: &rest.RequestIn{
  52. IRequest: &rest.Request{
  53. Type: rest.RequestTypeIn,
  54. Command: r.URL.Path,
  55. Data: mjson.Map{},
  56. Files: make(rest.RequestFiles),
  57. },
  58. GeneratorToken: s.server.TokenGenerate,
  59. ParserToken: s.server.TokenParse,
  60. Core: s.core,
  61. },
  62. }
  63. // Парсим Bearer токен и извлекаем claims
  64. authHeader := r.Header.Get("Authorization")
  65. if authHeader != "" {
  66. if parts := strings.Split(authHeader, " "); len(parts) == 2 && parts[0] == "Bearer" {
  67. tokenString := parts[1]
  68. auth, err := s.server.TokenParse(tokenString)
  69. if err == nil {
  70. rr.auth = auth
  71. }
  72. }
  73. }
  74. // Если это многокомпонентный запрос, обрабатываем файлы
  75. if strings.Index(r.Header.Get("Content-Type"), "multipart/form-data") == 0 {
  76. if s.Debug {
  77. log.Println("request multipart")
  78. }
  79. err := r.ParseMultipartForm(32 << 20) // max memory 32MB, после этого файлы будут сохранены во временных файлах
  80. log.Println(err)
  81. if err != nil {
  82. responseError(w, rest.ErrorMessage("ErrMultipartParse", err.Error()), 500)
  83. return
  84. }
  85. multiPartForm := r.MultipartForm
  86. data, check := multiPartForm.Value["data"]
  87. if check {
  88. if s.Debug {
  89. log.Println("data accepted")
  90. }
  91. rData := rr.RData()
  92. err := json.NewDecoder(bytes.NewBuffer([]byte(data[0]))).Decode(&rData)
  93. if err != nil {
  94. responseError(w, rest.ErrorMessage("ErrMultipartDataParse", err.Error()), 500)
  95. return
  96. }
  97. if s.Debug {
  98. rData.LogPretty()
  99. }
  100. }
  101. files := rr.RFiles()
  102. for filename, headers := range multiPartForm.File {
  103. for _, header := range headers {
  104. file, err := header.Open()
  105. if err != nil {
  106. err := rest.NewError(
  107. "ErrMultipartFileParse",
  108. mjson.Map{
  109. "filename": filename,
  110. "message": err.Error(),
  111. },
  112. )
  113. responseError(w, err, 500)
  114. }
  115. upFile := rest.NewReadCloserLen(
  116. file,
  117. header.Size,
  118. )
  119. rFile := &rest.MimeFile{
  120. Name: filename,
  121. MimeType: header.Header.Get("Content-Type"),
  122. File: upFile,
  123. }
  124. files[filename] = rFile
  125. }
  126. }
  127. defer rr.RClose()
  128. } else {
  129. if s.Debug {
  130. log.Println("request json")
  131. }
  132. data := rr.RData()
  133. err := json.NewDecoder(r.Body).Decode(&data)
  134. if err != nil {
  135. responseError(w, rest.ErrorMessage("ErrDataParse", err.Error()), 500)
  136. return
  137. }
  138. if s.Debug {
  139. data.LogPretty()
  140. }
  141. }
  142. // get command
  143. //command, check := s.commands.GetCommand(r.URL.Path)
  144. command, check := s.app.Executer(rr)
  145. if !check {
  146. responseNotFound(w)
  147. return
  148. }
  149. //log.Println(rr.RData)
  150. // serialize
  151. if err := rest.Serialize(rr.RData(), command); err != nil {
  152. responseError(w, err, 500)
  153. return
  154. }
  155. // validate
  156. if validator, check := command.(rest.IValidator); check {
  157. resp := validator.Validate(rr)
  158. if resp != nil {
  159. if err := resp.Write(w); err != nil {
  160. responseError(w, err, 500)
  161. }
  162. return
  163. }
  164. }
  165. // execute
  166. resp := command.Execute(rr)
  167. if err := resp.Write(w); err != nil {
  168. responseError(w, err, 500)
  169. }
  170. resp.RClose()
  171. }