feat: 🎸 Country Code Mapping Table Functionality

package.json

@@ -54,6 +54,7 @@
   },
   "dependencies": {
     "fastify": "^4.26.2",
+    "i18n-iso-countries": "^7.11.0",
     "ts-node": "^10.9.2"
   }
 }

src/cmd.ts

@@ -196,6 +196,7 @@ function datasetCommand(yargs: yargs.Argv<{}>): {
                 },
             }
         )
+        .command("getAllCountriesCallingCode", "Get all countries calling code")
         .command("isWinner", "The account is the dataset auditor winner", {
             datasetId: {
                 description: "Dataset Id",
@@ -553,6 +554,15 @@ async function dataset(
                 path: String(argv.path),
             })
             break
+        case "getDatasetState":
+            await new DatasetMetadatas().getDatasetState({
+                context,
+                datasetId: Number(argv.datasetId),
+            })
+            break
+        case "getAllCountriesCallingCode":
+            await new DatasetMetadatas().getAllCountriesCallingCode()
+            break
         case "submitDatasetProof":
             await new DatasetProofs().submitDatasetProof({
                 context,
@@ -601,12 +611,6 @@ async function dataset(
                 path: String(argv.path),
             })
             break
-        case "getDatasetState":
-            await new DatasetMetadatas().getDatasetState({
-                context,
-                datasetId: Number(argv.datasetId),
-            })
-            break
         default:
             console.log("Unknown command.")
     }

src/dataset/metadata/repo/index.ts

@@ -19,6 +19,7 @@
  ********************************************************************************/
 
 import fs from "fs"
+import * as countries from "i18n-iso-countries"
 import { DatasetState } from "@dataswapjs/dataswapjs"
 import { handleEvmError, logMethodCall } from "../../../shared/utils/utils"
 import { chainSuccessInterval, blockPeriod } from "../../../shared/constant"
@@ -192,14 +193,13 @@ export class DatasetMetadatas {
             fs.readFileSync(options.path).toString()
         ) as DatasetReplicaRequirements
 
-        const state = await handleEvmError(
-            options.context.evm.datasetMetadata.getDatasetState(
-                datasetReplicaRequirements.datasetId
-            )
-        )
-        if (state != DatasetState.MetadataSubmitted) {
-            console.log("Dataset state is not MetadataSubmitted, do nothing~")
-            return true
+        if (
+            !(await this.checkSubmissionRequirementsCriteria({
+                context: options.context,
+                datasetReplicaRequirements,
+            }))
+        ) {
+            return false
         }
 
         options.context.evm.datasetRequirement
@@ -211,8 +211,8 @@ export class DatasetMetadatas {
                 datasetReplicaRequirements.dataPreparers,
                 datasetReplicaRequirements.storageProviders,
                 datasetReplicaRequirements.regions,
-                datasetReplicaRequirements.countrys,
-                datasetReplicaRequirements.citys,
+                datasetReplicaRequirements.countries,
+                datasetReplicaRequirements.cities,
                 datasetReplicaRequirements.amount
             )
         )
@@ -236,4 +236,37 @@ export class DatasetMetadatas {
             )
         )
     }
+
+    @logMethodCall(["context"])
+    async getAllCountriesCallingCode(): Promise<{
+        [numericKey: string]: string
+    }> {
+        return countries.getNumericCodes()
+    }
+
+    private async checkSubmissionRequirementsCriteria(options: {
+        context: Context
+        datasetReplicaRequirements: DatasetReplicaRequirements
+    }): Promise<boolean> {
+        const state = await handleEvmError(
+            options.context.evm.datasetMetadata.getDatasetState(
+                options.datasetReplicaRequirements.datasetId
+            )
+        )
+        if (state != DatasetState.MetadataSubmitted) {
+            console.log("Dataset state is not MetadataSubmitted, do nothing~")
+            return false
+        }
+        for (const country of options.datasetReplicaRequirements.countries) {
+            if (!countries.isValid(Number(country))) {
+                console.log(
+                    "Dataset replic requirements countries invalid:",
+                    country
+                )
+                return false
+            }
+        }
+
+        return true
+    }
 }

src/dataset/metadata/types/index.ts

@@ -46,7 +46,7 @@ export interface DatasetReplicaRequirements {
     dataPreparers: string[][]
     storageProviders: string[][]
     regions: bigint[]
-    countrys: bigint[]
-    citys: bigint[][]
+    countries: bigint[]
+    cities: bigint[][]
     amount: bigint
 }

testdata/datasetReplicaRequirements.json

@@ -60,14 +60,14 @@
         3474,
         4211
     ],
-    "countrys": [
-        517,
-        1300,
-        2040,
-        3063,
-        4687
+    "countries": [
+        840,
+        528,
+        508,
+        504,
+        533
     ],
-    "citys": [
+    "cities": [
         [
             202186,
             1302254,
@@ -95,5 +95,5 @@
         ]
     ],
     "amount": 0,
-    "datasetId": 16
+    "datasetId": 17
 }