handler.go 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. package agi
  2. import (
  3. "io/ioutil"
  4. "net/http"
  5. "path/filepath"
  6. "imuslab.com/arozos/mod/utils"
  7. )
  8. //Handle AGI Exectuion Request with token, design for letting other web scripting language like php to interface with AGI
  9. func (g *Gateway) HandleAgiExecutionRequestWithToken(w http.ResponseWriter, r *http.Request) {
  10. token, err := utils.Mv(r, "token", false)
  11. if err != nil {
  12. //Username not defined
  13. utils.SendErrorResponse(w, "Token not defined or empty.")
  14. return
  15. }
  16. script, err := utils.Mv(r, "script", false)
  17. if err != nil {
  18. //Username not defined
  19. utils.SendErrorResponse(w, "Script path not defined or empty.")
  20. return
  21. }
  22. //Try to get the username from token
  23. username, err := g.Option.UserHandler.GetAuthAgent().GetUsernameFromToken(token)
  24. if err != nil {
  25. //This token is not valid
  26. w.WriteHeader(http.StatusUnauthorized)
  27. w.Write([]byte("401 - Unauthorized (Token not valid)"))
  28. return
  29. }
  30. //Check if user exists and have access to the script
  31. targetUser, err := g.Option.UserHandler.GetUserInfoFromUsername(username)
  32. if err != nil {
  33. //This user not exists
  34. w.WriteHeader(http.StatusUnauthorized)
  35. w.Write([]byte("401 - Unauthorized (User not exists)"))
  36. return
  37. }
  38. scriptScope := ""
  39. allowAccess := checkUserAccessToScript(targetUser, script, scriptScope)
  40. if !allowAccess {
  41. w.WriteHeader(http.StatusUnauthorized)
  42. w.Write([]byte("401 - Unauthorized (Permission Denied)"))
  43. return
  44. }
  45. //Get the content of the script
  46. scriptContentByte, err := ioutil.ReadFile(filepath.Join("./web/", script))
  47. if err != nil {
  48. w.WriteHeader(http.StatusNotFound)
  49. w.Write([]byte("404 - Script Not Found"))
  50. return
  51. }
  52. scriptContent := string(scriptContentByte)
  53. g.ExecuteAGIScript(scriptContent, nil, script, scriptScope, w, r, targetUser)
  54. }