From a026816c516b7bb81eafd963b02c3e51a7fdd2b9 Mon Sep 17 00:00:00 2001
From: David <david@taiko.xyz>
Date: Tue, 19 Sep 2023 23:57:27 +0800
Subject: [PATCH] feat(prover): always use the oracle prover private key when
 an oracle prover starting a server (#395)

---
 prover/prover.go | 24 ++++++++++++++----------
 1 file changed, 14 insertions(+), 10 deletions(-)

diff --git a/prover/prover.go b/prover/prover.go
index 22c1cecf6..452dd6b5f 100644
--- a/prover/prover.go
+++ b/prover/prover.go
@@ -127,16 +127,6 @@ func InitFromConfig(ctx context.Context, p *Prover, cfg *Config) (err error) {
 		return err
 	}
 
-	// Prover server
-	if p.srv, err = server.New(&server.NewProverServerOpts{
-		ProverPrivateKey: p.cfg.L1ProverPrivKey,
-		MinProofFee:      p.cfg.MinProofFee,
-		MaxExpiry:        p.cfg.MaxExpiry,
-		CapacityManager:  p.capacityManager,
-	}); err != nil {
-		return err
-	}
-
 	// Configs
 	protocolConfigs, err := p.rpc.TaikoL1.GetConfig(&bind.CallOpts{Context: ctx})
 	if err != nil {
@@ -214,6 +204,20 @@ func InitFromConfig(ctx context.Context, p *Prover, cfg *Config) (err error) {
 		return err
 	}
 
+	// Prover server
+	proverServerOpts := &server.NewProverServerOpts{
+		ProverPrivateKey: p.cfg.L1ProverPrivKey,
+		MinProofFee:      p.cfg.MinProofFee,
+		MaxExpiry:        p.cfg.MaxExpiry,
+		CapacityManager:  p.capacityManager,
+	}
+	if p.cfg.OracleProver {
+		proverServerOpts.ProverPrivateKey = p.cfg.OracleProverPrivateKey
+	}
+	if p.srv, err = server.New(proverServerOpts); err != nil {
+		return err
+	}
+
 	return nil
 }