From bae41b7c1eb84738b602ab30d374510a2307ad1d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Hyori=20=ED=9A=A8=EB=A6=AC?= <hyorigo@icloud.com>
Date: Mon, 26 Feb 2024 14:03:56 +0800
Subject: [PATCH] load cache

---
 machine.go | 22 +++++++++++++++++++---
 1 file changed, 19 insertions(+), 3 deletions(-)

diff --git a/machine.go b/machine.go
index d39fcb0d..c2e64501 100644
--- a/machine.go
+++ b/machine.go
@@ -6,12 +6,13 @@
 package starlet
 
 import (
+	"bytes"
 	"fmt"
-	"go.starlark.net/syntax"
 	"io/fs"
 	"sync"
 
 	"go.starlark.net/starlark"
+	"go.starlark.net/syntax"
 )
 
 // Machine is a thread-safe type that wraps Starlark runtime environments. Machine ensures thread safety by using a sync.RWMutex to control access to the environment's state.
@@ -325,12 +326,27 @@ func (m *Machine) DisableGlobalReassign() {
 
 func starlarkExecFile(opts *syntax.FileOptions, thread *starlark.Thread, filename string, src interface{}, predeclared starlark.StringDict) (starlark.StringDict, error) {
 	// Parse, resolve, and compile a Starlark source file.
-	_, mod, err := starlark.SourceProgramOptions(opts, filename, src, predeclared.Has)
+	_, prog, err := starlark.SourceProgramOptions(opts, filename, src, predeclared.Has)
+	if err != nil {
+		return nil, err
+	}
+
+	// Try to save it to the cache
+	buf := new(bytes.Buffer)
+	if err := prog.Write(buf); err != nil {
+		return nil, err
+	}
+	bs := buf.Bytes()
+	//cv := starlark.CompilerVersion
+
+	// Reload as a new program
+	np, err := starlark.CompiledProgram(bytes.NewReader(bs))
 	if err != nil {
 		return nil, err
 	}
+	prog = np
 
-	g, err := mod.Init(thread, predeclared)
+	g, err := prog.Init(thread, predeclared)
 	g.Freeze()
 	return g, err
 }