revert mock changes, more dev in db functions

aggsender/db/aggsender_db_storage.go

@@ -37,7 +37,7 @@ type AggSenderStorage interface {
 	// Add auth-proof in DB
 	AddAuthProof(ctx context.Context, authProof types.AuthProof) error
 	// Get auth-proof by identifier
-	GetAuthProof(identifier string) (*types.AuthProof, error)
+	GetAuthProof(endBlock uint64) (*types.AuthProof, error)
 	// Validate auth-proof
 	ValidateProof(req *types.ProofRequest) (bool, error)
 }
@@ -87,25 +87,24 @@ func (a *AggSenderSQLStorage) AddAuthProof(ctx context.Context, authProof types.
 		}
 	}()
 
-	if _, err = tx.Exec(`INSERT INTO auth_proof (identifier, proof) VALUES ($1, $2);`,
-		authProof.Identifier, authProof.Proof); err != nil {
+	if _, err = tx.Exec(`INSERT INTO auth_proof (start_block, end_block, proof) VALUES ($1, $2, $3);`,
+		authProof.StartBlock, authProof.EndBlock, authProof.Proof); err != nil {
 		return fmt.Errorf("error inserting auth proof: %w", err)
 	}
 	if err = tx.Commit(); err != nil {
 		return err
 	}
 
-	a.logger.Debugf("inserted auth proof - Identifier: %s", authProof.Identifier)
+	a.logger.Debugf("inserted auth proof - start block: %d", authProof.StartBlock, "end block: %d", authProof.EndBlock)
 
 	return nil
 }
 
-func (a *AggSenderSQLStorage) GetAuthProof(identifier string) (*types.AuthProof, error) {
+func (a *AggSenderSQLStorage) GetAuthProof(endBlock uint64) (*types.AuthProof, error) {
 	var authProof types.AuthProof
 	if err := meddler.QueryRow(a.db, &authProof,
-		"SELECT * FROM auth_proof WHERE identifier = $1;", identifier); err != nil {
-		// Fix this
-		return nil, getSelectQueryError(0, err)
+		"SELECT * FROM auth_proof WHERE start_block = $1;", endBlock); err != nil {
+		return nil, getSelectQueryError(endBlock, err)
 	}
 
 	return &authProof, nil

aggsender/db/aggsender_db_storage_test.go

@@ -445,7 +445,8 @@ func TestValidateProof(t *testing.T) {
 
 	t.Run("ValidateProof", func(t *testing.T) {
 		mockProofRequest := &types.ProofRequest{
-			Identifier: "identifier",
+			StartBlock: 0,
+			EndBlock:   1,
 			Proof:      "proof",
 		}
 		valid, err := storage.ValidateProof(mockProofRequest)

aggsender/db/migrations/0002.sql

@@ -3,7 +3,10 @@ DROP TABLE IF EXISTS auth_proof;
 
 -- +migrate Up
 CREATE TABLE auth_proof (
-    indentifier    VARCHAR,
+    start_block    INTEGER NOT NULL,
+    end_block      INTEGER NOT NULL,
     proof          VARCHAR,
-    PRIMARY KEY (indentifier)
+    created_at     INTEGER NOT NULL,
+    updated_at     INTEGER NOT NULL,
+    PRIMARY KEY (start_block, end_block)
 );

aggsender/grpc/handler.go

@@ -8,7 +8,6 @@ import (
 )
 
 func (s *server) ReceiveAuthProof(ctx context.Context, req *types.ProofRequest) (*types.ProofResponse, error) {
-	// Implement your logic here
 	log.Printf("Received proof: %v", req)
 	valid, err := s.aggsenderStorage.ValidateProof(req)
 	if err != nil {
@@ -24,7 +23,8 @@ func (s *server) ReceiveAuthProof(ctx context.Context, req *types.ProofRequest)
 	log.Printf("Proof is valid")
 	err = s.aggsenderStorage.AddAuthProof(context.Background(), types.AuthProof{
 		Proof:      req.Proof,
-		Identifier: req.Identifier,
+		StartBlock: req.StartBlock,
+		EndBlock:   req.EndBlock,
 	})
 	if err != nil {
 		log.Printf("Error adding proof: %v", err)

aggsender/proto/aggsender.proto

@@ -5,8 +5,9 @@ package types;
 option go_package = "github.com/agglayer/aggkit/aggsender/types";
 
 message ProofRequest {
-  string identifier = 1;
-  string proof = 2;
+  uint64 start_block = 1;
+  uint64 end_block = 2;
+  string proof = 3;
 }
 
 message ProofResponse {

aggsender/types/types.go

@@ -56,7 +56,8 @@ type Logger interface {
 }
 
 type AuthProof struct {
-	Identifier string
+	StartBlock uint64
+	EndBlock   uint64
 	Proof      string
 }