easyp-tech · hound672 · May 9, 2025 · May 16, 2025 · May 16, 2025
diff --git a/README.md b/README.md
@@ -182,6 +182,22 @@ for example:
     insteadOf = https://github.com/
 ```
 
+### Use mirrors
+
+For use your company private mirrors (e.g artifactory) you can use `mirrors` settings.
+
+```yaml
+mirrors:
+  - origin: onprem-vcs.loc
+    use: github.com
+```
+
+In that case dependency from your config with host `onprem-vcs.loc` will be replaced with `github.com`.
+
+So you are able to download deps from your mirror without modificate easyp config file.
+
+In case when you have to use private 
+
 ## Auto-completion
 
 ### zsh auto-completion

diff --git a/example.easyp.yaml b/example.easyp.yaml
@@ -7,9 +7,9 @@ lint:
     - COMMENT_SERVICE
     - IMPORT_USED
 deps:
-  - github.com/googleapis/googleapis
-  - github.com/bufbuild/protovalidate@v0.3.1
-  - github.com/grpc-ecosystem/grpc-gateway@v2.19.1
+  - onprem-vcs.loc/googleapis/googleapis
+#  - github.com/bufbuild/protovalidate@v0.3.1
+#  - github.com/grpc-ecosystem/grpc-gateway@v2.19.1
 
 breaking:
   ignore:
@@ -45,3 +45,6 @@ generate:
       opts:
         paths: source_relative
 
+mirrors:
+  - origin: onprem-vcs.loc
+    use: github.com
diff --git a/internal/adapters/repository/git/git.go b/internal/adapters/repository/git/git.go
@@ -3,6 +3,7 @@ package git
 import (
 	"context"
 	"fmt"
+	"log/slog"
 	"os"
 	"path/filepath"
 
@@ -39,13 +40,15 @@ type Console interface {
 
 // New returns gitRepo instance
 // remote: full remoteURL address without schema
-func New(ctx context.Context, remote string, cacheDir string, console Console) (repository.Repo, error) {
+func New(ctx context.Context, remoteURL string, cacheDir string, console Console) (repository.Repo, error) {
 	r := &gitRepo{
-		remoteURL: getRemote(remote),
+		remoteURL: getRemote(remoteURL),
 		cacheDir:  cacheDir,
 		console:   console,
 	}
 
+	slog.Debug("clone", "remoteURL", remoteURL, "cacheDir", cacheDir)
+
 	if _, err := os.Stat(filepath.Join(r.cacheDir, "objects")); err == nil {
 		// repo is already exists
 		return r, nil

diff --git a/internal/api/temporaly_helper.go b/internal/api/temporaly_helper.go
@@ -79,6 +79,14 @@ func buildCore(_ context.Context, cfg config.Config, dirWalker core.DirWalker) (
 		AgainstGitRef: cfg.BreakingCheck.AgainstGitRef,
 	}
 
+	mirrors := lo.Map(cfg.Mirrors, func(item config.Mirror, _ int) core.MirrorConfig {
+		return core.MirrorConfig{
+			Origin: item.Origin,
+			Use:    item.Use,
+		}
+	})
+	depAddressResolver := core.NewDepAddressResolver(mirrors)
+
 	app := core.New(
 		lintRules,
 		cfg.Lint.Ignore,
@@ -117,6 +125,7 @@ func buildCore(_ context.Context, cfg config.Config, dirWalker core.DirWalker) (
 		lockFile,
 		currentProjectGitWalker,
 		breakingCheckConfig,
+		depAddressResolver,
 	)
 
 	return app, nil

diff --git a/internal/config/config.go b/internal/config/config.go
@@ -48,6 +48,12 @@ type InputFilesDir struct {
 	Root string `yaml:"root"`
 }
 
+// Mirror is the mirror setup for dependency
+type Mirror struct {
+	Origin string `yaml:"origin"`
+	Use    string `yaml:"use"`
+}
+
 // Config is the configuration of easyp.
 type Config struct {
 	// LintConfig is the lint configuration.
@@ -61,6 +67,9 @@ type Config struct {
 
 	// BreakingCheck `breaking` command's configuration
 	BreakingCheck BreakingCheck `json:"breaking" yaml:"breaking"`
+
+	// Mirrors configs
+	Mirrors []Mirror `json:"mirrors" yaml:"mirrors"`
 }
 
 var errFileNotFound = errors.New("config file not found")

diff --git a/internal/core/core.go b/internal/core/core.go
@@ -22,6 +22,8 @@ type Core struct {
 
 	breakingCheckConfig     BreakingCheckConfig
 	currentProjectGitWalker CurrentProjectGitWalker
+
+	depAddressResolver DepAddressResolver
 }
 
 var (
@@ -43,6 +45,7 @@ func New(
 	lockFile LockFile,
 	currentProjectGitWalker CurrentProjectGitWalker,
 	breakingCheckConfig BreakingCheckConfig,
+	depAddressResolver DepAddressResolver,
 ) *Core {
 	return &Core{
 		rules:                   rules,
@@ -58,5 +61,6 @@ func New(
 		lockFile:                lockFile,
 		currentProjectGitWalker: currentProjectGitWalker,
 		breakingCheckConfig:     breakingCheckConfig,
+		depAddressResolver:      depAddressResolver,
 	}
 }
diff --git a/internal/core/dep_address_resolver.go b/internal/core/dep_address_resolver.go
@@ -0,0 +1,33 @@
+package core
+
+import (
+	"strings"
+)
+
+type MirrorConfig struct {
+	Origin string
+	Use    string
+}
+
+type DepAddressResolver struct {
+	mirrors []MirrorConfig
+}
+
+func NewDepAddressResolver(mirrors []MirrorConfig) DepAddressResolver {
+	return DepAddressResolver{
+		mirrors: mirrors,
+	}
+}
+
+func (r DepAddressResolver) Resolve(requestedModule string) string {
+	requestedModule = r.useMirrors(requestedModule)
+	return requestedModule
+}
+
+func (r DepAddressResolver) useMirrors(requestedModule string) string {
+	for _, mirror := range r.mirrors {
+		requestedModule = strings.Replace(requestedModule, mirror.Origin, mirror.Use, 1)
+	}
+
+	return requestedModule
+}
diff --git a/internal/core/dep_address_resolver_test.go b/internal/core/dep_address_resolver_test.go
@@ -0,0 +1,39 @@
+package core
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func Test_DepAddressResolver_Resolve(t *testing.T) {
+	mirrors := []MirrorConfig{
+		{
+			Origin: "github.com",
+			Use:    "gitlab.com",
+		},
+	}
+
+	resolver := NewDepAddressResolver(mirrors)
+
+	tests := map[string]struct {
+		requestedModule string
+		expected        string
+	}{
+		"not replaced": {
+			requestedModule: "vcs.com/github/octocat.git",
+			expected:        "vcs.com/github/octocat.git",
+		},
+		"replaced": {
+			requestedModule: "github.com/github/octocat.git",
+			expected:        "gitlab.com/github/octocat.git",
+		},
+	}
+
+	for name, test := range tests {
+		t.Run(name, func(t *testing.T) {
+			res := resolver.Resolve(test.requestedModule)
+			require.Equal(t, test.expected, res)
+		})
+	}
+}
diff --git a/internal/core/get.go b/internal/core/get.go
@@ -17,8 +17,8 @@ func (c *Core) Get(ctx context.Context, requestedModule models.Module) error {
 		return fmt.Errorf("c.storage.CreateCacheRepositoryDir: %w", err)
 	}
 
-	// TODO: use factory (git, svn etc)
-	repo, err := git.New(ctx, requestedModule.Name, cacheRepositoryDir, c.console)
+	remoteURL := c.depAddressResolver.Resolve(requestedModule.Name)
+	repo, err := git.New(ctx, remoteURL, cacheRepositoryDir, c.console)
 	if err != nil {
 		return fmt.Errorf("git.New: %w", err)
 	}