From f9ef6427f99089d79dd396c2ea85ead029c78517 Mon Sep 17 00:00:00 2001
From: Neo Anderson <neo@neo.yt>
Date: Sat, 15 Feb 2020 09:32:44 -0800
Subject: [PATCH] make a working API

---
 main.go      |  7 ++-----
 test/main.go | 35 +++++++++++++++++++++++++++++++++++
 2 files changed, 37 insertions(+), 5 deletions(-)
 create mode 100644 test/main.go

diff --git a/main.go b/main.go
index d7e67f8..e05f4fd 100644
--- a/main.go
+++ b/main.go
@@ -1,12 +1,9 @@
 package karaoke
 
 import (
-	"fmt"
+	"bytes"
 	"image"
 	"image/png"
-	"io/ioutil"
-	"log"
-	"os"
 )
 
 const (
@@ -71,7 +68,7 @@ func ScreenshotAtTime(cdgBytes []byte, timeInSec int) []byte {
 /** @Return raw PNG buffer **/
 func snap() []byte {
 	buf := new(bytes.Buffer)
-	png.Encode(out_file, internal_rgba_context)
+	png.Encode(buf, internal_rgba_context)
 	return buf.Bytes()
 }
 
diff --git a/test/main.go b/test/main.go
new file mode 100644
index 0000000..bc669cb
--- /dev/null
+++ b/test/main.go
@@ -0,0 +1,35 @@
+package main
+
+import (
+	"io/ioutil"
+	"log"
+	"os"
+	".."
+)
+
+func main() {
+	//load data
+	cdg_file_data, err := ioutil.ReadFile("../cdg/SC-SBI-REMIX - Billy Idol - Rebel Yell.cdg")
+	if err != nil {
+		log.Fatal("Couldn't read .cdg file")
+	}
+
+	png := karaoke.ScreenshotAtTime(cdg_file_data, 3)
+
+    file, err := os.OpenFile(
+		"screenshot.png",
+		os.O_WRONLY|os.O_TRUNC|os.O_CREATE,
+		0666,
+    )
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer file.Close()
+
+	bytesWritten, err := file.Write(png)
+	if err != nil {
+	  log.Fatal(err)
+	}
+
+	log.Printf("Wrote %d bytes to %s\n", bytesWritten, "screenshot.png")
+}
\ No newline at end of file