HFO4
6 years ago
parent
09de05548f
commit
45b54b3455
@ -1,42 +0,0 @@
|
|||||||
package payment
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
model "github.com/HFO4/cloudreve/models"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/serializer"
|
|
||||||
alipay "github.com/smartwalle/alipay/v3"
|
|
||||||
"net/url"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Alipay 支付宝当面付支付处理
|
|
||||||
type Alipay struct {
|
|
||||||
Client *alipay.Client
|
|
||||||
}
|
|
||||||
|
|
||||||
// Create 创建订单
|
|
||||||
func (pay *Alipay) Create(order *model.Order, pack *serializer.PackProduct, group *serializer.GroupProducts, user *model.User) (*OrderCreateRes, error) {
|
|
||||||
gateway, _ := url.Parse("/api/v3/callback/alipay")
|
|
||||||
var p = alipay.TradePreCreate{
|
|
||||||
Trade: alipay.Trade{
|
|
||||||
NotifyURL: model.GetSiteURL().ResolveReference(gateway).String(),
|
|
||||||
Subject: order.Name,
|
|
||||||
OutTradeNo: order.OrderNo,
|
|
||||||
TotalAmount: fmt.Sprintf("%.2f", float64(order.Price*order.Num)/100),
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, err := order.Create(); err != nil {
|
|
||||||
return nil, ErrInsertOrder.WithError(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
res, err := pay.Client.TradePreCreate(p)
|
|
||||||
if err != nil {
|
|
||||||
return nil, ErrIssueOrder.WithError(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return &OrderCreateRes{
|
|
||||||
Payment: true,
|
|
||||||
QRCode: res.Content.QRCode,
|
|
||||||
ID: order.OrderNo,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
@ -1,147 +0,0 @@
|
|||||||
package payment
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
model "github.com/HFO4/cloudreve/models"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/serializer"
|
|
||||||
"github.com/qingwg/payjs"
|
|
||||||
"github.com/smartwalle/alipay/v3"
|
|
||||||
"math/rand"
|
|
||||||
"net/url"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
// ErrUnknownPaymentMethod 未知支付方式
|
|
||||||
ErrUnknownPaymentMethod = serializer.NewError(serializer.CodeNotFound, "未知支付方式", nil)
|
|
||||||
// ErrUnsupportedPaymentMethod 未知支付方式
|
|
||||||
ErrUnsupportedPaymentMethod = serializer.NewError(serializer.CodeNotFound, "此订单不支持此支付方式", nil)
|
|
||||||
// ErrInsertOrder 无法插入订单记录
|
|
||||||
ErrInsertOrder = serializer.NewError(serializer.CodeDBError, "无法插入订单记录", nil)
|
|
||||||
// ErrScoreNotEnough 积分不足
|
|
||||||
ErrScoreNotEnough = serializer.NewError(serializer.CodeNoPermissionErr, "积分不足", nil)
|
|
||||||
// ErrCreateStoragePack 无法创建容量包
|
|
||||||
ErrCreateStoragePack = serializer.NewError(serializer.CodeNoPermissionErr, "无法创建容量包", nil)
|
|
||||||
// ErrGroupConflict 用户组冲突
|
|
||||||
ErrGroupConflict = serializer.NewError(serializer.CodeNoPermissionErr, "当前用户组仍未过期，请前往个人设置手动解约后继续", nil)
|
|
||||||
// ErrGroupInvalid 用户组冲突
|
|
||||||
ErrGroupInvalid = serializer.NewError(serializer.CodeNoPermissionErr, "用户组不可用", nil)
|
|
||||||
// ErrUpgradeGroup 用户组冲突
|
|
||||||
ErrUpgradeGroup = serializer.NewError(serializer.CodeDBError, "无法升级用户组", nil)
|
|
||||||
// ErrUInitPayment 无法初始化支付实例
|
|
||||||
ErrUInitPayment = serializer.NewError(serializer.CodeInternalSetting, "无法初始化支付实例", nil)
|
|
||||||
// ErrIssueOrder 订单接口请求失败
|
|
||||||
ErrIssueOrder = serializer.NewError(serializer.CodeInternalSetting, "无法创建订单", nil)
|
|
||||||
// ErrOrderNotFound 订单不存在
|
|
||||||
ErrOrderNotFound = serializer.NewError(serializer.CodeNotFound, "订单不存在", nil)
|
|
||||||
)
|
|
||||||
|
|
||||||
// Pay 支付处理接口
|
|
||||||
type Pay interface {
|
|
||||||
Create(order *model.Order, pack *serializer.PackProduct, group *serializer.GroupProducts, user *model.User) (*OrderCreateRes, error)
|
|
||||||
}
|
|
||||||
|
|
||||||
// OrderCreateRes 订单创建结果
|
|
||||||
type OrderCreateRes struct {
|
|
||||||
Payment bool `json:"payment"` // 是否需要支付
|
|
||||||
ID string `json:"id,omitempty"` // 订单号
|
|
||||||
QRCode string `json:"qr_code,omitempty"` // 支付二维码指向的地址
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewPaymentInstance 获取新的支付实例
|
|
||||||
func NewPaymentInstance(method string) (Pay, error) {
|
|
||||||
switch method {
|
|
||||||
case "score":
|
|
||||||
return &ScorePayment{}, nil
|
|
||||||
case "alipay":
|
|
||||||
options := model.GetSettingByNames("alipay_enabled", "appid", "appkey", "shopid")
|
|
||||||
if options["alipay_enabled"] != "1" {
|
|
||||||
return nil, ErrUnknownPaymentMethod
|
|
||||||
}
|
|
||||||
|
|
||||||
// 初始化支付宝客户端
|
|
||||||
var client, err = alipay.New(options["appid"], options["appkey"], true)
|
|
||||||
if err != nil {
|
|
||||||
return nil, ErrUInitPayment.WithError(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 加载支付宝公钥
|
|
||||||
err = client.LoadAliPayPublicKey(options["shopid"])
|
|
||||||
if err != nil {
|
|
||||||
return nil, ErrUInitPayment.WithError(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return &Alipay{Client: client}, nil
|
|
||||||
case "payjs":
|
|
||||||
options := model.GetSettingByNames("payjs_enabled", "payjs_secret", "payjs_id")
|
|
||||||
if options["payjs_enabled"] != "1" {
|
|
||||||
return nil, ErrUnknownPaymentMethod
|
|
||||||
}
|
|
||||||
|
|
||||||
callback, _ := url.Parse("/api/v3/callback/payjs")
|
|
||||||
payjsConfig := &payjs.Config{
|
|
||||||
Key: options["payjs_secret"],
|
|
||||||
MchID: options["payjs_id"],
|
|
||||||
NotifyUrl: model.GetSiteURL().ResolveReference(callback).String(),
|
|
||||||
}
|
|
||||||
|
|
||||||
return &PayJSClient{Client: payjs.New(payjsConfig)}, nil
|
|
||||||
|
|
||||||
default:
|
|
||||||
return nil, ErrUnknownPaymentMethod
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewOrder 创建新订单
|
|
||||||
func NewOrder(pack *serializer.PackProduct, group *serializer.GroupProducts, num int, method string, user *model.User) (*OrderCreateRes, error) {
|
|
||||||
// 获取支付实例
|
|
||||||
pay, err := NewPaymentInstance(method)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
var (
|
|
||||||
orderType int
|
|
||||||
productID int64
|
|
||||||
title string
|
|
||||||
price int
|
|
||||||
)
|
|
||||||
if pack != nil {
|
|
||||||
orderType = model.PackOrderType
|
|
||||||
productID = pack.ID
|
|
||||||
title = pack.Name
|
|
||||||
price = pack.Price
|
|
||||||
} else if group != nil {
|
|
||||||
orderType = model.GroupOrderType
|
|
||||||
productID = group.ID
|
|
||||||
title = group.Name
|
|
||||||
price = group.Price
|
|
||||||
} else {
|
|
||||||
orderType = model.ScoreOrderType
|
|
||||||
productID = 0
|
|
||||||
title = fmt.Sprintf("%d 积分", num)
|
|
||||||
price = model.GetIntSetting("score_price", 1)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 创建订单记录
|
|
||||||
order := &model.Order{
|
|
||||||
UserID: user.ID,
|
|
||||||
OrderNo: orderID(),
|
|
||||||
Type: orderType,
|
|
||||||
Method: method,
|
|
||||||
ProductID: productID,
|
|
||||||
Num: num,
|
|
||||||
Name: fmt.Sprintf("%s - %s", model.GetSettingByName("siteName"), title),
|
|
||||||
Price: price,
|
|
||||||
Status: model.OrderUnpaid,
|
|
||||||
}
|
|
||||||
|
|
||||||
return pay.Create(order, pack, group, user)
|
|
||||||
}
|
|
||||||
|
|
||||||
func orderID() string {
|
|
||||||
return fmt.Sprintf("%s%d",
|
|
||||||
time.Now().Format("20060102150405"),
|
|
||||||
100000+rand.Intn(900000),
|
|
||||||
)
|
|
||||||
}
|
|
||||||
@ -1,31 +0,0 @@
|
|||||||
package payment
|
|
||||||
|
|
||||||
import (
|
|
||||||
model "github.com/HFO4/cloudreve/models"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/serializer"
|
|
||||||
"github.com/qingwg/payjs"
|
|
||||||
)
|
|
||||||
|
|
||||||
// PayJSClient PayJS支付处理
|
|
||||||
type PayJSClient struct {
|
|
||||||
Client *payjs.PayJS
|
|
||||||
}
|
|
||||||
|
|
||||||
// Create 创建订单
|
|
||||||
func (pay *PayJSClient) Create(order *model.Order, pack *serializer.PackProduct, group *serializer.GroupProducts, user *model.User) (*OrderCreateRes, error) {
|
|
||||||
if _, err := order.Create(); err != nil {
|
|
||||||
return nil, ErrInsertOrder.WithError(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
PayNative := pay.Client.GetNative()
|
|
||||||
res, err := PayNative.Create(int64(order.Price*order.Num), order.Name, order.OrderNo, "", "")
|
|
||||||
if err != nil {
|
|
||||||
return nil, ErrIssueOrder.WithError(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return &OrderCreateRes{
|
|
||||||
Payment: true,
|
|
||||||
QRCode: res.CodeUrl,
|
|
||||||
ID: order.OrderNo,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
@ -1,127 +0,0 @@
|
|||||||
package payment
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"errors"
|
|
||||||
model "github.com/HFO4/cloudreve/models"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/serializer"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
// GivePack 创建容量包
|
|
||||||
func GivePack(user *model.User, packInfo *serializer.PackProduct, num int) error {
|
|
||||||
timeNow := time.Now()
|
|
||||||
expires := timeNow.Add(time.Duration(packInfo.Time*int64(num)) * time.Second)
|
|
||||||
pack := model.StoragePack{
|
|
||||||
Name: packInfo.Name,
|
|
||||||
UserID: user.ID,
|
|
||||||
ActiveTime: &timeNow,
|
|
||||||
ExpiredTime: &expires,
|
|
||||||
Size: packInfo.Size,
|
|
||||||
}
|
|
||||||
if _, err := pack.Create(); err != nil {
|
|
||||||
return ErrCreateStoragePack.WithError(err)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func checkGroupUpgrade(user *model.User, groupInfo *serializer.GroupProducts) error {
|
|
||||||
// 检查用户是否已有未过期用户
|
|
||||||
if user.PreviousGroupID != 0 {
|
|
||||||
return ErrGroupConflict
|
|
||||||
}
|
|
||||||
|
|
||||||
// 用户组不能相同
|
|
||||||
if user.GroupID == groupInfo.GroupID {
|
|
||||||
return ErrGroupInvalid
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// GiveGroup 升级用户组
|
|
||||||
func GiveGroup(user *model.User, groupInfo *serializer.GroupProducts, num int) error {
|
|
||||||
if err := checkGroupUpgrade(user, groupInfo); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
timeNow := time.Now()
|
|
||||||
expires := timeNow.Add(time.Duration(groupInfo.Time*int64(num)) * time.Second)
|
|
||||||
|
|
||||||
if err := user.UpgradeGroup(groupInfo.GroupID, &expires); err != nil {
|
|
||||||
return ErrUpgradeGroup.WithError(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// GiveScore 积分充值
|
|
||||||
func GiveScore(user *model.User, num int) error {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// GiveProduct “发货”
|
|
||||||
func GiveProduct(user *model.User, pack *serializer.PackProduct, group *serializer.GroupProducts, num int) error {
|
|
||||||
if pack != nil {
|
|
||||||
return GivePack(user, pack, num)
|
|
||||||
} else if group != nil {
|
|
||||||
return GiveGroup(user, group, num)
|
|
||||||
} else {
|
|
||||||
return GiveScore(user, num)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// OrderPaid 订单已支付处理
|
|
||||||
func OrderPaid(orderNo string) error {
|
|
||||||
order, err := model.GetOrderByNo(orderNo)
|
|
||||||
if err != nil {
|
|
||||||
return ErrOrderNotFound.WithError(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 更新订单状态为 已支付
|
|
||||||
order.UpdateStatus(model.OrderPaid)
|
|
||||||
|
|
||||||
user, err := model.GetActiveUserByID(order.UserID)
|
|
||||||
if err != nil {
|
|
||||||
return errors.New("用户不存在")
|
|
||||||
}
|
|
||||||
|
|
||||||
// 查询商品
|
|
||||||
options := model.GetSettingByNames("pack_data", "group_sell_data")
|
|
||||||
|
|
||||||
var (
|
|
||||||
packs []serializer.PackProduct
|
|
||||||
groups []serializer.GroupProducts
|
|
||||||
)
|
|
||||||
if err := json.Unmarshal([]byte(options["pack_data"]), &packs); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if err := json.Unmarshal([]byte(options["group_sell_data"]), &groups); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
// 查找要购买的商品
|
|
||||||
var (
|
|
||||||
pack *serializer.PackProduct
|
|
||||||
group *serializer.GroupProducts
|
|
||||||
)
|
|
||||||
if order.Type == model.GroupOrderType {
|
|
||||||
for _, v := range groups {
|
|
||||||
if v.ID == order.ProductID {
|
|
||||||
group = &v
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else if order.Type == model.PackOrderType {
|
|
||||||
for _, v := range packs {
|
|
||||||
if v.ID == order.ProductID {
|
|
||||||
pack = &v
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// "发货"
|
|
||||||
return GiveProduct(&user, pack, group, order.Num)
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -1,34 +0,0 @@
|
|||||||
package payment
|
|
||||||
|
|
||||||
import (
|
|
||||||
model "github.com/HFO4/cloudreve/models"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/serializer"
|
|
||||||
)
|
|
||||||
|
|
||||||
// ScorePayment 积分支付处理
|
|
||||||
type ScorePayment struct {
|
|
||||||
}
|
|
||||||
|
|
||||||
// Create 创建新订单
|
|
||||||
func (pay *ScorePayment) Create(order *model.Order, pack *serializer.PackProduct, group *serializer.GroupProducts, user *model.User) (*OrderCreateRes, error) {
|
|
||||||
if pack != nil {
|
|
||||||
order.Price = pack.Score
|
|
||||||
} else {
|
|
||||||
order.Price = group.Score
|
|
||||||
}
|
|
||||||
|
|
||||||
// 检查此订单是否可用积分支付
|
|
||||||
if order.Price == 0 {
|
|
||||||
return nil, ErrUnsupportedPaymentMethod
|
|
||||||
}
|
|
||||||
|
|
||||||
// 创建订单记录
|
|
||||||
order.Status = model.OrderPaid
|
|
||||||
if _, err := order.Create(); err != nil {
|
|
||||||
return nil, ErrInsertOrder.WithError(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return &OrderCreateRes{
|
|
||||||
Payment: false,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
@ -1,211 +0,0 @@
|
|||||||
package qq
|
|
||||||
|
|
||||||
import (
|
|
||||||
"crypto/md5"
|
|
||||||
"encoding/json"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
model "github.com/HFO4/cloudreve/models"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/request"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/serializer"
|
|
||||||
"github.com/gofrs/uuid"
|
|
||||||
"net/url"
|
|
||||||
"strings"
|
|
||||||
)
|
|
||||||
|
|
||||||
// LoginPage 登陆页面描述
|
|
||||||
type LoginPage struct {
|
|
||||||
URL string
|
|
||||||
SecretKey string
|
|
||||||
}
|
|
||||||
|
|
||||||
// UserCredentials 登陆成功后的凭证
|
|
||||||
type UserCredentials struct {
|
|
||||||
OpenID string
|
|
||||||
AccessToken string
|
|
||||||
}
|
|
||||||
|
|
||||||
// UserInfo 用户信息
|
|
||||||
type UserInfo struct {
|
|
||||||
Nick string
|
|
||||||
Avatar string
|
|
||||||
}
|
|
||||||
|
|
||||||
var (
|
|
||||||
// ErrNotEnabled 未开启登录功能
|
|
||||||
ErrNotEnabled = serializer.NewError(serializer.CodeNoPermissionErr, "QQ登录功能未开启", nil)
|
|
||||||
// ErrObtainAccessToken 无法获取AccessToken
|
|
||||||
ErrObtainAccessToken = serializer.NewError(serializer.CodeParamErr, "无法获取AccessToken", nil)
|
|
||||||
// ErrObtainOpenID 无法获取OpenID
|
|
||||||
ErrObtainOpenID = serializer.NewError(serializer.CodeParamErr, "无法获取OpenID", nil)
|
|
||||||
//ErrDecodeResponse 无法解析服务端响应
|
|
||||||
ErrDecodeResponse = serializer.NewError(serializer.CodeInternalSetting, "无法解析服务端响应", nil)
|
|
||||||
)
|
|
||||||
|
|
||||||
// NewLoginRequest 新建登录会话
|
|
||||||
func NewLoginRequest() (*LoginPage, error) {
|
|
||||||
// 获取相关设定
|
|
||||||
options := model.GetSettingByNames("qq_login", "qq_login_id")
|
|
||||||
if options["qq_login"] == "0" {
|
|
||||||
return nil, ErrNotEnabled
|
|
||||||
}
|
|
||||||
|
|
||||||
// 生成唯一ID
|
|
||||||
u2, err := uuid.NewV4()
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
secret := fmt.Sprintf("%x", md5.Sum(u2.Bytes()))
|
|
||||||
|
|
||||||
// 生成登录地址
|
|
||||||
loginURL, _ := url.Parse("https://graph.qq.com/oauth2.0/authorize?response_type=code")
|
|
||||||
queries := loginURL.Query()
|
|
||||||
queries.Add("client_id", options["qq_login_id"])
|
|
||||||
queries.Add("redirect_uri", getCallbackURL())
|
|
||||||
queries.Add("state", secret)
|
|
||||||
loginURL.RawQuery = queries.Encode()
|
|
||||||
|
|
||||||
return &LoginPage{
|
|
||||||
URL: loginURL.String(),
|
|
||||||
SecretKey: secret,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func getCallbackURL() string {
|
|
||||||
//return "https://drive.aoaoao.me/Callback/QQ"
|
|
||||||
// 生成回调地址
|
|
||||||
gateway, _ := url.Parse("/#/login/qq")
|
|
||||||
callback := model.GetSiteURL().ResolveReference(gateway).String()
|
|
||||||
|
|
||||||
return callback
|
|
||||||
}
|
|
||||||
|
|
||||||
func getAccessTokenURL(code string) string {
|
|
||||||
// 获取相关设定
|
|
||||||
options := model.GetSettingByNames("qq_login_id", "qq_login_key")
|
|
||||||
|
|
||||||
api, _ := url.Parse("https://graph.qq.com/oauth2.0/token?grant_type=authorization_code")
|
|
||||||
queries := api.Query()
|
|
||||||
queries.Add("client_id", options["qq_login_id"])
|
|
||||||
queries.Add("redirect_uri", getCallbackURL())
|
|
||||||
queries.Add("client_secret", options["qq_login_key"])
|
|
||||||
queries.Add("code", code)
|
|
||||||
api.RawQuery = queries.Encode()
|
|
||||||
|
|
||||||
return api.String()
|
|
||||||
}
|
|
||||||
|
|
||||||
func getUserInfoURL(openid, ak string) string {
|
|
||||||
// 获取相关设定
|
|
||||||
options := model.GetSettingByNames("qq_login_id", "qq_login_key")
|
|
||||||
|
|
||||||
api, _ := url.Parse("https://graph.qq.com/user/get_user_info")
|
|
||||||
queries := api.Query()
|
|
||||||
queries.Add("oauth_consumer_key", options["qq_login_id"])
|
|
||||||
queries.Add("openid", openid)
|
|
||||||
queries.Add("access_token", ak)
|
|
||||||
api.RawQuery = queries.Encode()
|
|
||||||
|
|
||||||
return api.String()
|
|
||||||
}
|
|
||||||
|
|
||||||
func getResponse(body string) (map[string]interface{}, error) {
|
|
||||||
var res map[string]interface{}
|
|
||||||
|
|
||||||
if !strings.Contains(body, "callback") {
|
|
||||||
return res, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
body = strings.TrimPrefix(body, "callback(")
|
|
||||||
body = strings.TrimSuffix(body, ");\n")
|
|
||||||
|
|
||||||
err := json.Unmarshal([]byte(body), &res)
|
|
||||||
|
|
||||||
return res, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Callback 处理回调,返回openid和access key
|
|
||||||
func Callback(code string) (*UserCredentials, error) {
|
|
||||||
// 获取相关设定
|
|
||||||
options := model.GetSettingByNames("qq_login")
|
|
||||||
if options["qq_login"] == "0" {
|
|
||||||
return nil, ErrNotEnabled
|
|
||||||
}
|
|
||||||
|
|
||||||
api := getAccessTokenURL(code)
|
|
||||||
|
|
||||||
// 获取AccessToken
|
|
||||||
client := request.HTTPClient{}
|
|
||||||
res := client.Request("GET", api, nil)
|
|
||||||
resp, err := res.GetResponse()
|
|
||||||
if err != nil {
|
|
||||||
return nil, ErrObtainAccessToken.WithError(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 如果服务端返回错误
|
|
||||||
errResp, err := getResponse(resp)
|
|
||||||
if msg, ok := errResp["error_description"]; err == nil && ok {
|
|
||||||
return nil, ErrObtainAccessToken.WithError(errors.New(msg.(string)))
|
|
||||||
}
|
|
||||||
|
|
||||||
// 获取AccessToken
|
|
||||||
vals, err := url.ParseQuery(resp)
|
|
||||||
if err != nil {
|
|
||||||
return nil, ErrDecodeResponse.WithError(err)
|
|
||||||
}
|
|
||||||
accessToken := vals.Get("access_token")
|
|
||||||
|
|
||||||
// 用 AccessToken 换取OpenID
|
|
||||||
res = client.Request("GET", "https://graph.qq.com/oauth2.0/me?access_token="+accessToken, nil)
|
|
||||||
resp, err = res.GetResponse()
|
|
||||||
if err != nil {
|
|
||||||
return nil, ErrObtainOpenID.WithError(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 解析服务端响应
|
|
||||||
errResp, err = getResponse(resp)
|
|
||||||
if msg, ok := errResp["error_description"]; err == nil && ok {
|
|
||||||
return nil, ErrObtainOpenID.WithError(errors.New(msg.(string)))
|
|
||||||
}
|
|
||||||
|
|
||||||
if openid, ok := errResp["openid"]; ok {
|
|
||||||
return &UserCredentials{
|
|
||||||
OpenID: openid.(string),
|
|
||||||
AccessToken: accessToken,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil, ErrDecodeResponse
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetUserInfo 使用凭证获取用户信息
|
|
||||||
func GetUserInfo(credential *UserCredentials) (*UserInfo, error) {
|
|
||||||
api := getUserInfoURL(credential.OpenID, credential.AccessToken)
|
|
||||||
|
|
||||||
// 获取用户信息
|
|
||||||
client := request.HTTPClient{}
|
|
||||||
res := client.Request("GET", api, nil)
|
|
||||||
resp, err := res.GetResponse()
|
|
||||||
if err != nil {
|
|
||||||
return nil, ErrObtainAccessToken.WithError(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var resSerialized map[string]interface{}
|
|
||||||
if err := json.Unmarshal([]byte(resp), &resSerialized); err != nil {
|
|
||||||
return nil, ErrDecodeResponse.WithError(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 如果服务端返回错误
|
|
||||||
if msg, ok := resSerialized["msg"]; ok && msg.(string) != "" {
|
|
||||||
return nil, ErrObtainAccessToken.WithError(errors.New(msg.(string)))
|
|
||||||
}
|
|
||||||
|
|
||||||
if avatar, ok := resSerialized["figureurl_qq_2"]; ok {
|
|
||||||
return &UserInfo{
|
|
||||||
Nick: resSerialized["nickname"].(string),
|
|
||||||
Avatar: avatar.(string),
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil, ErrDecodeResponse
|
|
||||||
}
|
|
||||||
@ -1,120 +0,0 @@
|
|||||||
package serializer
|
|
||||||
|
|
||||||
import (
|
|
||||||
model "github.com/HFO4/cloudreve/models"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/hashid"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/util"
|
|
||||||
)
|
|
||||||
|
|
||||||
type quota struct {
|
|
||||||
Base uint64 `json:"base"`
|
|
||||||
Pack uint64 `json:"pack"`
|
|
||||||
Used uint64 `json:"used"`
|
|
||||||
Total uint64 `json:"total"`
|
|
||||||
Packs []storagePacks `json:"packs"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type storagePacks struct {
|
|
||||||
Name string `json:"name"`
|
|
||||||
Size uint64 `json:"size"`
|
|
||||||
ActivateDate string `json:"activate_date"`
|
|
||||||
Expiration int `json:"expiration"`
|
|
||||||
ExpirationDate string `json:"expiration_date"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// MountedFolders 已挂载的目录
|
|
||||||
type MountedFolders struct {
|
|
||||||
ID string `json:"id"`
|
|
||||||
Name string `json:"name"`
|
|
||||||
PolicyName string `json:"policy_name"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type policyOptions struct {
|
|
||||||
Name string `json:"name"`
|
|
||||||
ID string `json:"id"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// BuildPolicySettingRes 构建存储策略选项选择
|
|
||||||
func BuildPolicySettingRes(policies []model.Policy, current *model.Policy) Response {
|
|
||||||
options := make([]policyOptions, 0, len(policies))
|
|
||||||
for _, policy := range policies {
|
|
||||||
options = append(options, policyOptions{
|
|
||||||
Name: policy.Name,
|
|
||||||
ID: hashid.HashID(policy.ID, hashid.PolicyID),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
return Response{
|
|
||||||
Data: map[string]interface{}{
|
|
||||||
"options": options,
|
|
||||||
"current": policyOptions{
|
|
||||||
Name: current.Name,
|
|
||||||
ID: hashid.HashID(current.ID, hashid.PolicyID),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// BuildUserQuotaResponse 序列化用户存储配额概况响应
|
|
||||||
func BuildUserQuotaResponse(user *model.User, packs []model.StoragePack) Response {
|
|
||||||
packSize := user.GetAvailablePackSize()
|
|
||||||
res := quota{
|
|
||||||
Base: user.Group.MaxStorage,
|
|
||||||
Pack: packSize,
|
|
||||||
Used: user.Storage,
|
|
||||||
Total: packSize + user.Group.MaxStorage,
|
|
||||||
Packs: make([]storagePacks, 0, len(packs)),
|
|
||||||
}
|
|
||||||
for _, pack := range packs {
|
|
||||||
res.Packs = append(res.Packs, storagePacks{
|
|
||||||
Name: pack.Name,
|
|
||||||
Size: pack.Size,
|
|
||||||
ActivateDate: pack.ActiveTime.Format("2006-01-02 15:04:05"),
|
|
||||||
Expiration: int(pack.ExpiredTime.Sub(*pack.ActiveTime).Seconds()),
|
|
||||||
ExpirationDate: pack.ExpiredTime.Format("2006-01-02 15:04:05"),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
return Response{
|
|
||||||
Data: res,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// PackProduct 容量包商品
|
|
||||||
type PackProduct struct {
|
|
||||||
ID int64 `json:"id"`
|
|
||||||
Name string `json:"name"`
|
|
||||||
Size uint64 `json:"size"`
|
|
||||||
Time int64 `json:"time"`
|
|
||||||
Price int `json:"price"`
|
|
||||||
Score int `json:"score"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// GroupProducts 用户组商品
|
|
||||||
type GroupProducts struct {
|
|
||||||
ID int64 `json:"id"`
|
|
||||||
Name string `json:"name"`
|
|
||||||
GroupID uint `json:"group_id"`
|
|
||||||
Time int64 `json:"time"`
|
|
||||||
Price int `json:"price"`
|
|
||||||
Score int `json:"score"`
|
|
||||||
Des []string `json:"des"`
|
|
||||||
Highlight bool `json:"highlight"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// BuildProductResponse 构建增值服务商品响应
|
|
||||||
func BuildProductResponse(groups []GroupProducts, packs []PackProduct, alipay, payjs bool, scorePrice int) Response {
|
|
||||||
// 隐藏响应中的用户组ID
|
|
||||||
for i := 0; i < len(groups); i++ {
|
|
||||||
groups[i].GroupID = 0
|
|
||||||
}
|
|
||||||
return Response{
|
|
||||||
Data: map[string]interface{}{
|
|
||||||
"packs": packs,
|
|
||||||
"groups": groups,
|
|
||||||
"alipay": alipay,
|
|
||||||
"payjs": payjs,
|
|
||||||
"score_price": scorePrice,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,144 +0,0 @@
|
|||||||
package controllers
|
|
||||||
|
|
||||||
import (
|
|
||||||
"github.com/HFO4/cloudreve/pkg/payment"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/util"
|
|
||||||
"github.com/HFO4/cloudreve/service/vas"
|
|
||||||
"github.com/gin-gonic/gin"
|
|
||||||
"github.com/qingwg/payjs/notify"
|
|
||||||
"github.com/smartwalle/alipay/v3"
|
|
||||||
)
|
|
||||||
|
|
||||||
// GetQuota 获取容量配额信息
|
|
||||||
func GetQuota(c *gin.Context) {
|
|
||||||
var service vas.GeneralVASService
|
|
||||||
if err := c.ShouldBindUri(&service); err == nil {
|
|
||||||
res := service.Quota(c, CurrentUser(c))
|
|
||||||
c.JSON(200, res)
|
|
||||||
} else {
|
|
||||||
c.JSON(200, ErrorResponse(err))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetProduct 获取商品信息
|
|
||||||
func GetProduct(c *gin.Context) {
|
|
||||||
var service vas.GeneralVASService
|
|
||||||
if err := c.ShouldBindUri(&service); err == nil {
|
|
||||||
res := service.Products(c, CurrentUser(c))
|
|
||||||
c.JSON(200, res)
|
|
||||||
} else {
|
|
||||||
c.JSON(200, ErrorResponse(err))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewOrder 新建支付订单
|
|
||||||
func NewOrder(c *gin.Context) {
|
|
||||||
var service vas.CreateOrderService
|
|
||||||
if err := c.ShouldBindJSON(&service); err == nil {
|
|
||||||
res := service.Create(c, CurrentUser(c))
|
|
||||||
c.JSON(200, res)
|
|
||||||
} else {
|
|
||||||
c.JSON(200, ErrorResponse(err))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// OrderStatus 查询订单状态
|
|
||||||
func OrderStatus(c *gin.Context) {
|
|
||||||
var service vas.OrderService
|
|
||||||
if err := c.ShouldBindUri(&service); err == nil {
|
|
||||||
res := service.Status(c, CurrentUser(c))
|
|
||||||
c.JSON(200, res)
|
|
||||||
} else {
|
|
||||||
c.JSON(200, ErrorResponse(err))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetRedeemInfo 获取兑换码信息
|
|
||||||
func GetRedeemInfo(c *gin.Context) {
|
|
||||||
var service vas.RedeemService
|
|
||||||
if err := c.ShouldBindUri(&service); err == nil {
|
|
||||||
res := service.Query(c)
|
|
||||||
c.JSON(200, res)
|
|
||||||
} else {
|
|
||||||
c.JSON(200, ErrorResponse(err))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// DoRedeem 获取兑换码信息
|
|
||||||
func DoRedeem(c *gin.Context) {
|
|
||||||
var service vas.RedeemService
|
|
||||||
if err := c.ShouldBindUri(&service); err == nil {
|
|
||||||
res := service.Redeem(c, CurrentUser(c))
|
|
||||||
c.JSON(200, res)
|
|
||||||
} else {
|
|
||||||
c.JSON(200, ErrorResponse(err))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// AlipayCallback 支付宝回调
|
|
||||||
func AlipayCallback(c *gin.Context) {
|
|
||||||
pay, err := payment.NewPaymentInstance("alipay")
|
|
||||||
if err != nil {
|
|
||||||
util.Log().Debug("[支付宝回调] 无法创建支付宝客户端, %s", err)
|
|
||||||
c.Status(400)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
res, err := pay.(*payment.Alipay).Client.GetTradeNotification(c.Request)
|
|
||||||
if err != nil {
|
|
||||||
util.Log().Debug("[支付宝回调] 回调验证失败, %s", err)
|
|
||||||
c.Status(403)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if res != nil && res.TradeStatus == "TRADE_SUCCESS" {
|
|
||||||
// 支付成功
|
|
||||||
if err := payment.OrderPaid(res.OutTradeNo); err != nil {
|
|
||||||
util.Log().Debug("[支付宝回调] 支付处理失败, %s", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 确认收到通知消息
|
|
||||||
alipay.AckNotification(c.Writer)
|
|
||||||
}
|
|
||||||
|
|
||||||
// PayJSCallback PayJS回调
|
|
||||||
func PayJSCallback(c *gin.Context) {
|
|
||||||
pay, err := payment.NewPaymentInstance("payjs")
|
|
||||||
if err != nil {
|
|
||||||
util.Log().Debug("[PayJS回调] 无法创建支付宝客户端, %s", err)
|
|
||||||
c.Status(400)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
payNotify := pay.(*payment.PayJSClient).Client.GetNotify(c.Request, c.Writer)
|
|
||||||
|
|
||||||
//设置接收消息的处理方法
|
|
||||||
payNotify.SetMessageHandler(func(msg notify.Message) {
|
|
||||||
if err := payment.OrderPaid(msg.OutTradeNo); err != nil {
|
|
||||||
util.Log().Debug("[PayJS回调] 支付处理失败, %s", err)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
//处理消息接收以及回复
|
|
||||||
err = payNotify.Serve()
|
|
||||||
if err != nil {
|
|
||||||
util.Log().Debug("[PayJS回调] 回调处理失败, %s", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
//发送回复的消息
|
|
||||||
payNotify.SendResponseMsg()
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// QQCallback QQ互联回调
|
|
||||||
func QQCallback(c *gin.Context) {
|
|
||||||
var service vas.QQCallbackService
|
|
||||||
if err := c.ShouldBindJSON(&service); err == nil {
|
|
||||||
res := service.Callback(c, CurrentUser(c))
|
|
||||||
c.JSON(200, res)
|
|
||||||
} else {
|
|
||||||
c.JSON(200, ErrorResponse(err))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,98 +0,0 @@
|
|||||||
package admin
|
|
||||||
|
|
||||||
import (
|
|
||||||
model "github.com/HFO4/cloudreve/models"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/serializer"
|
|
||||||
"github.com/gofrs/uuid"
|
|
||||||
)
|
|
||||||
|
|
||||||
// GenerateRedeemsService 兑换码生成服务
|
|
||||||
type GenerateRedeemsService struct {
|
|
||||||
Num int `json:"num" binding:"required,min=1,max=100"`
|
|
||||||
ID int64 `json:"id"`
|
|
||||||
Time int `json:"time" binding:"required,min=1"`
|
|
||||||
Type int `json:"type" binding:"min=0,max=2"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// SingleIDService 单ID服务
|
|
||||||
type SingleIDService struct {
|
|
||||||
ID uint `uri:"id" binding:"required"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// DeleteRedeem 删除兑换码
|
|
||||||
func (service *SingleIDService) DeleteRedeem() serializer.Response {
|
|
||||||
if err := model.DB.Where("id = ?", service.ID).Delete(&model.Redeem{}).Error; err != nil {
|
|
||||||
return serializer.DBErr("无法删除兑换码", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return serializer.Response{}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Generate 生成兑换码
|
|
||||||
func (service *GenerateRedeemsService) Generate() serializer.Response {
|
|
||||||
res := make([]string, service.Num)
|
|
||||||
redeem := model.Redeem{}
|
|
||||||
|
|
||||||
// 开始事务
|
|
||||||
tx := model.DB.Begin()
|
|
||||||
if err := tx.Error; err != nil {
|
|
||||||
return serializer.DBErr("无法开启事务", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 创建每个兑换码
|
|
||||||
for i := 0; i < service.Num; i++ {
|
|
||||||
redeem.Model.ID = 0
|
|
||||||
redeem.Num = service.Time
|
|
||||||
redeem.Type = service.Type
|
|
||||||
redeem.ProductID = service.ID
|
|
||||||
redeem.Used = false
|
|
||||||
|
|
||||||
// 生成唯一兑换码
|
|
||||||
u2, err := uuid.NewV4()
|
|
||||||
if err != nil {
|
|
||||||
tx.Rollback()
|
|
||||||
return serializer.Err(serializer.CodeInternalSetting, "无法生成兑换码", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
redeem.Code = u2.String()
|
|
||||||
if err := tx.Create(&redeem).Error; err != nil {
|
|
||||||
tx.Rollback()
|
|
||||||
return serializer.DBErr("无法创建兑换码记录", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
res[i] = redeem.Code
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := tx.Commit().Error; err != nil {
|
|
||||||
return serializer.DBErr("无法创建兑换码记录", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return serializer.Response{Data: res}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// Redeems 列出激活码
|
|
||||||
func (service *AdminListService) Redeems() serializer.Response {
|
|
||||||
var res []model.Redeem
|
|
||||||
total := 0
|
|
||||||
|
|
||||||
tx := model.DB.Model(&model.Redeem{})
|
|
||||||
if service.OrderBy != "" {
|
|
||||||
tx = tx.Order(service.OrderBy)
|
|
||||||
}
|
|
||||||
|
|
||||||
for k, v := range service.Conditions {
|
|
||||||
tx = tx.Where("? = ?", k, v)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 计算总数用于分页
|
|
||||||
tx.Count(&total)
|
|
||||||
|
|
||||||
// 查询记录
|
|
||||||
tx.Limit(service.PageSize).Offset((service.Page - 1) * service.PageSize).Find(&res)
|
|
||||||
|
|
||||||
return serializer.Response{Data: map[string]interface{}{
|
|
||||||
"total": total,
|
|
||||||
"items": res,
|
|
||||||
}}
|
|
||||||
}
|
|
||||||
@ -1,219 +0,0 @@
|
|||||||
package vas
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
model "github.com/HFO4/cloudreve/models"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/payment"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/serializer"
|
|
||||||
"github.com/gin-gonic/gin"
|
|
||||||
)
|
|
||||||
|
|
||||||
// CreateOrderService 创建订单服务
|
|
||||||
type CreateOrderService struct {
|
|
||||||
Action string `json:"action" binding:"required,eq=group|eq=pack|eq=score"`
|
|
||||||
Method string `json:"method" binding:"required,eq=alipay|eq=score|eq=payjs"`
|
|
||||||
ID int64 `json:"id" binding:"required"`
|
|
||||||
Num int `json:"num" binding:"required,min=1"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// RedeemService 兑换服务
|
|
||||||
type RedeemService struct {
|
|
||||||
Code string `uri:"code" binding:"required,max=64"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// OrderService 订单查询
|
|
||||||
type OrderService struct {
|
|
||||||
ID string `uri:"id" binding:"required"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// Status 查询订单状态
|
|
||||||
func (service *OrderService) Status(c *gin.Context, user *model.User) serializer.Response {
|
|
||||||
order, _ := model.GetOrderByNo(service.ID)
|
|
||||||
if order == nil || order.UserID != user.ID {
|
|
||||||
return serializer.Err(serializer.CodeNotFound, "订单不存在", nil)
|
|
||||||
}
|
|
||||||
|
|
||||||
return serializer.Response{Data: order.Status}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Redeem 开始兑换
|
|
||||||
func (service *RedeemService) Redeem(c *gin.Context, user *model.User) serializer.Response {
|
|
||||||
redeem, err := model.GetAvailableRedeem(service.Code)
|
|
||||||
if err != nil {
|
|
||||||
return serializer.Err(serializer.CodeDBError, "兑换码无效", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 取得当前商品信息
|
|
||||||
packs, groups, err := decodeProductInfo()
|
|
||||||
if err != nil {
|
|
||||||
return serializer.Err(serializer.CodeInternalSetting, "无法解析商品设置", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 查找要购买的商品
|
|
||||||
var (
|
|
||||||
pack *serializer.PackProduct
|
|
||||||
group *serializer.GroupProducts
|
|
||||||
)
|
|
||||||
if redeem.Type == model.GroupOrderType {
|
|
||||||
for _, v := range groups {
|
|
||||||
if v.ID == redeem.ProductID {
|
|
||||||
group = &v
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if group == nil {
|
|
||||||
return serializer.Err(serializer.CodeNotFound, "商品已失效", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
} else if redeem.Type == model.PackOrderType {
|
|
||||||
for _, v := range packs {
|
|
||||||
if v.ID == redeem.ProductID {
|
|
||||||
pack = &v
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if pack == nil {
|
|
||||||
return serializer.Err(serializer.CodeNotFound, "商品已失效", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
err = payment.GiveProduct(user, pack, group, redeem.Num)
|
|
||||||
if err != nil {
|
|
||||||
return serializer.Err(serializer.CodeNotSet, "兑换失败", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
redeem.Use()
|
|
||||||
|
|
||||||
return serializer.Response{}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// Query 检查兑换码信息
|
|
||||||
func (service *RedeemService) Query(c *gin.Context) serializer.Response {
|
|
||||||
redeem, err := model.GetAvailableRedeem(service.Code)
|
|
||||||
if err != nil {
|
|
||||||
return serializer.Err(serializer.CodeDBError, "兑换码无效", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var (
|
|
||||||
name = "积分"
|
|
||||||
productTime int64
|
|
||||||
)
|
|
||||||
if redeem.Type != model.ScoreOrderType {
|
|
||||||
packs, groups, err := decodeProductInfo()
|
|
||||||
if err != nil {
|
|
||||||
return serializer.Err(serializer.CodeInternalSetting, "无法解析商品设置", err)
|
|
||||||
}
|
|
||||||
if redeem.Type == model.GroupOrderType {
|
|
||||||
for _, v := range groups {
|
|
||||||
if v.ID == redeem.ProductID {
|
|
||||||
name = v.Name
|
|
||||||
productTime = v.Time
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
for _, v := range packs {
|
|
||||||
if v.ID == redeem.ProductID {
|
|
||||||
name = v.Name
|
|
||||||
productTime = v.Time
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if name == "积分" {
|
|
||||||
return serializer.Err(serializer.CodeNotFound, "商品已失效", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
return serializer.Response{
|
|
||||||
Data: struct {
|
|
||||||
Name string `json:"name"`
|
|
||||||
Type int `json:"type"`
|
|
||||||
Num int `json:"num"`
|
|
||||||
Time int64 `json:"time"`
|
|
||||||
}{
|
|
||||||
name, redeem.Type, redeem.Num, productTime,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Create 创建新订单
|
|
||||||
func (service *CreateOrderService) Create(c *gin.Context, user *model.User) serializer.Response {
|
|
||||||
// 取得当前商品信息
|
|
||||||
packs, groups, err := decodeProductInfo()
|
|
||||||
if err != nil {
|
|
||||||
return serializer.Err(serializer.CodeInternalSetting, "无法解析商品设置", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 查找要购买的商品
|
|
||||||
var (
|
|
||||||
pack *serializer.PackProduct
|
|
||||||
group *serializer.GroupProducts
|
|
||||||
)
|
|
||||||
if service.Action == "group" {
|
|
||||||
for _, v := range groups {
|
|
||||||
if v.ID == service.ID {
|
|
||||||
group = &v
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else if service.Action == "pack" {
|
|
||||||
for _, v := range packs {
|
|
||||||
if v.ID == service.ID {
|
|
||||||
pack = &v
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 购买积分
|
|
||||||
if pack == nil && group == nil {
|
|
||||||
if service.Method == "score" {
|
|
||||||
return serializer.Err(serializer.CodeNotFound, "不支持此支付方式", nil)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 创建订单
|
|
||||||
res, err := payment.NewOrder(pack, group, service.Num, service.Method, user)
|
|
||||||
if err != nil {
|
|
||||||
return serializer.Err(serializer.CodeNotSet, err.Error(), err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return serializer.Response{Data: res}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// Products 获取商品信息
|
|
||||||
func (service *GeneralVASService) Products(c *gin.Context, user *model.User) serializer.Response {
|
|
||||||
options := model.GetSettingByNames("alipay_enabled", "payjs_enabled", "score_price")
|
|
||||||
scorePrice := model.GetIntSetting("score_price", 0)
|
|
||||||
packs, groups, err := decodeProductInfo()
|
|
||||||
if err != nil {
|
|
||||||
return serializer.Err(serializer.CodeInternalSetting, "无法解析商品设置", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return serializer.BuildProductResponse(groups, packs, options["alipay_enabled"] == "1", options["payjs_enabled"] == "1", scorePrice)
|
|
||||||
}
|
|
||||||
|
|
||||||
func decodeProductInfo() ([]serializer.PackProduct, []serializer.GroupProducts, error) {
|
|
||||||
options := model.GetSettingByNames("pack_data", "group_sell_data", "alipay_enabled", "payjs_enabled")
|
|
||||||
|
|
||||||
var (
|
|
||||||
packs []serializer.PackProduct
|
|
||||||
groups []serializer.GroupProducts
|
|
||||||
)
|
|
||||||
if err := json.Unmarshal([]byte(options["pack_data"]), &packs); err != nil {
|
|
||||||
return nil, nil, err
|
|
||||||
}
|
|
||||||
if err := json.Unmarshal([]byte(options["group_sell_data"]), &groups); err != nil {
|
|
||||||
return nil, nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return packs, groups, nil
|
|
||||||
}
|
|
||||||
@ -1,17 +0,0 @@
|
|||||||
package vas
|
|
||||||
|
|
||||||
import (
|
|
||||||
model "github.com/HFO4/cloudreve/models"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/serializer"
|
|
||||||
"github.com/gin-gonic/gin"
|
|
||||||
)
|
|
||||||
|
|
||||||
// GeneralVASService 通用增值服务
|
|
||||||
type GeneralVASService struct {
|
|
||||||
}
|
|
||||||
|
|
||||||
// Quota 获取容量配额信息
|
|
||||||
func (service *GeneralVASService) Quota(c *gin.Context, user *model.User) serializer.Response {
|
|
||||||
packs := user.GetAvailableStoragePacks()
|
|
||||||
return serializer.BuildUserQuotaResponse(user, packs)
|
|
||||||
}
|
|
||||||
Loading…
Reference in new issue