Save identity known heights from people chain, #960

opensquare-network · Sep 27, 2024 · 4b9046e · 4b9046e
1 parent a2cf1df
commit 4b9046e
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 14 deletions.
diff --git a/backend/packages/identity-scan/src/scripts/known/identity.js b/backend/packages/identity-scan/src/scripts/known/identity.js
@@ -1,13 +1,11 @@
 const {
   identity: { getIdentityTimelineCol },
 } = require("@statescan/mongo");
-const {
-  utils: { saveHeightsCommon },
-} = require("@statescan/common");
+const { saveHeights } = require("./save");
 
 async function saveIdentityHeights() {
   const timelineCol = await getIdentityTimelineCol();
-  await saveHeightsCommon(timelineCol);
+  await saveHeights(timelineCol);
 }
 
 module.exports = {

diff --git a/backend/packages/identity-scan/src/scripts/known/index.js b/backend/packages/identity-scan/src/scripts/known/index.js
@@ -1,17 +1,29 @@
 require("dotenv").config();
 
-const {
-  utils: { saveHeightsCommon },
-} = require("@statescan/common");
 const { saveIdentityHeights } = require("./identity");
 const { saveRequestHeights } = require("./request");
 const {
   identity: { getRegistrarsTimelineCol },
 } = require("@statescan/mongo");
+const {
+  env: { currentChain },
+  mongo: {
+    known: { saveKnownHeights, getKnownHeightDb },
+  },
+} = require("@osn/scan-common");
+const { saveHeights } = require("./save");
+
+const knownHeights = Object.freeze({
+  polkadot: [22572436],
+  kusama: [23780225],
+});
 
 (async () => {
   await saveIdentityHeights();
   await saveRequestHeights();
-  await saveHeightsCommon(await getRegistrarsTimelineCol());
+  await saveHeights(await getRegistrarsTimelineCol());
+
+  const known = knownHeights[currentChain()];
+  await saveKnownHeights(known);
   process.exit(0);
 })();
diff --git a/backend/packages/identity-scan/src/scripts/known/request.js b/backend/packages/identity-scan/src/scripts/known/request.js
@@ -1,13 +1,11 @@
 const {
   identity: { getRequestCol, getRequestTimelineCol },
 } = require("@statescan/mongo");
-const {
-  utils: { saveHeightsCommon },
-} = require("@statescan/common");
+const { saveHeights } = require("./save");
 
 async function saveRequestHeights() {
-  await saveHeightsCommon(await getRequestTimelineCol());
-  await saveHeightsCommon(await getRequestCol());
+  await saveHeights(await getRequestTimelineCol());
+  await saveHeights(await getRequestCol());
 }
 
 module.exports = {

diff --git a/backend/packages/identity-scan/src/scripts/known/save.js b/backend/packages/identity-scan/src/scripts/known/save.js
@@ -22,7 +22,7 @@ async function savePeopleChainHeights(heights = []) {
   await bulk.execute();
 }
 
-async function saveHeightsCommon(col) {
+async function saveHeights(col) {
   const items = await col.find({}).toArray();
   const heights = [];
   const peopleChainHeights = [];
@@ -57,3 +57,7 @@ async function saveHeightsCommon(col) {
   const uniquePeopleChainHeights = [...new Set(peopleChainHeights)];
   await savePeopleChainHeights(uniquePeopleChainHeights);
 }
+
+module.exports = {
+  saveHeights,
+};