Refactors main.go, removes time.Sleep and some log

henrixapp · Sep 22, 2020 · 44671fd · 44671fd
1 parent a5ff7b6
commit 44671fd
Show file tree

Hide file tree

Showing 2 changed files with 53 additions and 81 deletions.
diff --git a/main.go b/main.go
@@ -10,7 +10,6 @@ import (
 	"io/ioutil"
 	"log"
 	"syscall/js"
-	"time"
 
 	"github.com/nfnt/resize"
 	api "github.com/pdfcpu/pdfcpu/pkg/api"
@@ -83,91 +82,66 @@ func Log(a ...interface{}) {
 
 	if !LogCallback.IsUndefined() {
 		LogCallback.Invoke(js.Null(), fmt.Sprint(a))
-	} else {
-		/*div := doc.Call("createElement", "div")
-		div.Set("textContent", fmt.Sprint(a))
-		body.Call("appendChild", div)*/
 	}
-	time.Sleep(200 * time.Millisecond)
 }
-func main() {
-	onCompress := js.FuncOf(func(this js.Value, args []js.Value) interface{} {
-		LogCallback = args[len(args)-1:][0]
-		Log("called")
-		Log("File size in Bytes:", args[0].Get("length").Int())
-		array := make([]byte, args[0].Get("length").Int())
-		js.CopyBytesToGo(array, args[0])
-		buffi := bytes.NewReader(array)
-		ctx, err := pdfcpu.Read(buffi, pdfcpu.NewDefaultConfiguration())
-		if err != nil {
-			Log("Error while loading file", err)
-		}
-		//Log(ctx)
-		ctx.EnsurePageCount()
-		count := ctx.PageCount
-		pdfcpu.OptimizeXRefTable(ctx)
-		Log("Page count:", count)
-		images := make([]CompressedImage, 0)
-
-		for i := 1; i <= count; i++ {
-			Log("Processing page no:", i)
-			imageObjNrs := ctx.ImageObjNrs(i)
-			Log("Images on page:", len(imageObjNrs))
-			for _, objNr := range imageObjNrs {
-				imageF, _ := ctx.ExtractImage(objNr)
-				Log("Image file name:", imageF.Name)
-				image1, format, err := image.Decode(imageF)
-				Log("Image format:", format)
+
+var onCompress = js.FuncOf(func(this js.Value, args []js.Value) interface{} {
+	LogCallback = args[len(args)-1:][0]
+	Log("called")
+	Log("File size in Bytes:", args[0].Get("length").Int())
+	array := make([]byte, args[0].Get("length").Int())
+	js.CopyBytesToGo(array, args[0])
+	buffi := bytes.NewReader(array)
+	ctx, err := pdfcpu.Read(buffi, pdfcpu.NewDefaultConfiguration())
+	if err != nil {
+		Log("Error while loading file", err)
+	}
+	//Log(ctx)
+	ctx.EnsurePageCount()
+	count := ctx.PageCount
+	pdfcpu.OptimizeXRefTable(ctx)
+	Log("Page count:", count)
+	images := make([]CompressedImage, 0)
+
+	for i := 1; i <= count; i++ {
+		Log("Processing page no:", i)
+		imageObjNrs := ctx.ImageObjNrs(i)
+		Log("Images on page:", len(imageObjNrs))
+		for _, objNr := range imageObjNrs {
+			imageF, _ := ctx.ExtractImage(objNr)
+			image1, format, err := image.Decode(imageF)
+			Log("Image format:", format)
+			if err != nil {
+				Log("e", err)
+			}
+			Log("Image size:", image1.Bounds().Dx(), image1.Bounds().Dy())
+			if image1.Bounds().Dx() > 1000 {
+				Log("Compress this image.....")
+				smaller := resize.Thumbnail(1240, 1754, image1, resize.Lanczos2)
+				var b bytes.Buffer
+				w := bufio.NewWriter(&b)
+				err := jpeg.Encode(w, smaller, nil)
+				images = append(images, CompressedImage{ObjNr: objNr})
 				if err != nil {
-					Log("e", err)
+					Log("Error enconding", err)
 				}
-				Log("Image size:", image1.Bounds().Dx(), image1.Bounds().Dy())
-				if image1.Bounds().Dx() > 1000 {
-					Log("Compress this image.....")
-					smaller := resize.Thumbnail(1240, 1754, image1, resize.Lanczos2)
-					//smaller := resize.Resize(uint(image1.Bounds().Dx()/2.0), 0, image1, resize.Lanczos2)
-					var b bytes.Buffer
-					w := bufio.NewWriter(&b)
-					err := jpeg.Encode(w, smaller, nil)
-					images = append(images, CompressedImage{ObjNr: objNr})
-					if err != nil {
-						Log("Error enconding", err)
-					}
-					ctx.DeleteObject(objNr)
-					buf := new(bytes.Buffer)
-					err = jpeg.Encode(buf, smaller, nil)
-					images[len(images)-1].Ref, _, _, _ = createImageResource(ctx.XRefTable, buf)
-					//imageO.ImageDict.StreamLength
-				}
-				//imageO.ImageDict.Raw =
+				ctx.DeleteObject(objNr)
+				buf := new(bytes.Buffer)
+				err = jpeg.Encode(buf, smaller, nil)
+				images[len(images)-1].Ref, _, _, _ = createImageResource(ctx.XRefTable, buf)
 			}
-
 		}
-		/*fmt.Println(ctx.XRefTable.FindObject(1))
-		root, _ := ctx.XRefTable.FindObject(int(ctx.Root.ObjectNumber))
-		fmt.Println((root.(pdfcpu.Dict)["Pages"]).(pdfcpu.IndirectRef).ObjectNumber)
-		rootPages, _ := ctx.XRefTable.FindObject(int((root.(pdfcpu.Dict)["Pages"]).(pdfcpu.IndirectRef).ObjectNumber))
-		fmt.Println(rootPages.(pdfcpu.Dict).ArrayEntry("Kids"))
-		for _, pageR := range rootPages.(pdfcpu.Dict).ArrayEntry("Kids") {
-			page, _ := ctx.FindObject(int(pageR.(pdfcpu.IndirectRef).ObjectNumber))
-			fmt.Println(page)
-			resources, _ := ctx.FindObject(int(page.(pdfcpu.Dict)["Resources"].(pdfcpu.IndirectRef).ObjectNumber))
-			fmt.Println(resources)
-			n := resources.(pdfcpu.Dict)["XObject"]
-			fmt.Println(reflect.TypeOf(n.(pdfcpu.Dict)["Img1"]))
-		}*/
-		ctx.EnsureVersionForWriting()
-		Log("Write file...")
-		wr := new(bytes.Buffer)
-		api.WriteContext(ctx, wr)
-		js.CopyBytesToJS(args[0], wr.Bytes())
-		args[1].Set("l", len(wr.Bytes()))
-		return len(wr.Bytes())
-	})
+	}
+	ctx.EnsureVersionForWriting()
+	Log("Write file...")
+	wr := new(bytes.Buffer)
+	api.WriteContext(ctx, wr)
+	js.CopyBytesToJS(args[0], wr.Bytes())
+	args[1].Set("l", len(wr.Bytes()))
+	return len(wr.Bytes())
+})
 
+func main() {
 	js.Global().Set("compress", onCompress)
 	<-done
-	if false {
-
-	}
 }
diff --git a/worker.js b/worker.js
@@ -26,6 +26,4 @@ self.addEventListener('message', function (e) {
             time: b - a
         });
     });
-    console.log(e);
-    console.log(self);
 }, false);