💩 Use class metadata image as nft image fallback

likecoin · Feb 10, 2023 · e8a0bfe · e8a0bfe
1 parent ffcc490
commit e8a0bfe
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 9 deletions.
diff --git a/src/routes/likernft/metadata.ts b/src/routes/likernft/metadata.ts
@@ -92,10 +92,15 @@ router.get(
       }
       const size = Math.min(Math.max(inputSizeNum, 1), 1920);
       const iscnPrefix = await getISCNPrefixByClassId(classId);
-      const { data } = await getNFTISCNData(iscnPrefix);
-      if (!data) throw new ValidationError('ISCN_NOT_FOUND', 404);
-      const iscnId = data && data['@id'] as string;
+      const [{ data: ISCNData }, chainData] = await Promise.all([
+        getNFTISCNData(iscnPrefix),
+        getNFTClassDataById(classId),
+      ]);
+      if (!chainData) throw new ValidationError('CLASS_ID_NOT_FOUND', 404);
+      if (!ISCNData) throw new ValidationError('ISCN_NOT_FOUND', 404);
+      const iscnId = ISCNData && ISCNData['@id'] as string;
       if (!iscnId) throw new ValidationError('ISCN_ID_NOT_FOUND', 404);
+      const { image: chainImage } = chainData.data.metadata;
       let iscnData = await iscnInfoCollection.doc(encodeURIComponent(iscnId)).get();
       if (!iscnData.exists) {
         await axios.post(
@@ -105,16 +110,16 @@ router.get(
         await sleep(1000);
         iscnData = await iscnInfoCollection.doc(encodeURIComponent(iscnId)).get();
       }
-      let image = '';
+      let iscnImage = '';
       let title = 'Writing NFT';
       if (iscnData.exists) {
-        ({ image, title = 'Writing NFT' } = iscnData.data());
+        ({ image: iscnImage, title = 'Writing NFT' } = iscnData.data());
       }
       const {
         image: basicImage,
         contentType,
         isDefault: isImageMissing,
-      } = await getBasicImage(image, title);
+      } = await getBasicImage(iscnImage, chainImage, title);
       const resizedImage = getResizedImage(size);
       // Disable image mask for now
       // const combinedImage = await getCombinedImage();

diff --git a/src/util/api/likernft/metadata.ts b/src/util/api/likernft/metadata.ts
@@ -50,12 +50,20 @@ export function parseImageURLFromMetadata(image: string): string {
   return image.replace('ar://', 'https://arweave.net/').replace('ipfs://', 'https://ipfs.io/ipfs/')
 }
 
-export async function getBasicImage(image, title) {
+export async function getBasicImage(iscnImage, chainImage, title) {
   let imageBuffer;
   let contentType;
   let isDefault = true;
-  if (image) {
-    const imageData = (await axios.get(encodedURL(image), { responseType: 'stream' }).catch(() => ({} as any)));
+  if (iscnImage) {
+    const imageData = (await axios.get(encodedURL(iscnImage), { responseType: 'stream' }).catch(() => ({} as any)));
+    if (imageData && imageData.data) {
+      imageBuffer = imageData.data;
+      contentType = imageData.headers['content-type'] || 'image/png';
+      isDefault = false;
+    }
+  }
+  if (chainImage && !imageBuffer) {
+    const imageData = (await axios.get(encodedURL(chainImage), { responseType: 'stream' }).catch(() => ({} as any)));
     if (imageData && imageData.data) {
       imageBuffer = imageData.data;
       contentType = imageData.headers['content-type'] || 'image/png';