actions.go 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127
  1. package main
  2. import (
  3. "encoding/hex"
  4. "errors"
  5. "math/rand"
  6. "net/http"
  7. "sparrow/pkg/models"
  8. "sparrow/pkg/rpcs"
  9. "sparrow/pkg/server"
  10. "sparrow/pkg/token"
  11. "github.com/martini-contrib/render"
  12. )
  13. const (
  14. ErrOK = 0
  15. ErrSystemFault = 10001
  16. ErrDeviceNotFound = 10002
  17. ErrWrongSecret = 10003
  18. ErrProtocolNotSuported = 10004
  19. )
  20. func renderError(code int, err error) Common {
  21. result := Common{}
  22. result.Code = code
  23. result.Message = err.Error()
  24. server.Log.Error(err.Error())
  25. return result
  26. }
  27. // DeviceRegisterArgs device register args
  28. type DeviceRegisterArgs struct {
  29. ProductKey string `json:"product_key" binding:"required"`
  30. DeviceCode string `json:"device_code" binding:"required"`
  31. Version string `json:"version" binding:"required"`
  32. ModuleName string `json:"module" binding:"required"`
  33. }
  34. // DeviceAuthArgs device authentication args
  35. type DeviceAuthArgs struct {
  36. DeviceId int64 `json:"device_id" binding:"required"`
  37. DeviceSecret string `json:"device_secret" binding:"required"`
  38. Protocol string `json:"protocol" binding:"required"`
  39. }
  40. // RegisterDevice 设备激活
  41. func RegisterDevice(args DeviceRegisterArgs, r render.Render) {
  42. server.Log.Printf("ACTION RegisterDevice, args:: %v ", args)
  43. rpcargs := &rpcs.ArgsDeviceRegister{
  44. ProductKey: args.ProductKey,
  45. DeviceCode: args.DeviceCode,
  46. DeviceVersion: args.Version,
  47. ModuleName: args.ModuleName,
  48. }
  49. device := &models.Device{}
  50. err := server.RPCCallByName("registry", "Registry.RegisterDevice", rpcargs, device)
  51. if err != nil {
  52. r.JSON(http.StatusOK, renderError(ErrSystemFault, err))
  53. return
  54. }
  55. server.Log.Infof("register device success: %v", device)
  56. result := DeviceRegisterResponse{}
  57. result.Data = DeviceRegisterData{
  58. DeviceId: int64(device.ID),
  59. DeviceSecret: device.DeviceSecret,
  60. DeviceKey: device.DeviceKey,
  61. DeviceIdentifier: device.DeviceIdentifier,
  62. }
  63. r.JSON(http.StatusOK, result)
  64. return
  65. }
  66. // AuthDevice device auth
  67. func AuthDevice(args DeviceAuthArgs, r render.Render) {
  68. server.Log.Printf("ACTION AuthDevice, args:: %v", args)
  69. device := &models.Device{}
  70. err := server.RPCCallByName("registry", "Registry.FindDeviceById", int64(args.DeviceId), device)
  71. if err != nil {
  72. r.JSON(http.StatusOK, renderError(ErrDeviceNotFound, err))
  73. return
  74. }
  75. if device.DeviceSecret != args.DeviceSecret {
  76. // device secret is wrong.
  77. r.JSON(http.StatusOK, renderError(ErrWrongSecret, errors.New("wrong device secret.")))
  78. return
  79. }
  80. hepler := token.NewHelper(*confRedisHost)
  81. token, err := hepler.GenerateToken(uint64(int64(device.ID)))
  82. if err != nil {
  83. r.JSON(http.StatusOK, renderError(ErrSystemFault, err))
  84. return
  85. }
  86. var hosts []string
  87. switch args.Protocol {
  88. case "http":
  89. hosts, err = server.GetServerHosts(args.Protocol+"access", "httphost")
  90. case "mqtt":
  91. hosts, err = server.GetServerHosts(args.Protocol+"access", "tcphost")
  92. case "coap":
  93. hosts, err = server.GetServerHosts(args.Protocol+"access", "udphost")
  94. default:
  95. err = errors.New("unsuported protocol: " + args.Protocol)
  96. }
  97. if err != nil {
  98. r.JSON(http.StatusOK, renderError(ErrProtocolNotSuported, err))
  99. return
  100. }
  101. // just get a random host
  102. host := hosts[rand.Intn(len(hosts))]
  103. result := DeviceAuthResponse{}
  104. result.Data = DeviceAuthData{
  105. AccessToken: hex.EncodeToString(token),
  106. AccessAddr: host,
  107. }
  108. server.Log.Infof("auth device success: %v, token: %x, access: %v", device, token, host)
  109. r.JSON(http.StatusOK, result)
  110. return
  111. }