Add infra tools

- public key generator from google credentials file
- fast genesis script for Aggregator/Signer

Author: jpraynaud

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

@@ -0,0 +1,23 @@
+#!/bin/bash
+AGGREGATOR_NODE=$1
+
+exec_sql_aggregator() {
+  sqlite3 data/preview/$1/mithril/stores/aggregator.sqlite3 "$2"
+}
+
+fix_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;")
+
+  CREATE_EPOCH=$LAST_EPOCH
+  CREATE_EPOCH=$(expr $CREATE_EPOCH - $3)
+  CREATE_EPOCH_HASH=$(echo -n $CREATE_EPOCH | sha256sum | awk '{print $1}')
+  exec_sql_aggregator $1 "INSERT OR IGNORE INTO $2(key_hash, key, value) VALUES('$CREATE_EPOCH_HASH','$CREATE_EPOCH','$LAST_VALUE');"
+  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
+

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

@@ -0,0 +1,23 @@
+#!/bin/bash
+SIGNER_NODE=$1
+
+exec_sql_signer() {
+  sqlite3 data/preview/$1/mithril/stores/signer.sqlite3 "$2"
+}
+
+fix_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;")
+
+  CREATE_EPOCH=$LAST_EPOCH
+  CREATE_EPOCH=$(expr $CREATE_EPOCH - $3)
+  CREATE_EPOCH_HASH=$(echo -n $CREATE_EPOCH | sha256sum | awk '{print $1}')
+  exec_sql_signer $1 "INSERT OR IGNORE INTO $2(key_hash, key, value) VALUES('$CREATE_EPOCH_HASH','$CREATE_EPOCH','$LAST_VALUE');"
+  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
+

mithril-infra/assets/tools/google-credentials-public-key.sh

@@ -0,0 +1,16 @@
+#!/bin/bash
+
+GOOGLE_CREDENTIALS_FILE=$1
+SSH_KEYS_FILE=$2
+VM_USER=$3
+
+# Create a public key from the credentials file and append it to the ssh_keys file
+cat $GOOGLE_CREDENTIALS_FILE | jq -r .private_key > temp.sk
+CREDENTIALS_EMAIL=$(cat $GOOGLE_CREDENTIALS_FILE | jq -r .client_email)
+chmod 0600 temp.sk
+ssh-keygen -q -t rsa -y -f temp.sk -C $CREDENTIALS_EMAIL -b 2048 > temp.pub 
+PUBLIC_KEY=$(cat temp.pub)
+rm -f temp.{sk,pub}
+
+# Add entry to ssh_keys file
+echo $VM_USER:$PUBLIC_KEY $CREDENTIALS_EMAIL >> $SSH_KEYS_FILE