小程序分销

miniprogram/src/api/index.ts

@@ -4,8 +4,14 @@ import { request } from '../utils/request'
 export const getConfigs = () => request({ url: '/configs/' })
 export const getConfigDetail = (id: number) => request({ url: `/configs/${id}/` })
 
+const getInviteCode = () => Taro.getStorageSync('invite_code') || ''
+
 // Orders
-export const createOrder = (data: any) => request({ url: '/orders/', method: 'POST', data })
+export const createOrder = (data: any) => {
+    const code = getInviteCode()
+    if (code) data.ref_code = code
+    return request({ url: '/orders/', method: 'POST', data })
+}
 export const getOrder = (id: number) => request({ url: `/orders/${id}/` })
 export const getMyOrders = () => request({ url: '/orders/' }) 
 export const prepayMiniprogram = (orderId: number) => request({ url: `/orders/${orderId}/prepay_miniprogram/`, method: 'POST' })
@@ -14,20 +20,28 @@ export const queryOrderStatus = (orderId: number) => request({ url: `/orders/${o
 // AI Services
 export const getServices = () => request({ url: '/services/' })
 export const getServiceDetail = (id: number) => request({ url: `/services/${id}/` })
-export const createServiceOrder = (data: any) => request({ url: '/service-orders/', method: 'POST', data })
+export const createServiceOrder = (data: any) => {
+    const code = getInviteCode()
+    if (code) data.ref_code = code
+    return request({ url: '/service-orders/', method: 'POST', data })
+}
 
 // VB Courses
 export const getVBCourses = () => request({ url: '/courses/' })
 export const getVBCourseDetail = (id: number) => request({ url: `/courses/${id}/` })
 
 // Distributor
-export const distributorRegister = (data: any) => request({ url: '/distributor/register/', method: 'POST', data })
+export const distributorRegister = (data: any) => {
+    const code = getInviteCode()
+    if (code && !data.invite_code) data.invite_code = code
+    return request({ url: '/distributor/register/', method: 'POST', data })
+}
 export const distributorInfo = () => request({ url: '/distributor/info/' })
 export const distributorInvite = () => request({ url: '/distributor/invite/', method: 'POST' })
 export const distributorWithdraw = (amount: number) => request({ url: '/distributor/withdraw/', method: 'POST', data: { amount } })
-// TODO: Verify if these exist in the API docs
-// export const distributorTeam = () => request({ url: '/distributor/team/' })
-// export const distributorHistory = () => request({ url: '/distributor/history/' })
+export const distributorTeam = () => request({ url: '/distributor/team/' })
+export const distributorEarnings = (params?: any) => request({ url: '/distributor/earnings/', data: params })
+export const distributorOrders = (params?: any) => request({ url: '/distributor/orders/', data: params })
 
 // User
 export const updateUserInfo = (data: any) => request({ url: '/wechat/update/', method: 'POST', data })