refactor: Into ECS

1 files changed, 39 insertions, 0 deletions

diff --git a/entity.go b/entity.go
new file mode 100644
index 0000000..6066772
--- /dev/null
+++ b/entity.go
@@ -0,0 +1,39 @@
+package main
+
+import "github.com/hajimehoshi/ebiten/v2"
+
+type Component[T any] struct {
+	IsEnabled bool
+	Value     T
+}
+
+func C[T any](t T) Component[T] {
+	return Component[T]{IsEnabled: true, Value: t}
+}
+
+type Entity struct {
+	Position  Component[Point]
+	Image     Component[*ebiten.Image]
+	Selection Component[Selection]
+	Move      Component[Move]
+}
+
+func (e Entity) Bounds() Rectangle {
+	return Rectangle{
+		e.Position.Value,
+		e.Position.Value.Add(e.Image.Value.Bounds().Size()),
+	}
+}
+
+func Draw(screen *ebiten.Image, e *Entity) {
+	if e.Image.IsEnabled && e.Position.IsEnabled {
+		op := &ebiten.DrawImageOptions{}
+		op.GeoM.Translate(float64(e.Position.Value.X), float64(e.Position.Value.Y))
+		if e.Selection.IsEnabled && e.Selection.Value.IsSelected {
+			op.GeoM.Translate(-10, -10)
+			screen.DrawImage(e.Selection.Value.SelectedImage, op)
+		} else {
+			screen.DrawImage(e.Image.Value, op)
+		}
+	}
+}