From 003d515b96385f23e44b7d6e7fcb52c5abeb3bff Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Philipp=20Gill=C3=A9?= <philipp.gille@gmail.com>
Date: Sat, 27 Jan 2024 15:16:31 +0100
Subject: [PATCH] Add replace function in Magefile

---
 magefiles/magefile.go | 53 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)

diff --git a/magefiles/magefile.go b/magefiles/magefile.go
index a2faba9d..e3483d15 100644
--- a/magefiles/magefile.go
+++ b/magefiles/magefile.go
@@ -4,6 +4,8 @@ package main
 
 import (
 	"errors"
+	"os"
+	"path/filepath"
 	"runtime"
 
 	"github.com/bitfield/script"
@@ -83,3 +85,54 @@ func Clean() error {
 	p.Wait()
 	return p.Error()
 }
+
+// Replace replaces the tagged/versioned gokv dependencies with the local version.
+// This is useful before tagging/releasing a new version of a helper package,
+// to ensure all store implementations work with those latest helper package versions.
+func Replace() error {
+	rootDir, err := os.Getwd()
+	if err != nil {
+		return err
+	}
+
+	impls, err := script.File("./build/implementations").Slice()
+	if err != nil {
+		return err
+	}
+	replaceText := `
+replace (
+	github.com/philippgille/gokv => ../
+	github.com/philippgille/gokv/encoding => ../encoding
+	github.com/philippgille/gokv/test => ../test
+	github.com/philippgille/gokv/util => ../util
+	github.com/philippgille/gokv/sql => ../sql
+)
+`
+	for _, impl := range impls {
+		modFile := filepath.Join(".", impl, "go.mod")
+		content, err := os.ReadFile(modFile)
+		if err != nil {
+			return err
+		}
+		newContent := string(content) + replaceText
+
+		err = os.WriteFile(modFile, []byte(newContent), os.ModePerm)
+		if err != nil {
+			return err
+		}
+
+		err = os.Chdir(filepath.Join(".", impl))
+		if err != nil {
+			return err
+		}
+		err = sh.Run("go", "mod", "tidy")
+		if err != nil {
+			return err
+		}
+		err = os.Chdir(rootDir)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}