102 lines
2.6 KiB
TypeScript
102 lines
2.6 KiB
TypeScript
import type { Prisma } from "@prisma/client";
|
|
import fetch from "node-fetch";
|
|
|
|
import { uploadAvatar } from "@calcom/lib/server/avatar";
|
|
import { resizeBase64Image } from "@calcom/lib/server/resizeBase64Image";
|
|
import prisma from "@calcom/prisma";
|
|
|
|
interface IPrefillAvatar {
|
|
email: string;
|
|
}
|
|
|
|
async function downloadImageDataFromUrl(url: string) {
|
|
try {
|
|
const response = await fetch(url);
|
|
|
|
if (!response.ok) {
|
|
console.log("Error fetching image from: ", url);
|
|
return null;
|
|
}
|
|
|
|
const imageBuffer = await response.buffer();
|
|
const base64Image = `data:image/png;base64,${imageBuffer.toString("base64")}`;
|
|
|
|
return base64Image;
|
|
} catch (error) {
|
|
console.log(error);
|
|
return null;
|
|
}
|
|
}
|
|
|
|
export const prefillAvatar = async ({ email }: IPrefillAvatar) => {
|
|
const imageUrl = await getImageUrlAvatarAPI(email);
|
|
if (!imageUrl) return;
|
|
|
|
const base64Image = await downloadImageDataFromUrl(imageUrl);
|
|
if (!base64Image) return;
|
|
|
|
const avatar = await resizeBase64Image(base64Image);
|
|
const user = await prisma.user.findFirst({
|
|
where: { email: email },
|
|
});
|
|
|
|
if (!user) {
|
|
return;
|
|
}
|
|
const avatarUrl = await uploadAvatar({ userId: user.id, avatar });
|
|
|
|
const data: Prisma.UserUpdateInput = {};
|
|
data.avatarUrl = avatarUrl;
|
|
|
|
await prisma.user.update({
|
|
where: { email: email },
|
|
data,
|
|
});
|
|
};
|
|
|
|
const getImageUrlAvatarAPI = async (email: string) => {
|
|
if (!process.env.AVATARAPI_USERNAME || !process.env.AVATARAPI_PASSWORD) {
|
|
console.info("No avatar api credentials found");
|
|
return null;
|
|
}
|
|
|
|
const controller = new AbortController();
|
|
const timeout = setTimeout(() => controller.abort(), 10_000);
|
|
|
|
try {
|
|
const response = await fetch("https://avatarapi.com/v2/api.aspx", {
|
|
method: "POST",
|
|
headers: {
|
|
"Content-Type": "text/plain",
|
|
},
|
|
body: JSON.stringify({
|
|
username: process.env.AVATARAPI_USERNAME,
|
|
password: process.env.AVATARAPI_PASSWORD,
|
|
email,
|
|
}),
|
|
signal: controller.signal,
|
|
});
|
|
|
|
clearTimeout(timeout);
|
|
|
|
const info = await response.json();
|
|
|
|
if (!info.Success) {
|
|
if (info.Error === "Not found") {
|
|
// Expected case: no avatar for this email
|
|
return null;
|
|
}
|
|
console.warn("Avatar API error:", info.Error);
|
|
return null;
|
|
}
|
|
return info.Image as string;
|
|
} catch (error: unknown) {
|
|
if (error instanceof DOMException && error.name === "AbortError") {
|
|
console.warn("Avatar API request timed out");
|
|
} else {
|
|
console.error("Avatar API request failed:", error);
|
|
}
|
|
return null;
|
|
}
|
|
};
|