From a42c9b27e793cd65e9a2c7f3dc15fc9c631cfe05 Mon Sep 17 00:00:00 2001
From: Arsen Musayelyan <arsen@arsenm.dev>
Date: Sun, 2 Oct 2022 20:09:12 -0700
Subject: [PATCH] Add version() function

---
 build.go | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/build.go b/build.go
index 2cadad1..923e65f 100644
--- a/build.go
+++ b/build.go
@@ -19,6 +19,7 @@
 package main
 
 import (
+	"bytes"
 	"context"
 	"encoding/hex"
 	"io"
@@ -233,6 +234,26 @@ func buildPackage(ctx context.Context, script string, mgr manager.Manager) ([]st
 		}
 	}
 
+	fn, ok = dec.GetFunc("version")
+	if ok {
+		log.Info("Executing version()").Send()
+
+		buf := &bytes.Buffer{}
+
+		err = fn(
+			ctx,
+			interp.Dir(srcdir),
+			interp.StdIO(os.Stdin, buf, os.Stderr),
+		)
+		if err != nil {
+			return nil, nil, err
+		}
+
+		vars.Version = strings.TrimSpace(buf.String())
+
+		log.Info("Updating version").Str("new", vars.Version).Send()
+	}
+
 	fn, ok = dec.GetFunc("build")
 	if ok {
 		log.Info("Executing build()").Send()