whatsapp.go 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333
  1. // Code generated by ent, DO NOT EDIT.
  2. package ent
  3. import (
  4. "encoding/json"
  5. "fmt"
  6. "strings"
  7. "time"
  8. "wechat-api/ent/agent"
  9. "wechat-api/ent/whatsapp"
  10. "entgo.io/ent"
  11. "entgo.io/ent/dialect/sql"
  12. )
  13. // Whatsapp is the model entity for the Whatsapp schema.
  14. type Whatsapp struct {
  15. config `json:"-"`
  16. // ID of the ent.
  17. ID uint64 `json:"id,omitempty"`
  18. // Create Time | 创建日期
  19. CreatedAt time.Time `json:"created_at,omitempty"`
  20. // Update Time | 修改日期
  21. UpdatedAt time.Time `json:"updated_at,omitempty"`
  22. // Status 1: normal 2: ban | 状态 1 正常 2 禁用
  23. Status uint8 `json:"status,omitempty"`
  24. // Delete Time | 删除日期
  25. DeletedAt time.Time `json:"deleted_at,omitempty"`
  26. // ak
  27. Ak string `json:"ak,omitempty"`
  28. // 端口号
  29. Sk string `json:"sk,omitempty"`
  30. // 回调地址
  31. Callback string `json:"callback,omitempty"`
  32. // 微信账号
  33. Account string `json:"account,omitempty"`
  34. // 昵称
  35. Nickname string `json:"nickname,omitempty"`
  36. // 手机号
  37. Phone string `json:"phone,omitempty"`
  38. // 机构 ID
  39. OrganizationID uint64 `json:"organization_id,omitempty"`
  40. // 模式ID
  41. AgentID uint64 `json:"agent_id,omitempty"`
  42. // 大模型服务地址
  43. APIBase string `json:"api_base,omitempty"`
  44. // 大模型服务密钥
  45. APIKey string `json:"api_key,omitempty"`
  46. // 白名单
  47. AllowList []string `json:"allow_list,omitempty"`
  48. // 群白名单
  49. GroupAllowList []string `json:"group_allow_list,omitempty"`
  50. // 黑名单
  51. BlockList []string `json:"block_list,omitempty"`
  52. // 群黑名单
  53. GroupBlockList []string `json:"group_block_list,omitempty"`
  54. // Edges holds the relations/edges for other nodes in the graph.
  55. // The values are being populated by the WhatsappQuery when eager-loading is set.
  56. Edges WhatsappEdges `json:"edges"`
  57. selectValues sql.SelectValues
  58. }
  59. // WhatsappEdges holds the relations/edges for other nodes in the graph.
  60. type WhatsappEdges struct {
  61. // Agent holds the value of the agent edge.
  62. Agent *Agent `json:"agent,omitempty"`
  63. // loadedTypes holds the information for reporting if a
  64. // type was loaded (or requested) in eager-loading or not.
  65. loadedTypes [1]bool
  66. }
  67. // AgentOrErr returns the Agent value or an error if the edge
  68. // was not loaded in eager-loading, or loaded but was not found.
  69. func (e WhatsappEdges) AgentOrErr() (*Agent, error) {
  70. if e.Agent != nil {
  71. return e.Agent, nil
  72. } else if e.loadedTypes[0] {
  73. return nil, &NotFoundError{label: agent.Label}
  74. }
  75. return nil, &NotLoadedError{edge: "agent"}
  76. }
  77. // scanValues returns the types for scanning values from sql.Rows.
  78. func (*Whatsapp) scanValues(columns []string) ([]any, error) {
  79. values := make([]any, len(columns))
  80. for i := range columns {
  81. switch columns[i] {
  82. case whatsapp.FieldAllowList, whatsapp.FieldGroupAllowList, whatsapp.FieldBlockList, whatsapp.FieldGroupBlockList:
  83. values[i] = new([]byte)
  84. case whatsapp.FieldID, whatsapp.FieldStatus, whatsapp.FieldOrganizationID, whatsapp.FieldAgentID:
  85. values[i] = new(sql.NullInt64)
  86. case whatsapp.FieldAk, whatsapp.FieldSk, whatsapp.FieldCallback, whatsapp.FieldAccount, whatsapp.FieldNickname, whatsapp.FieldPhone, whatsapp.FieldAPIBase, whatsapp.FieldAPIKey:
  87. values[i] = new(sql.NullString)
  88. case whatsapp.FieldCreatedAt, whatsapp.FieldUpdatedAt, whatsapp.FieldDeletedAt:
  89. values[i] = new(sql.NullTime)
  90. default:
  91. values[i] = new(sql.UnknownType)
  92. }
  93. }
  94. return values, nil
  95. }
  96. // assignValues assigns the values that were returned from sql.Rows (after scanning)
  97. // to the Whatsapp fields.
  98. func (w *Whatsapp) assignValues(columns []string, values []any) error {
  99. if m, n := len(values), len(columns); m < n {
  100. return fmt.Errorf("mismatch number of scan values: %d != %d", m, n)
  101. }
  102. for i := range columns {
  103. switch columns[i] {
  104. case whatsapp.FieldID:
  105. value, ok := values[i].(*sql.NullInt64)
  106. if !ok {
  107. return fmt.Errorf("unexpected type %T for field id", value)
  108. }
  109. w.ID = uint64(value.Int64)
  110. case whatsapp.FieldCreatedAt:
  111. if value, ok := values[i].(*sql.NullTime); !ok {
  112. return fmt.Errorf("unexpected type %T for field created_at", values[i])
  113. } else if value.Valid {
  114. w.CreatedAt = value.Time
  115. }
  116. case whatsapp.FieldUpdatedAt:
  117. if value, ok := values[i].(*sql.NullTime); !ok {
  118. return fmt.Errorf("unexpected type %T for field updated_at", values[i])
  119. } else if value.Valid {
  120. w.UpdatedAt = value.Time
  121. }
  122. case whatsapp.FieldStatus:
  123. if value, ok := values[i].(*sql.NullInt64); !ok {
  124. return fmt.Errorf("unexpected type %T for field status", values[i])
  125. } else if value.Valid {
  126. w.Status = uint8(value.Int64)
  127. }
  128. case whatsapp.FieldDeletedAt:
  129. if value, ok := values[i].(*sql.NullTime); !ok {
  130. return fmt.Errorf("unexpected type %T for field deleted_at", values[i])
  131. } else if value.Valid {
  132. w.DeletedAt = value.Time
  133. }
  134. case whatsapp.FieldAk:
  135. if value, ok := values[i].(*sql.NullString); !ok {
  136. return fmt.Errorf("unexpected type %T for field ak", values[i])
  137. } else if value.Valid {
  138. w.Ak = value.String
  139. }
  140. case whatsapp.FieldSk:
  141. if value, ok := values[i].(*sql.NullString); !ok {
  142. return fmt.Errorf("unexpected type %T for field sk", values[i])
  143. } else if value.Valid {
  144. w.Sk = value.String
  145. }
  146. case whatsapp.FieldCallback:
  147. if value, ok := values[i].(*sql.NullString); !ok {
  148. return fmt.Errorf("unexpected type %T for field callback", values[i])
  149. } else if value.Valid {
  150. w.Callback = value.String
  151. }
  152. case whatsapp.FieldAccount:
  153. if value, ok := values[i].(*sql.NullString); !ok {
  154. return fmt.Errorf("unexpected type %T for field account", values[i])
  155. } else if value.Valid {
  156. w.Account = value.String
  157. }
  158. case whatsapp.FieldNickname:
  159. if value, ok := values[i].(*sql.NullString); !ok {
  160. return fmt.Errorf("unexpected type %T for field nickname", values[i])
  161. } else if value.Valid {
  162. w.Nickname = value.String
  163. }
  164. case whatsapp.FieldPhone:
  165. if value, ok := values[i].(*sql.NullString); !ok {
  166. return fmt.Errorf("unexpected type %T for field phone", values[i])
  167. } else if value.Valid {
  168. w.Phone = value.String
  169. }
  170. case whatsapp.FieldOrganizationID:
  171. if value, ok := values[i].(*sql.NullInt64); !ok {
  172. return fmt.Errorf("unexpected type %T for field organization_id", values[i])
  173. } else if value.Valid {
  174. w.OrganizationID = uint64(value.Int64)
  175. }
  176. case whatsapp.FieldAgentID:
  177. if value, ok := values[i].(*sql.NullInt64); !ok {
  178. return fmt.Errorf("unexpected type %T for field agent_id", values[i])
  179. } else if value.Valid {
  180. w.AgentID = uint64(value.Int64)
  181. }
  182. case whatsapp.FieldAPIBase:
  183. if value, ok := values[i].(*sql.NullString); !ok {
  184. return fmt.Errorf("unexpected type %T for field api_base", values[i])
  185. } else if value.Valid {
  186. w.APIBase = value.String
  187. }
  188. case whatsapp.FieldAPIKey:
  189. if value, ok := values[i].(*sql.NullString); !ok {
  190. return fmt.Errorf("unexpected type %T for field api_key", values[i])
  191. } else if value.Valid {
  192. w.APIKey = value.String
  193. }
  194. case whatsapp.FieldAllowList:
  195. if value, ok := values[i].(*[]byte); !ok {
  196. return fmt.Errorf("unexpected type %T for field allow_list", values[i])
  197. } else if value != nil && len(*value) > 0 {
  198. if err := json.Unmarshal(*value, &w.AllowList); err != nil {
  199. return fmt.Errorf("unmarshal field allow_list: %w", err)
  200. }
  201. }
  202. case whatsapp.FieldGroupAllowList:
  203. if value, ok := values[i].(*[]byte); !ok {
  204. return fmt.Errorf("unexpected type %T for field group_allow_list", values[i])
  205. } else if value != nil && len(*value) > 0 {
  206. if err := json.Unmarshal(*value, &w.GroupAllowList); err != nil {
  207. return fmt.Errorf("unmarshal field group_allow_list: %w", err)
  208. }
  209. }
  210. case whatsapp.FieldBlockList:
  211. if value, ok := values[i].(*[]byte); !ok {
  212. return fmt.Errorf("unexpected type %T for field block_list", values[i])
  213. } else if value != nil && len(*value) > 0 {
  214. if err := json.Unmarshal(*value, &w.BlockList); err != nil {
  215. return fmt.Errorf("unmarshal field block_list: %w", err)
  216. }
  217. }
  218. case whatsapp.FieldGroupBlockList:
  219. if value, ok := values[i].(*[]byte); !ok {
  220. return fmt.Errorf("unexpected type %T for field group_block_list", values[i])
  221. } else if value != nil && len(*value) > 0 {
  222. if err := json.Unmarshal(*value, &w.GroupBlockList); err != nil {
  223. return fmt.Errorf("unmarshal field group_block_list: %w", err)
  224. }
  225. }
  226. default:
  227. w.selectValues.Set(columns[i], values[i])
  228. }
  229. }
  230. return nil
  231. }
  232. // Value returns the ent.Value that was dynamically selected and assigned to the Whatsapp.
  233. // This includes values selected through modifiers, order, etc.
  234. func (w *Whatsapp) Value(name string) (ent.Value, error) {
  235. return w.selectValues.Get(name)
  236. }
  237. // QueryAgent queries the "agent" edge of the Whatsapp entity.
  238. func (w *Whatsapp) QueryAgent() *AgentQuery {
  239. return NewWhatsappClient(w.config).QueryAgent(w)
  240. }
  241. // Update returns a builder for updating this Whatsapp.
  242. // Note that you need to call Whatsapp.Unwrap() before calling this method if this Whatsapp
  243. // was returned from a transaction, and the transaction was committed or rolled back.
  244. func (w *Whatsapp) Update() *WhatsappUpdateOne {
  245. return NewWhatsappClient(w.config).UpdateOne(w)
  246. }
  247. // Unwrap unwraps the Whatsapp entity that was returned from a transaction after it was closed,
  248. // so that all future queries will be executed through the driver which created the transaction.
  249. func (w *Whatsapp) Unwrap() *Whatsapp {
  250. _tx, ok := w.config.driver.(*txDriver)
  251. if !ok {
  252. panic("ent: Whatsapp is not a transactional entity")
  253. }
  254. w.config.driver = _tx.drv
  255. return w
  256. }
  257. // String implements the fmt.Stringer.
  258. func (w *Whatsapp) String() string {
  259. var builder strings.Builder
  260. builder.WriteString("Whatsapp(")
  261. builder.WriteString(fmt.Sprintf("id=%v, ", w.ID))
  262. builder.WriteString("created_at=")
  263. builder.WriteString(w.CreatedAt.Format(time.ANSIC))
  264. builder.WriteString(", ")
  265. builder.WriteString("updated_at=")
  266. builder.WriteString(w.UpdatedAt.Format(time.ANSIC))
  267. builder.WriteString(", ")
  268. builder.WriteString("status=")
  269. builder.WriteString(fmt.Sprintf("%v", w.Status))
  270. builder.WriteString(", ")
  271. builder.WriteString("deleted_at=")
  272. builder.WriteString(w.DeletedAt.Format(time.ANSIC))
  273. builder.WriteString(", ")
  274. builder.WriteString("ak=")
  275. builder.WriteString(w.Ak)
  276. builder.WriteString(", ")
  277. builder.WriteString("sk=")
  278. builder.WriteString(w.Sk)
  279. builder.WriteString(", ")
  280. builder.WriteString("callback=")
  281. builder.WriteString(w.Callback)
  282. builder.WriteString(", ")
  283. builder.WriteString("account=")
  284. builder.WriteString(w.Account)
  285. builder.WriteString(", ")
  286. builder.WriteString("nickname=")
  287. builder.WriteString(w.Nickname)
  288. builder.WriteString(", ")
  289. builder.WriteString("phone=")
  290. builder.WriteString(w.Phone)
  291. builder.WriteString(", ")
  292. builder.WriteString("organization_id=")
  293. builder.WriteString(fmt.Sprintf("%v", w.OrganizationID))
  294. builder.WriteString(", ")
  295. builder.WriteString("agent_id=")
  296. builder.WriteString(fmt.Sprintf("%v", w.AgentID))
  297. builder.WriteString(", ")
  298. builder.WriteString("api_base=")
  299. builder.WriteString(w.APIBase)
  300. builder.WriteString(", ")
  301. builder.WriteString("api_key=")
  302. builder.WriteString(w.APIKey)
  303. builder.WriteString(", ")
  304. builder.WriteString("allow_list=")
  305. builder.WriteString(fmt.Sprintf("%v", w.AllowList))
  306. builder.WriteString(", ")
  307. builder.WriteString("group_allow_list=")
  308. builder.WriteString(fmt.Sprintf("%v", w.GroupAllowList))
  309. builder.WriteString(", ")
  310. builder.WriteString("block_list=")
  311. builder.WriteString(fmt.Sprintf("%v", w.BlockList))
  312. builder.WriteString(", ")
  313. builder.WriteString("group_block_list=")
  314. builder.WriteString(fmt.Sprintf("%v", w.GroupBlockList))
  315. builder.WriteByte(')')
  316. return builder.String()
  317. }
  318. // Whatsapps is a parsable slice of Whatsapp.
  319. type Whatsapps []*Whatsapp