131 lines
3.9 KiB
TypeScript
131 lines
3.9 KiB
TypeScript
import {NextApiRequest, NextApiResponse} from "next";
|
|
import {createUserWithEmailAndPassword, getAuth} from "firebase/auth";
|
|
import {app} from "@/firebase";
|
|
import {sessionOptions} from "@/lib/session";
|
|
import {withIronSessionApiRoute} from "iron-session/next";
|
|
import {getFirestore, doc, setDoc, query, collection, where, getDocs} from "firebase/firestore";
|
|
import {CorporateInformation, DemographicInformation, Type} from "@/interfaces/user";
|
|
import {addUserToGroupOnCreation} from "@/utils/registration";
|
|
import moment from "moment";
|
|
|
|
const auth = getAuth(app);
|
|
const db = getFirestore(app);
|
|
|
|
export default withIronSessionApiRoute(register, sessionOptions);
|
|
|
|
const DEFAULT_DESIRED_LEVELS = {
|
|
reading: 9,
|
|
listening: 9,
|
|
writing: 9,
|
|
speaking: 9,
|
|
};
|
|
|
|
const DEFAULT_LEVELS = {
|
|
reading: 0,
|
|
listening: 0,
|
|
writing: 0,
|
|
speaking: 0,
|
|
};
|
|
|
|
async function register(req: NextApiRequest, res: NextApiResponse) {
|
|
const {type} = req.body as {
|
|
type: "individual" | "corporate";
|
|
};
|
|
|
|
if (type === "individual") return registerIndividual(req, res);
|
|
if (type === "corporate") return registerCorporate(req, res);
|
|
}
|
|
|
|
async function registerIndividual(req: NextApiRequest, res: NextApiResponse) {
|
|
const {email, passport_id, password, code} = req.body as {
|
|
email: string;
|
|
passport_id?: string;
|
|
password: string;
|
|
code?: string;
|
|
};
|
|
|
|
const codeQuery = query(collection(db, "codes"), where("code", "==", code));
|
|
const codeDocs = (await getDocs(codeQuery)).docs.filter((x) => !Object.keys(x.data()).includes("userId"));
|
|
|
|
if (code && code.length > 0 && codeDocs.length === 0) {
|
|
res.status(400).json({error: "Invalid Code!"});
|
|
return;
|
|
}
|
|
|
|
const codeData = codeDocs.length > 0 ? (codeDocs[0].data() as {code: string; type: Type; creator?: string; expiryDate: Date | null}) : undefined;
|
|
|
|
createUserWithEmailAndPassword(auth, email, password)
|
|
.then(async (userCredentials) => {
|
|
const userId = userCredentials.user.uid;
|
|
delete req.body.password;
|
|
|
|
const user = {
|
|
...req.body,
|
|
desiredLevels: DEFAULT_DESIRED_LEVELS,
|
|
levels: DEFAULT_LEVELS,
|
|
bio: "",
|
|
isFirstLogin: codeData ? codeData.type === "student" : true,
|
|
focus: "academic",
|
|
type: email.endsWith("@ecrop.dev") ? "developer" : codeData ? codeData.type : "student",
|
|
subscriptionExpirationDate: codeData ? codeData.expiryDate : moment().subtract(1, "days").toISOString(),
|
|
...(passport_id ? {demographicInformation: {passport_id}} : {}),
|
|
registrationDate: new Date().toISOString(),
|
|
status: code ? "active" : "paymentDue",
|
|
};
|
|
|
|
await setDoc(doc(db, "users", userId), user);
|
|
|
|
if (codeDocs.length > 0 && codeData) {
|
|
await setDoc(codeDocs[0].ref, {userId: userId}, {merge: true});
|
|
if (codeData.creator) await addUserToGroupOnCreation(userId, codeData.type, codeData.creator);
|
|
}
|
|
|
|
req.session.user = {...user, id: userId};
|
|
await req.session.save();
|
|
|
|
res.status(200).json({user: {...user, id: userId}});
|
|
})
|
|
.catch((error) => {
|
|
console.log(error);
|
|
res.status(401).json({error});
|
|
});
|
|
}
|
|
|
|
async function registerCorporate(req: NextApiRequest, res: NextApiResponse) {
|
|
const {email, password} = req.body as {
|
|
email: string;
|
|
password: string;
|
|
corporateInformation: CorporateInformation;
|
|
};
|
|
|
|
createUserWithEmailAndPassword(auth, email, password)
|
|
.then(async (userCredentials) => {
|
|
const userId = userCredentials.user.uid;
|
|
delete req.body.password;
|
|
|
|
const user = {
|
|
...req.body,
|
|
desiredLevels: DEFAULT_DESIRED_LEVELS,
|
|
levels: DEFAULT_LEVELS,
|
|
bio: "",
|
|
isFirstLogin: false,
|
|
focus: "academic",
|
|
type: "corporate",
|
|
subscriptionExpirationDate: req.body.subscriptionExpirationDate || null,
|
|
status: "paymentDue",
|
|
registrationDate: new Date().toISOString(),
|
|
};
|
|
|
|
await setDoc(doc(db, "users", userId), user);
|
|
|
|
req.session.user = {...user, id: userId};
|
|
await req.session.save();
|
|
|
|
res.status(200).json({user: {...user, id: userId}});
|
|
})
|
|
.catch((error) => {
|
|
console.log(error);
|
|
res.status(401).json({error});
|
|
});
|
|
}
|