Add Module

Author: Tsuzu

module.go

@@ -0,0 +1,53 @@
+package gopackages
+
+import (
+	"path/filepath"
+
+	"golang.org/x/xerrors"
+)
+
+type Module struct {
+	module  string
+	rootDir string
+}
+
+func NewModule(in string) (*Module, error) {
+	gmp, err := GetGoModPath(in)
+
+	if err != nil {
+		return nil, xerrors.Errorf("failed to call GetGoModPath: %w", err)
+	}
+
+	module, err := GetGoModule(gmp)
+
+	if err != nil {
+		return nil, xerrors.Errorf("failed to call GetGoModule: %w", err)
+	}
+
+	rootDir, err := filepath.Abs(filepath.Dir(gmp))
+
+	if err != nil {
+		return nil, xerrors.Errorf("failed to get absolute path for the directory of go.mod: %w", err)
+	}
+
+	return &Module{
+		module:  module,
+		rootDir: rootDir,
+	}, nil
+}
+
+func (m *Module) GetImportPath(path string) (string, error) {
+	abs, err := filepath.Abs(path)
+
+	if err != nil {
+		return "", xerrors.Errorf("failed to get absolute path for %s: %w", path, err)
+	}
+
+	rel, err := filepath.Rel(m.rootDir, abs)
+
+	if err != nil {
+		return "", xerrors.Errorf("failed to calculate relative path from %s to %s: %w", m.rootDir, abs, err)
+	}
+
+	return filepath.Join(m.module, rel), nil
+}

module_test.go

@@ -0,0 +1,42 @@
+package gopackages
+
+import "testing"
+
+func TestModule_GetImportPath(t *testing.T) {
+	m, err := NewModule(".")
+
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	type args struct {
+		path string
+	}
+	tests := []struct {
+		name    string
+		args    args
+		want    string
+		wantErr bool
+	}{
+		{
+			name: "normal",
+			args: args{
+				path: "./tests",
+			},
+			want:    "github.com/go-utils/gopackages/tests",
+			wantErr: false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := m.GetImportPath(tt.args.path)
+			if (err != nil) != tt.wantErr {
+				t.Errorf("Module.GetImportPath() error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+			if got != tt.want {
+				t.Errorf("Module.GetImportPath() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}