buf to png

go/gopath/src/fohow.com/apps/controllers/tool_controller/qrcode_controller.go

@@ -4,7 +4,6 @@ import (
 	"context"
 	"fmt"
 	"github.com/chromedp/chromedp"
-	"io/ioutil"
 	"log"
 	"time"
 
@@ -55,6 +54,8 @@ func (self *ToolController) GetHaibao() {
 }
 
 func (self *ToolController) GetHaibaoWithGoogle() {
+	uri := self.GetString("url")
+	//title := self.GetString("title")
 	// 创建上下文
 	ctx, cancel := chromedp.NewContext(context.Background())
 	defer cancel()
@@ -66,20 +67,22 @@ func (self *ToolController) GetHaibaoWithGoogle() {
 	// 运行任务（导航到网页，并捕获屏幕截图）
 	var buf []byte
 	err := chromedp.Run(ctx,
-		chromedp.Navigate(`https://www.baidu.com`), // 替换为你想截图的网页地址
-		chromedp.FullScreenshot(&buf, 90),          // 调整质量参数
+		chromedp.Navigate(uri),            // 替换为你想截图的网页地址
+		chromedp.FullScreenshot(&buf, 90), // 调整质量参数
 	)
 	if err != nil {
 		log.Fatal(err)
 	}
+	//fileName := fmt.Sprintf("%s.png", title)
+	//// 将截图保存到文件
+	//err = ioutil.WriteFile(fileName, buf, 0644)
+	//if err != nil {
+	//	log.Fatal(err)
+	//}
+	self.Ctx.Output.Header("Content-Type", "image/png")
+	self.Ctx.Output.Body(buf)
 
-	// 将截图保存到文件
-	err = ioutil.WriteFile("screenshot.png", buf, 0644)
-	if err != nil {
-		log.Fatal(err)
-	}
-
-	log.Println("Screenshot saved as screenshot.png")
+	//log.Println("Screenshot saved as screenshot.png")
 }
 
 func (self *ToolController) GetQrcode() {