WIP

Author: noel2004

coordinator/internal/controller/proxy/auth.go

@@ -0,0 +1,46 @@
+package proxy
+
+import (
+	"fmt"
+
+	"github.com/gin-gonic/gin"
+
+	"scroll-tech/coordinator/internal/config"
+	"scroll-tech/coordinator/internal/controller/api"
+	"scroll-tech/coordinator/internal/logic/auth"
+	"scroll-tech/coordinator/internal/logic/verifier"
+	"scroll-tech/coordinator/internal/types"
+)
+
+// AuthController is login API
+type AuthController struct {
+	*api.AuthController
+	clients Clients
+}
+
+// NewAuthController returns an LoginController instance
+func NewAuthController(cfg *config.ProxyConfig, clients Clients, vf *verifier.Verifier) *AuthController {
+
+	loginLogic := auth.NewLoginLogicWithSimpleDEduplicator(cfg.ProxyManager.Verifier, vf)
+	auth := api.NewAuthControllerWithLogic(loginLogic)
+	return &AuthController{
+		AuthController: auth,
+		clients:        clients,
+	}
+}
+
+// Login extended the Login hander in api controller
+func (a *AuthController) Login(c *gin.Context) (interface{}, error) {
+
+	ret, err := a.AuthController.Login(c)
+	if err != nil {
+		return nil, err
+	}
+	loginParam := ret.(types.LoginParameterWithHardForkName)
+	// band recursive proxy now ...
+	if loginParam.Message.ProverProviderType == types.ProverProviderTypeProxy {
+		return nil, fmt.Errorf("do not allow recursive proxy for login %v", loginParam.Message)
+	}
+
+	return loginParam, nil
+}

coordinator/internal/controller/proxy/get_task.go

@@ -0,0 +1,45 @@
+package proxy
+
+import (
+	"github.com/gin-gonic/gin"
+	"github.com/prometheus/client_golang/prometheus"
+	"github.com/scroll-tech/go-ethereum/params"
+	"gorm.io/gorm"
+
+	"scroll-tech/common/types/message"
+
+	"scroll-tech/coordinator/internal/config"
+	"scroll-tech/coordinator/internal/logic/provertask"
+	"scroll-tech/coordinator/internal/logic/verifier"
+	coordinatorType "scroll-tech/coordinator/internal/types"
+)
+
+// GetTaskController the get prover task api controller
+type GetTaskController struct {
+	proverTasks map[message.ProofType]provertask.ProverTask
+
+	getTaskAccessCounter *prometheus.CounterVec
+}
+
+// NewGetTaskController create a get prover task controller
+func NewGetTaskController(cfg *config.Config, chainCfg *params.ChainConfig, db *gorm.DB, verifier *verifier.Verifier, reg prometheus.Registerer) *GetTaskController {
+	// TODO: implement proxy get task controller initialization
+	return &GetTaskController{
+		proverTasks: make(map[message.ProofType]provertask.ProverTask),
+	}
+}
+
+func (ptc *GetTaskController) incGetTaskAccessCounter(ctx *gin.Context) error {
+	// TODO: implement proxy get task access counter
+	return nil
+}
+
+// GetTasks get assigned chunk/batch task
+func (ptc *GetTaskController) GetTasks(ctx *gin.Context) {
+	// TODO: implement proxy get tasks logic
+}
+
+func (ptc *GetTaskController) proofType(para *coordinatorType.GetTaskParameter) message.ProofType {
+	// TODO: implement proxy proof type logic
+	return message.ProofTypeChunk
+}

coordinator/internal/controller/proxy/submit_proof.go

@@ -0,0 +1,29 @@
+package proxy
+
+import (
+	"github.com/gin-gonic/gin"
+	"github.com/prometheus/client_golang/prometheus"
+	"github.com/scroll-tech/go-ethereum/params"
+	"gorm.io/gorm"
+
+	"scroll-tech/coordinator/internal/config"
+	"scroll-tech/coordinator/internal/logic/submitproof"
+	"scroll-tech/coordinator/internal/logic/verifier"
+)
+
+// SubmitProofController the submit proof api controller
+type SubmitProofController struct {
+	submitProofReceiverLogic *submitproof.ProofReceiverLogic
+}
+
+// NewSubmitProofController create the submit proof api controller instance
+func NewSubmitProofController(cfg *config.Config, chainCfg *params.ChainConfig, db *gorm.DB, vf *verifier.Verifier, reg prometheus.Registerer) *SubmitProofController {
+	return &SubmitProofController{
+		submitProofReceiverLogic: submitproof.NewSubmitProofReceiverLogic(cfg.ProverManager, chainCfg, db, vf, reg),
+	}
+}
+
+// SubmitProof prover submit the proof to coordinator
+func (spc *SubmitProofController) SubmitProof(ctx *gin.Context) {
+	// TODO: implement proxy submit proof logic
+}

coordinator/internal/middleware/proxy_bearer.go

@@ -0,0 +1 @@
+package middleware