forked from all-in-aigc/aicover
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathorder.ts
63 lines (55 loc) · 1.92 KB
/
order.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
import { getUserOrders, updateOrderStatus } from "@/models/order";
import { Order } from "@/types/order";
import Stripe from "stripe";
import { UserCredits } from "@/types/user";
import { getUserCoversCount } from "@/models/cover";
export async function handleOrderSession(session_id: string) {
const stripe = new Stripe(process.env.STRIPE_PRIVATE_KEY || "");
try {
const session = await stripe.checkout.sessions.retrieve(session_id);
console.log("order session: ", session);
if (!session || !session.metadata || !session.metadata.order_no) {
console.log("invalid session", session_id);
throw new Error("invalid session");
}
const order_no = session.metadata.order_no;
const paied_at = new Date().toISOString();
updateOrderStatus(order_no, 2, paied_at);
console.log("update success order status: ", order_no, paied_at);
} catch (e) {
console.log("handle order session failed: ", e);
throw e;
}
}
export async function getUserCredits(user_email: string): Promise<UserCredits> {
let user_credits: UserCredits = {
one_time_credits: 1,
monthly_credits: 0,
total_credits: 1,
used_credits: 0,
left_credits: 1,
};
try {
const used_credits = await getUserCoversCount(user_email);
user_credits.used_credits = Number(used_credits);
const orders = await getUserOrders(user_email);
if (orders) {
orders.forEach((order: Order) => {
if (order.plan === "monthly") {
user_credits.monthly_credits += order.credits;
} else {
user_credits.one_time_credits += order.credits;
}
user_credits.total_credits += order.credits;
});
}
user_credits.left_credits = user_credits.total_credits - used_credits;
if (user_credits.left_credits < 0) {
user_credits.left_credits = 0;
}
return user_credits;
} catch (e) {
console.log("get user credits failed: ", e);
return user_credits;
}
}