Create getReferralQrCode() method

aldrin-labs · bathord · Apr 29, 2024 · Apr 29, 2024 · Apr 29, 2024 · Apr 29, 2024
commit 82031f00bdfbffe1bce35b6d14ce481b90bb33ae
diff --git a/src/chains/referral/utils.ts b/src/chains/referral/utils.ts
@@ -1,6 +1,9 @@
+import { QRCodeCanvas } from '@loskir/styled-qr-code-node';
+import { InputFile } from 'grammy';
 import { BotContext, MyConversation } from '../../types';
-import { RINBOT_URL } from '../sui.config';
-import { PARAMS_SEPARATOR, REF_PARAM_KEY } from './config';
+import { RINBOT_LOGO_URL, RINBOT_URL } from '../sui.config';
+import { imageUrlToBase64 } from '../utils';
+import { PARAMS_SEPARATOR, REFERRAL_QR_CODE_CONFIG, REF_PARAM_KEY } from './config';
 import { getReferrerPublicKey } from './redis/utils';
 
 export function generateReferralId(): string {
@@ -11,6 +14,16 @@ export function getReferralLink(referralId: string): string {
   return `${RINBOT_URL}?start=${REF_PARAM_KEY}=${referralId}`;
 }
 
+export async function getReferralQrCode(referralLink: string): Promise<InputFile> {
+  const rinbotLogoBase64 = await imageUrlToBase64(RINBOT_LOGO_URL);
+  const qrCode = new QRCodeCanvas({ ...REFERRAL_QR_CODE_CONFIG, data: referralLink, image: rinbotLogoBase64 });
+
+  const buffer = await qrCode.toBuffer();
+  const file = new InputFile(buffer);
+
+  return file;
+}
+
 export function getReferrerIdByParams(params: string) {
   const separatorIndex = params.indexOf(PARAMS_SEPARATOR);