Merge branch 'develop'

Author: nykma

cmd/lambda_worker/main.go

@@ -67,7 +67,7 @@ func handler(ctx context.Context, sqs_event events.SQSEvent) error {
 			arweaveMsgs = append(arweaveMsgs, &message)
 		case types.QueueActions.Revalidate:
 			if err := revalidate_single(ctx, &message); err != nil {
-				return err
+				fmt.Printf("error revalidating proof record %d: %s\n", message.ProofID, err)
 			}
 		default:
 			logrus.Warnf("unsupported queue action: %s", message.Action)
@@ -217,7 +217,7 @@ func arweave_bundle_single(pc *model.ProofChain, previous *model.ProofChain) (*a
 
 func revalidate_single(ctx context.Context, message *types.QueueMessage) error {
 	proof := model.Proof{}
-	tx := model.DB.Preload("ProofChain.Previous").First(&proof, message.ProofID)
+	tx := model.DB.Preload("ProofChain").Preload("ProofChain.Previous").Where("id = ?", message.ProofID).First(&proof)
 	if tx.Error != nil {
 		return xerrors.Errorf("%w", tx.Error)
 	}

controller/proof_exists.go

@@ -56,6 +56,7 @@ func proofExists(c *gin.Context) {
 		errorResp(c, http.StatusNotFound, xerrors.Errorf("Record not found for %s: %s", req.Platform, req.Identity))
 		return
 	}
+	go triggerRevalidate(found.ID)
 
 	c.JSON(http.StatusOK, ProofExistsResponse{
 		CreatedAt:     strconv.FormatInt(found.CreatedAt.Unix(), 10),

controller/proof_query.go

@@ -8,6 +8,7 @@ import (
 	"github.com/gin-gonic/gin"
 	"github.com/nextdotid/proof-server/model"
 	"github.com/nextdotid/proof-server/types"
+	"github.com/nextdotid/proof-server/util/sqs"
 	"github.com/samber/lo"
 	"golang.org/x/xerrors"
 )
@@ -123,6 +124,12 @@ func performProofQuery(req ProofQueryRequest) ([]ProofQueryResponseSingle, Proof
 	if tx.Error != nil || tx.RowsAffected == int64(0) || len(proofs) == 0 {
 		return result, pagination
 	}
+	// Trigger revalidate procedure
+	lo.ForEach(proofs, func(proof model.Proof, i int) {
+		if proof.IsOutdated() {
+			go triggerRevalidate(proof.ID)
+		}
+	})
 
 	personas := lo.Map(proofs, func(p model.Proof, _index int) string {
 		return p.Persona
@@ -166,3 +173,16 @@ func performProofQuery(req ProofQueryRequest) ([]ProofQueryResponseSingle, Proof
 	}
 	return result, pagination
 }
+
+func triggerRevalidate(proofID int64) error {
+	msg := types.QueueMessage{
+		Action:  types.QueueActions.Revalidate,
+		ProofID: proofID,
+	}
+
+	if err := sqs.Send(msg); err != nil {
+		return xerrors.Errorf("Failed to send message to queue: %w", err)
+	}
+
+	return nil
+}

model/proof.go

@@ -8,6 +8,9 @@ import (
 	"golang.org/x/xerrors"
 )
 
+// EXPIRED_IN is the time after which a proof is considered expired and should perform revalidate.
+const EXPIRED_IN = time.Hour * 24 * 3
+
 // Proof is final proof state of a user (persona).
 type Proof struct {
 	ID            int64 `gorm:"primarykey"`
@@ -39,6 +42,11 @@ func FindAllProofByPersona(persona any) (proofs []Proof, err error) {
 	return proofs, nil
 }
 
+// IsOutdated returns true if proof is outdated and should do a revalidate.
+func (proof *Proof) IsOutdated() bool {
+	return proof.LastCheckedAt.Add(EXPIRED_IN).Before(time.Now())
+}
+
 // Revalidate validates current proof, will update `IsValid` and
 // `LastCheckedAt`. Must be used after `DB.Preload("ProofChain")`.
 func (proof *Proof) Revalidate() (result bool, err error) {