rest.go 3.9 KB

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