Update infra genesis tools

Author: jpraynaud

mithril-infra/assets/tools/fast-genesis-aggregator.sh

@@ -5,7 +5,7 @@ exec_sql_aggregator() {
   sqlite3 data/preview/$1/mithril/stores/aggregator.sqlite3 "$2"
 }
 
-fix_aggregator() {
+fix_table_aggregator() {
   LAST_EPOCH=$(exec_sql_aggregator $1 "SELECT MAX(key) FROM $2;")
   LAST_VALUE=$(exec_sql_aggregator $1 "SELECT value FROM $2 WHERE key = $LAST_EPOCH;")
 
@@ -16,8 +16,13 @@ fix_aggregator() {
   exec_sql_aggregator $1 "SELECT * FROM $2 WHERE key = $CREATE_EPOCH;"
 }
 
-fix_aggregator $AGGREGATOR_NODE stake 1
-fix_aggregator $AGGREGATOR_NODE stake 2
-fix_aggregator $AGGREGATOR_NODE verification_key 1
-fix_aggregator $AGGREGATOR_NODE verification_key 2
+fix_table_aggregator $AGGREGATOR_NODE stake 1
+fix_table_aggregator $AGGREGATOR_NODE stake 2
+fix_table_aggregator $AGGREGATOR_NODE stake 3
+fix_table_aggregator $AGGREGATOR_NODE verification_key 1
+fix_table_aggregator $AGGREGATOR_NODE verification_key 2
+fix_table_aggregator $AGGREGATOR_NODE verification_key 3
+fix_table_aggregator $AGGREGATOR_NODE protocol_parameters 1
+fix_table_aggregator $AGGREGATOR_NODE protocol_parameters 2
+fix_table_aggregator $AGGREGATOR_NODE protocol_parameters 3
 

mithril-infra/assets/tools/fast-genesis-signer.sh

@@ -5,7 +5,7 @@ exec_sql_signer() {
   sqlite3 data/preview/$1/mithril/stores/signer.sqlite3 "$2"
 }
 
-fix_signer() {
+fix_table_signer() {
   LAST_EPOCH=$(exec_sql_signer $1 "SELECT MAX(key) FROM $2;")
   LAST_VALUE=$(exec_sql_signer $1 "SELECT value FROM $2 WHERE key = $LAST_EPOCH;")
 
@@ -16,8 +16,10 @@ fix_signer() {
   exec_sql_signer $1 "SELECT * FROM $2 WHERE key = $CREATE_EPOCH;"
 }
 
-fix_signer $SIGNER_NODE stake 1
-fix_signer $SIGNER_NODE stake 2
-fix_signer $SIGNER_NODE protocol_initializer 1
-fix_signer $SIGNER_NODE protocol_initializer 2
+fix_table_signer $SIGNER_NODE stake 1
+fix_table_signer $SIGNER_NODE stake 2
+fix_table_signer $SIGNER_NODE stake 3
+fix_table_signer $SIGNER_NODE protocol_initializer 1
+fix_table_signer $SIGNER_NODE protocol_initializer 2
+fix_table_signer $SIGNER_NODE protocol_initializer 3
 

mithril-infra/assets/tools/update-genesis-certificate.sh

@@ -0,0 +1,20 @@
+#!/bin/bash
+AGGREGATOR_NODE=$1
+GENESIS_KEY_NEW=$2
+GENESIS_VALUE_NEW=$3
+
+exec_sql_aggregator() {
+  sqlite3 data/preview/$1/mithril/stores/aggregator.sqlite3 "$2"
+}
+
+fix_genesis_certificate() {
+  GENESIS_KEY_NEW='"'$2'"'
+  GENESIS_VALUE_NEW=$3
+  GENESIS_KEY_NEW_HASH=$(echo -n $GENESIS_KEY_NEW | sha256sum | awk '{print $1}')
+  exec_sql_aggregator $1 "SELECT * FROM certificate WHERE json_extract(value, '$.genesis_signature') <> '';"
+  exec_sql_aggregator $1 "DELETE FROM certificate;"
+  exec_sql_aggregator $1 "INSERT OR IGNORE INTO certificate(key_hash, key, value) VALUES('$GENESIS_KEY_NEW_HASH','$GENESIS_KEY_NEW','$GENESIS_VALUE_NEW');"
+  exec_sql_aggregator $1 "SELECT * FROM certificate WHERE json_extract(value, '$.genesis_signature') <> '';"
+}
+
+fix_genesis_certificate $AGGREGATOR_NODE $GENESIS_KEY_NEW $GENESIS_VALUE_NEW