feat: Add lewd tag to collection and nft

db/migrations/1684773436602-Data.js

@@ -0,0 +1,15 @@
+module.exports = class Data1684773436602 {
+  name = 'Data1684773436602'
+
+  async up(db) {
+    await db.query(`ALTER TABLE "nft_entity" ADD "lewd" boolean NOT NULL DEFAULT false`)
+    await db.query(`ALTER TABLE "nft_entity" ALTER COLUMN "lewd" DROP DEFAULT`)
+    await db.query(`ALTER TABLE "collection_entity" ADD "lewd" boolean NOT NULL DEFAULT false`)
+    await db.query(`ALTER TABLE "collection_entity" ALTER COLUMN "lewd" DROP DEFAULT`)
+  }
+
+  async down(db) {
+    await db.query(`ALTER TABLE "nft_entity" DROP COLUMN "lewd"`)
+    await db.query(`ALTER TABLE "collection_entity" DROP COLUMN "lewd"`)
+  }
+}

schema.graphql

@@ -11,6 +11,7 @@ type CollectionEntity @entity {
   id: ID!
   image: String
   issuer: String
+  lewd: Boolean!
   max: Int!
   media: String
   meta: MetadataEntity
@@ -41,6 +42,7 @@ type NFTEntity @entity {
   instance: String
   image: String
   issuer: String
+  lewd: Boolean!
   media: String
   meta: MetadataEntity
   metadata: String

src/mappings/shared/metadata.ts

@@ -3,7 +3,7 @@ import { Optional, TokenMetadata } from '@kodadot1/metasquid/types'
 import { isEmpty } from '@kodadot1/minimark/utils'
 
 import { logger } from '@kodadot1/metasquid/logger'
-import { MetadataEntity as Metadata } from '../../model/generated'
+import { Attribute, MetadataEntity as Metadata, MetadataEntity } from '../../model/generated'
 import { fetchMetadata } from '../utils/metadata'
 import { attributeFrom, Store } from '../utils/types'
 
@@ -42,3 +42,11 @@ export async function handleMetadata(id: string, name: string, store: Store): Pr
   }
   return final
 }
+
+export const isLewd = (metadata: MetadataEntity) => {
+  return Boolean(
+    metadata.attributes?.find((item) => {
+      return item.trait === 'NSFW'
+    })
+  )
+}

src/mappings/v1/create.ts

@@ -8,7 +8,7 @@ import { unwrap } from '../utils/extract'
 import { getCreateCollection } from '../utils/getters'
 import { error, success } from '../utils/logger'
 import { Action, Collection, Context } from '../utils/types'
-import { handleMetadata } from '../shared/metadata'
+import { handleMetadata, isLewd } from '../shared/metadata'
 
 const OPERATION = Action.CREATE
 
@@ -33,6 +33,7 @@ export async function createCollection(context: Context): Promise<void> {
     final.hash = md5(collection.id)
     final.highestSale = BigInt(0)
     final.issuer = caller
+    final.lewd = false
     final.max = Number(collection.max) || 0
     final.metadata = collection.metadata
     final.name = (collection.name || '').trim()
@@ -49,6 +50,7 @@ export async function createCollection(context: Context): Promise<void> {
       final.meta = metadata
       final.image = metadata?.image
       final.media = metadata?.animationUrl
+      final.lewd = metadata ? isLewd(metadata) : false
     }
 
     await context.store.save(final)

src/mappings/v1/mint.ts

@@ -9,7 +9,7 @@ import { getCreateToken } from '../utils/getters'
 import { error, success } from '../utils/logger'
 import { Action, Context, getNftId, NFT, Optional } from '../utils/types'
 import { createEvent } from '../shared/event'
-import { handleMetadata } from '../shared/metadata'
+import { handleMetadata, isLewd } from '../shared/metadata'
 import { calculateCollectionOwnerCountAndDistribution } from '../utils/helper'
 
 const OPERATION = Action.MINT
@@ -34,6 +34,7 @@ export async function mintItem(context: Context): Promise<void> {
     final.blockNumber = BigInt(blockNumber)
     final.name = nft.name
     final.instance = nft.instance
+    final.lewd = false
     final.transferable = nft.transferable
     final.collection = collection
     final.sn = nft.sn
@@ -62,6 +63,10 @@ export async function mintItem(context: Context): Promise<void> {
       final.meta = metadata
       final.image = metadata?.image
       final.media = metadata?.animationUrl
+      if (metadata && isLewd(metadata)) {
+        final.lewd = true
+        collection.lewd = true
+      }
     }
 
     await context.store.save(final)

src/mappings/v2/create.ts

@@ -4,7 +4,7 @@ import { Optional } from '@kodadot1/metasquid/types'
 
 import md5 from 'md5'
 import { CollectionEntity } from '../../model'
-import { handleMetadata } from '../shared'
+import { handleMetadata, isLewd } from '../shared'
 import { unwrap } from '../utils/extract'
 import { error, success } from '../utils/logger'
 import { Action, Collection, Context } from '../utils/types'
@@ -33,6 +33,7 @@ export async function createCollection(context: Context): Promise<void> {
     final.hash = md5(collection.id)
     final.highestSale = BigInt(0)
     final.issuer = caller
+    final.lewd = false
     final.max = Number(collection.max) || 0
     final.metadata = collection.metadata
     final.name = (collection.name || '').trim()
@@ -49,6 +50,7 @@ export async function createCollection(context: Context): Promise<void> {
       final.meta = metadata
       final.image = metadata?.image
       final.media = metadata?.animationUrl
+      final.lewd = metadata ? isLewd(metadata) : false
       if (metadata?.name && !final.name) {
         final.name = metadata.name
       }

src/mappings/v2/mint.ts

@@ -7,7 +7,7 @@ import { isOwnerOrElseError } from '../utils/consolidator'
 
 import { CollectionEntity, NFTEntity, Property } from '../../model/generated'
 import { createEvent } from '../shared/event'
-import { handleMetadata } from '../shared/metadata'
+import { handleMetadata, isLewd } from '../shared/metadata'
 import { findRootItemById } from '../utils/entity'
 import { calculateCollectionOwnerCountAndDistribution, isDummyAddress } from '../utils/helper'
 import logger, { error, success } from '../utils/logger'
@@ -34,6 +34,7 @@ export async function mintItem(context: Context): Promise<void> {
     final.blockNumber = BigInt(blockNumber)
     final.name = nft.name
     final.instance = nft.symbol
+    final.lewd = false
     final.transferable = nft.transferable
     final.collection = collection
     final.sn = nft.sn
@@ -55,6 +56,10 @@ export async function mintItem(context: Context): Promise<void> {
       final.meta = metadata
       final.image = metadata?.image
       final.media = metadata?.animationUrl
+      if (metadata && isLewd(metadata)) {
+        final.lewd = true
+        collection.lewd = true
+      }
       if (metadata?.name && !final.name) {
         final.name = metadata.name
       }

src/model/generated/collectionEntity.model.ts

@@ -47,6 +47,9 @@ export class CollectionEntity {
     @Column_("text", {nullable: true})
     issuer!: string | undefined | null
 
+    @Column_("bool", {nullable: false})
+    lewd!: boolean
+
     @Column_("int4", {nullable: false})
     max!: number
 

src/model/generated/nftEntity.model.ts

@@ -60,6 +60,9 @@ export class NFTEntity {
     @Column_("text", {nullable: true})
     issuer!: string | undefined | null
 
+    @Column_("bool", {nullable: false})
+    lewd!: boolean
+
     @Column_("text", {nullable: true})
     media!: string | undefined | null