Auto detect chain id and chain name

Tzkt.Api/Models/State.cs

@@ -1,12 +1,19 @@
 using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Threading.Tasks;
 
 namespace Tzkt.Api.Models
 {
     public class State
     {
+        /// <summary>
+        /// Alias name of the chain (or "private" if it's not on the list of known chains)
+        /// </summary>
+        public string Chain { get; set; }
+
+        /// <summary>
+        /// Unique identificator of the chain
+        /// </summary>
+        public string ChainId { get; set; }
+
         /// <summary>
         /// Current cycle
         /// </summary>

Tzkt.Api/Repositories/StateRepository.cs

@@ -23,6 +23,8 @@ public State Get()
             var appState = State.Current;
             return new State
             {
+                Chain = appState.Chain,
+                ChainId = appState.ChainId,
                 KnownLevel = appState.KnownHead,
                 LastSync = appState.LastSync,
                 Hash = appState.Hash,

Tzkt.Api/Services/Cache/State/RawModels/RawState.cs

@@ -4,6 +4,10 @@ namespace Tzkt.Api.Services.Cache
 {
     public class RawState
     {
+        public string Chain { get; set; }
+
+        public string ChainId { get; set; }
+
         public int KnownHead { get; set; }
 
         public DateTime LastSync { get; set; }

Tzkt.Data/Migrations/20210714124853_Initial.cs

@@ -14,6 +14,8 @@ protected override void Up(MigrationBuilder migrationBuilder)
                 {
                     Id = table.Column<int>(type: "integer", nullable: false)
                         .Annotation("Npgsql:ValueGenerationStrategy", NpgsqlValueGenerationStrategy.IdentityByDefaultColumn),
+                    Chain = table.Column<string>(type: "text", nullable: true),
+                    ChainId = table.Column<string>(type: "text", nullable: true),
                     KnownHead = table.Column<int>(type: "integer", nullable: false),
                     LastSync = table.Column<DateTime>(type: "timestamp without time zone", nullable: false),
                     Cycle = table.Column<int>(type: "integer", nullable: false),
@@ -1205,8 +1207,8 @@ protected override void Up(MigrationBuilder migrationBuilder)
 
             migrationBuilder.InsertData(
                 table: "AppState",
-                columns: new[] { "Id", "AccountCounter", "AccountsCount", "ActivationOpsCount", "BallotOpsCount", "BigMapCounter", "BigMapKeyCounter", "BigMapUpdateCounter", "BlocksCount", "CommitmentsCount", "Cycle", "CyclesCount", "DelegationOpsCount", "DoubleBakingOpsCount", "DoubleEndorsingOpsCount", "EndorsementOpsCount", "Hash", "KnownHead", "LastSync", "Level", "ManagerCounter", "MigrationOpsCount", "NextProtocol", "NonceRevelationOpsCount", "OperationCounter", "OriginationOpsCount", "ProposalOpsCount", "ProposalsCount", "Protocol", "ProtocolsCount", "QuoteBtc", "QuoteCny", "QuoteEth", "QuoteEur", "QuoteJpy", "QuoteKrw", "QuoteLevel", "QuoteUsd", "RevealOpsCount", "RevelationPenaltyOpsCount", "ScriptCounter", "StorageCounter", "Timestamp", "TransactionOpsCount", "VotingEpoch", "VotingPeriod" },
-                values: new object[] { -1, 0, 0, 0, 0, 0, 0, 0, 0, 0, -1, 0, 0, 0, 0, 0, "", 0, new DateTime(1, 1, 1, 0, 0, 0, 0, DateTimeKind.Unspecified), -1, 0, 0, "", 0, 0, 0, 0, 0, "", 0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, -1, 0.0, 0, 0, 0, 0, new DateTime(1, 1, 1, 0, 0, 0, 0, DateTimeKind.Unspecified), 0, -1, -1 });
+                columns: new[] { "Id", "AccountCounter", "AccountsCount", "ActivationOpsCount", "BallotOpsCount", "BigMapCounter", "BigMapKeyCounter", "BigMapUpdateCounter", "BlocksCount", "Chain", "ChainId", "CommitmentsCount", "Cycle", "CyclesCount", "DelegationOpsCount", "DoubleBakingOpsCount", "DoubleEndorsingOpsCount", "EndorsementOpsCount", "Hash", "KnownHead", "LastSync", "Level", "ManagerCounter", "MigrationOpsCount", "NextProtocol", "NonceRevelationOpsCount", "OperationCounter", "OriginationOpsCount", "ProposalOpsCount", "ProposalsCount", "Protocol", "ProtocolsCount", "QuoteBtc", "QuoteCny", "QuoteEth", "QuoteEur", "QuoteJpy", "QuoteKrw", "QuoteLevel", "QuoteUsd", "RevealOpsCount", "RevelationPenaltyOpsCount", "ScriptCounter", "StorageCounter", "Timestamp", "TransactionOpsCount", "VotingEpoch", "VotingPeriod" },
+                values: new object[] { -1, 0, 0, 0, 0, 0, 0, 0, 0, null, null, 0, -1, 0, 0, 0, 0, 0, "", 0, new DateTime(1, 1, 1, 0, 0, 0, 0, DateTimeKind.Unspecified), -1, 0, 0, "", 0, 0, 0, 0, 0, "", 0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, -1, 0.0, 0, 0, 0, 0, new DateTime(1, 1, 1, 0, 0, 0, 0, DateTimeKind.Unspecified), 0, -1, -1 });
 
             migrationBuilder.CreateIndex(
                 name: "IX_Accounts_Address",

Tzkt.Data/Migrations/TzktContextModelSnapshot.cs

@@ -170,6 +170,12 @@ protected override void BuildModel(ModelBuilder modelBuilder)
                     b.Property<int>("BlocksCount")
                         .HasColumnType("integer");
 
+                    b.Property<string>("Chain")
+                        .HasColumnType("text");
+
+                    b.Property<string>("ChainId")
+                        .HasColumnType("text");
+
                     b.Property<int>("CommitmentsCount")
                         .HasColumnType("integer");
 

Tzkt.Data/Models/AppState.cs

@@ -3,23 +3,26 @@
 
 namespace Tzkt.Data.Models
 {
-    //TODO: add chain id
     public class AppState
     {
         public int Id { get; set; }
+        public string Chain { get; set; }
+        public string ChainId { get; set; }
         public int KnownHead { get; set; }
         public DateTime LastSync { get; set; }
 
+        #region head
         public int Cycle { get; set; }
         public int Level { get; set; }
         public DateTime Timestamp { get; set; }
         public string Protocol { get; set; }
         public string NextProtocol { get; set; }
         public string Hash { get; set; }
-
         public int VotingEpoch { get; set; }
         public int VotingPeriod { get; set; }
+        #endregion
 
+        #region counters
         public int AccountCounter { get; set; }
         public int OperationCounter { get; set; }
         public int ManagerCounter { get; set; }
@@ -28,6 +31,7 @@ public class AppState
         public int BigMapUpdateCounter { get; set; }
         public int StorageCounter { get; set; }
         public int ScriptCounter { get; set; }
+        #endregion
 
         #region entities count
         public int CommitmentsCount { get; set; }

Tzkt.Sync/Protocols/Handlers/Genesis/GenesisHandler.cs

@@ -67,6 +67,8 @@ public override Task Commit(JsonElement rawBlock)
 
             #region update state
             var state = Cache.AppState.Get();
+            state.ChainId = rawBlock.RequiredString("chain_id");
+            state.Chain = Chains.GetName(state.ChainId);
             state.Cycle = -1;
             state.Level = block.Level;
             state.Timestamp = block.Timestamp;
@@ -93,6 +95,8 @@ await Db.Database.ExecuteSqlRawAsync(@"
 
             #region update state
             var state = Cache.AppState.Get();
+            state.ChainId = null;
+            state.Chain = null;
             state.Cycle = -1;
             state.Level = -1;
             state.Timestamp = DateTime.MinValue;

Tzkt.Sync/Protocols/Handlers/Initiator/Validation/Validator.cs

@@ -15,6 +15,9 @@ public Validator(ProtocolHandler protocol)
 
         public Task ValidateBlock(JsonElement block)
         {
+            if (block.RequiredString("chain_id") != Cache.AppState.GetChainId())
+                throw new ValidationException("invalid chain");
+
             if (block.Required("header").RequiredInt32("level") != Cache.AppState.GetNextLevel())
                 throw new ValidationException("invalid block level", true);
 

Tzkt.Sync/Protocols/Handlers/Proto1/Validation/Validator.cs

@@ -22,6 +22,9 @@ class Validator : IValidator
 
         public virtual async Task ValidateBlock(JsonElement block)
         {
+            if (block.RequiredString("chain_id") != Cache.AppState.GetChainId())
+                throw new ValidationException("invalid chain");
+
             var protocol = block.RequiredString("protocol");
             Protocol = await Cache.Protocols.GetAsync(protocol);
 

Tzkt.Sync/Protocols/Helpers/Chains.cs

@@ -0,0 +1,14 @@
+namespace Tzkt.Sync.Protocols
+{
+    static class Chains
+    {
+        public static string GetName(string chainId) => chainId switch
+        {
+            "NetXdQprcVkpaWU" => "mainnet",
+            "NetXSgo1ZT2DRUG" => "edo2net",
+            "NetXxkAx4woPLyu" => "florencenet",
+            "NetXz969SFaFn8k" => "granadanet",
+            _ => "private"
+        };
+    }
+}

Tzkt.Sync/Services/Cache/AppStateCache.cs

@@ -27,6 +27,11 @@ public AppState Get()
             return AppState;
         }
 
+        public string GetChainId()
+        {
+            return AppState.ChainId;
+        }
+
         public int GetLevel()
         {
             return AppState.Level;

Tzkt.Sync/Services/TezosNode/TezosNode.cs

@@ -1,6 +1,4 @@
 using System;
-using System.IO;
-using System.Collections.Generic;
 using System.Threading.Tasks;
 using Microsoft.Extensions.Configuration;
 using Microsoft.Extensions.DependencyInjection;
@@ -10,7 +8,6 @@ namespace Tzkt.Sync.Services
 {
     public sealed class TezosNode : IDisposable
     {
-        readonly string ChainId;
         readonly TzktClient Rpc;
 
         Header Header;
@@ -20,7 +17,6 @@ public sealed class TezosNode : IDisposable
         public TezosNode(IConfiguration config)
         {
             var nodeConf = config.GetTezosNodeConfig();
-            ChainId = nodeConf.ChainId;
             Rpc = new TzktClient(nodeConf.Endpoint, nodeConf.Timeout);
         }
 
@@ -36,9 +32,6 @@ public async Task<Header> GetHeaderAsync()
             {
                 var header = await Rpc.GetObjectAsync<Header>("chains/main/blocks/head/header");
 
-                if (header.ChainId != ChainId)
-                    throw new Exception("Invalid chain");
-
                 if (header.Protocol != Header?.Protocol)
                     Constants = await Rpc.GetObjectAsync<Constants>("chains/main/blocks/head/context/constants");
 
@@ -52,14 +45,9 @@ public async Task<Header> GetHeaderAsync()
             return Header;
         }
 
-        public async Task<Header> GetHeaderAsync(int level)
+        public Task<Header> GetHeaderAsync(int level)
         {
-            var header = await Rpc.GetObjectAsync<Header>($"chains/main/blocks/{level}/header");
-
-            if (header.ChainId != ChainId)
-                throw new Exception("Invalid chain");
-
-            return header;
+            return Rpc.GetObjectAsync<Header>($"chains/main/blocks/{level}/header");
         }
 
         public async Task<bool> HasUpdatesAsync(int level)

Tzkt.Sync/Services/TezosNode/TezosNodeConfig.cs

@@ -4,7 +4,6 @@ namespace Tzkt.Sync.Services
 {
     public class TezosNodeConfig
     {
-        public string ChainId { get; set; }
         public string Endpoint { get; set; }
         public int Timeout { get; set; }
     }

Tzkt.Sync/appsettings.json

@@ -4,7 +4,6 @@
     "Validation": true
   },
   "TezosNode": {
-    "ChainId": "NetXdQprcVkpaWU",
     "Endpoint": "https://mainnet-tezos.giganode.io/",
     "Timeout": 60
   },