Created
This commit is contained in:
scheibling
219
customer.go
Normal file
219
customer.go
Normal file
@@ -0,0 +1,219 @@
|
||||
package woogo
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
|
||||
"git.cloudyne.io/go/woogo/entity"
|
||||
validation "github.com/go-ozzo/ozzo-validation/v4"
|
||||
"github.com/go-ozzo/ozzo-validation/v4/is"
|
||||
jsoniter "github.com/json-iterator/go"
|
||||
)
|
||||
|
||||
// https://woocommerce.github.io/woocommerce-rest-api-docs/?php#customers
|
||||
|
||||
type customerService service
|
||||
|
||||
type CustomersQueryParams struct {
|
||||
queryParams
|
||||
Search string `url:"search,omitempty"`
|
||||
Exclude []int `url:"exclude,omitempty"`
|
||||
Include []int `url:"include,omitempty"`
|
||||
Email string `url:"email,omitempty"`
|
||||
Role string `url:"role,omitempty"`
|
||||
}
|
||||
|
||||
func (m CustomersQueryParams) Validate() error {
|
||||
return validation.ValidateStruct(&m,
|
||||
validation.Field(&m.OrderBy, validation.When(m.OrderBy != "", validation.In("id", "include", "name", "registered_date").Error("无效的排序字段"))),
|
||||
validation.Field(&m.Email, validation.When(m.Email != "", is.EmailFormat.Error("无效的邮箱"))),
|
||||
validation.Field(&m.Role, validation.When(m.Role != "", validation.In("all", "administrator", "editor", "author", "contributor", "subscriber", "shop_manager").Error("无效的角色"))),
|
||||
)
|
||||
}
|
||||
|
||||
// All List all customers
|
||||
func (s customerService) All(params CustomersQueryParams) (items []entity.Customer, total, totalPages int, isLastPage bool, err error) {
|
||||
if err = params.Validate(); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
params.TidyVars()
|
||||
resp, err := s.httpClient.R().SetQueryParamsFromValues(toValues(params)).Get("/customers")
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if resp.IsSuccess() {
|
||||
err = jsoniter.Unmarshal(resp.Body(), &items)
|
||||
total, totalPages, isLastPage = parseResponseTotal(params.Page, resp)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// One Retrieve a customer
|
||||
func (s customerService) One(id int) (item entity.Customer, err error) {
|
||||
resp, err := s.httpClient.R().Get(fmt.Sprintf("/customers/%d", id))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if resp.IsSuccess() {
|
||||
err = jsoniter.Unmarshal(resp.Body(), &item)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// CreateCustomerRequest Create customer request
|
||||
type CreateCustomerRequest struct {
|
||||
Email string `json:"email,omitempty"`
|
||||
FirstName string `json:"first_name,omitempty"`
|
||||
LastName string `json:"last_name,omitempty"`
|
||||
Username string `json:"username,omitempty"`
|
||||
Password string `json:"password,omitempty"`
|
||||
Billing *entity.Billing `json:"billing,omitempty"`
|
||||
Shipping *entity.Shipping `json:"shipping,omitempty"`
|
||||
MetaData []entity.Meta `json:"meta_data,omitempty"`
|
||||
}
|
||||
|
||||
func (m CreateCustomerRequest) Validate() error {
|
||||
return validation.ValidateStruct(&m,
|
||||
validation.Field(&m.Email,
|
||||
validation.Required.Error("邮箱不能为空"),
|
||||
is.EmailFormat.Error("无效的邮箱"),
|
||||
),
|
||||
validation.Field(&m.FirstName, validation.Required.Error("姓不能为空")),
|
||||
validation.Field(&m.LastName, validation.Required.Error("名不能为空")),
|
||||
validation.Field(&m.Username, validation.Required.Error("登录帐号不能为空")),
|
||||
validation.Field(&m.Password, validation.Required.Error("登录密码不能为空")),
|
||||
validation.Field(&m.Billing),
|
||||
)
|
||||
}
|
||||
|
||||
// Create create a customer
|
||||
func (s customerService) Create(req CreateCustomerRequest) (item entity.Customer, err error) {
|
||||
if err = req.Validate(); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
resp, err := s.httpClient.R().SetBody(req).Post("/customers")
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if resp.IsSuccess() {
|
||||
err = jsoniter.Unmarshal(resp.Body(), &item)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// Update customer
|
||||
|
||||
type UpdateCustomerRequest struct {
|
||||
Email string `json:"email,omitempty"`
|
||||
FirstName string `json:"first_name,omitempty"`
|
||||
LastName string `json:"last_name,omitempty"`
|
||||
Billing *entity.Billing `json:"billing,omitempty"`
|
||||
Shipping *entity.Shipping `json:"shipping,omitempty"`
|
||||
MetaData []entity.Meta `json:"meta_data,omitempty"`
|
||||
}
|
||||
|
||||
func (m UpdateCustomerRequest) Validate() error {
|
||||
return validation.ValidateStruct(&m,
|
||||
validation.Field(&m.Email, validation.When(m.Email != "", is.EmailFormat.Error("无效的邮箱"))),
|
||||
validation.Field(&m.Billing),
|
||||
)
|
||||
}
|
||||
|
||||
// Update update a customer
|
||||
func (s customerService) Update(id int, req UpdateCustomerRequest) (item entity.Customer, err error) {
|
||||
if err = req.Validate(); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
resp, err := s.httpClient.R().SetBody(req).Put(fmt.Sprintf("/customers/%d", id))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if resp.IsSuccess() {
|
||||
err = jsoniter.Unmarshal(resp.Body(), &item)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// Delete Delete a customer
|
||||
|
||||
type customerDeleteParams struct {
|
||||
Force bool `json:"force,omitempty"`
|
||||
Reassign []int `json:"reassign,omitempty"`
|
||||
}
|
||||
|
||||
func (s customerService) Delete(id int, params customerDeleteParams) (item entity.Customer, err error) {
|
||||
resp, err := s.httpClient.R().SetBody(params).Delete(fmt.Sprintf("/customers/%d", id))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if resp.IsSuccess() {
|
||||
err = jsoniter.Unmarshal(resp.Body(), &item)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// Batch update customers
|
||||
|
||||
type BatchCreateCustomerRequest = CreateCustomerRequest
|
||||
|
||||
type BatchUpdateCustomerRequest struct {
|
||||
ID string `json:"id"`
|
||||
BatchCreateCustomerRequest
|
||||
}
|
||||
|
||||
type BatchCustomerRequest struct {
|
||||
Create []BatchCreateCustomerRequest `json:"create,omitempty"`
|
||||
Update []BatchUpdateCustomerRequest `json:"update,omitempty"`
|
||||
Delete []int `json:"delete,omitempty"`
|
||||
}
|
||||
|
||||
func (m BatchCustomerRequest) Validate() error {
|
||||
if len(m.Create) == 0 && len(m.Update) == 0 && len(m.Delete) == 0 {
|
||||
return errors.New("无效的请求数据")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
type BatchCustomerResult struct {
|
||||
Create []entity.Customer `json:"create"`
|
||||
Update []entity.Customer `json:"update"`
|
||||
Delete []entity.Customer `json:"delete"`
|
||||
}
|
||||
|
||||
// Batch Batch create/update/delete customers
|
||||
func (s customerService) Batch(req BatchCustomerRequest) (res BatchCustomerResult, err error) {
|
||||
if err = req.Validate(); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
resp, err := s.httpClient.R().SetBody(req).Post("/customers/batch")
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if resp.IsSuccess() {
|
||||
err = jsoniter.Unmarshal(resp.Body(), &res)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// Downloads retrieve a customer downloads
|
||||
func (s customerService) Downloads(customerId int) (items []entity.CustomerDownload, err error) {
|
||||
resp, err := s.httpClient.R().Get(fmt.Sprintf("/customers/%d/downloads", customerId))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if resp.IsSuccess() {
|
||||
err = jsoniter.Unmarshal(resp.Body(), &items)
|
||||
}
|
||||
return
|
||||
}
|
||||
Reference in New Issue
Block a user