From 8dc64cb7e0a7dd3ede5cb4fcd999ac799fe6b8dd Mon Sep 17 00:00:00 2001
From: Googol Lee <i@googol.im>
Date: Wed, 20 Sep 2023 11:16:57 +0200
Subject: [PATCH] add builder to module.

---
 module.go      | 44 +++++++++++++++++++++++++++++++++++++++-----
 module_test.go |  9 +++++++++
 2 files changed, 48 insertions(+), 5 deletions(-)
 create mode 100644 module_test.go

diff --git a/module.go b/module.go
index 1df03be..81cd0b9 100644
--- a/module.go
+++ b/module.go
@@ -11,7 +11,7 @@ var ErrModuleDependError = errors.New("depend error")
 
 type Module interface {
 	Name() moduleName
-	DependOn() []Module
+	DependOn() []moduleName
 	CheckHealthy(context.Context) error
 }
 
@@ -23,15 +23,16 @@ type moduleName string
 
 type ModuleType[T ModuleImplementer] struct {
 	name    moduleName
-	depends []Module
+	builder func(context.Context) (T, error)
+	depends []moduleName
 }
 
-func DefineModule[T ModuleImplementer](depends ...Module) *ModuleType[T] {
+func NewModule[T ModuleImplementer](builder func(context.Context) (T, error)) *ModuleType[T] {
 	var t T
 	name := reflect.TypeOf(t).Name()
 	return &ModuleType[T]{
 		name:    moduleName(name),
-		depends: depends,
+		builder: builder,
 	}
 }
 
@@ -39,10 +40,43 @@ func (m ModuleType[T]) Name() moduleName {
 	return m.name
 }
 
-func (m ModuleType[T]) DependOn() []Module {
+func (m ModuleType[T]) DependOn() []moduleName {
 	return m.depends
 }
 
+type buildContext struct {
+	context.Context
+	deps map[moduleName]struct{}
+}
+
+func (c *buildContext) Value(key any) any {
+	name, ok := key.(moduleName)
+	if !ok {
+		return c.Context.Value(key)
+	}
+
+	c.deps[name] = struct{}{}
+	return c.Context.Value(name)
+}
+
+func (m *ModuleType[T]) Build(ctx context.Context) (T, error) {
+	buildContext := &buildContext{
+		Context: ctx,
+		deps:    make(map[moduleName]struct{}),
+	}
+	ret, err := m.builder(buildContext)
+	if err != nil {
+		return ret, err
+	}
+
+	m.depends = make([]moduleName, 0, len(buildContext.deps))
+	for name := range buildContext.deps {
+		m.depends = append(m.depends, name)
+	}
+
+	return ret, nil
+}
+
 func (m ModuleType[T]) Value(ctx context.Context) T {
 	var n T
 	v := ctx.Value(m.name)
diff --git a/module_test.go b/module_test.go
new file mode 100644
index 0000000..7e48012
--- /dev/null
+++ b/module_test.go
@@ -0,0 +1,9 @@
+package espresso
+
+import "context"
+
+type fakeModule struct{}
+
+func (fakeModule) CheckHealthy(context.Context) error { return nil }
+
+var _ Module = &ModuleType[fakeModule]{}