在go语言中用JQuery + html2canvas实现拍摄浏览器的屏幕截图示例

Go  /  管理员 发布于 2年前   568

如何对浏览器进行屏幕捕捉，并将捕捉到的屏幕截图发送至网络服务器，保存为PNG文件，然后下载到客户端的电脑中？

解决方式：

下面使用JQuery和JavaScript HTML渲染器-html2canvas可以通过以下方式捕获浏览器内容：

＜div＞或整个页面，并将base64数据传递回Golang web服务器。

作用场景：

这个功能在你想在一定时间内下载一个页面的屏幕截图或用于技术支持的情况下是非常有用的。


示例代码：

package main
 import (
         "bytes"
         "encoding/base64"
         "fmt"
         "image"
         "image/png"
         "io/ioutil"
         "net/http"
         "os"
 )
 
 func SaveFile(w http.ResponseWriter, r *http.Request) {
         if r.Method == "POST" {
                 imgBase64 := r.FormValue("data")
                 //删除 "data:image/png;base64,"
                 imgBase64cleaned := imgBase64[len("data:image/png;base64,"):len(imgBase64)]
                 
                 //解码base64到缓冲区的字节数
                 imgBytes, _ := base64.StdEncoding.DecodeString(imgBase64cleaned)
                 
                 //将[]字节转换为图像以保存在文件中
                 img, _, _ := image.Decode(bytes.NewReader(imgBytes))
                 imgFile, err := os.Create("./screen-capture.png")
                 if err != nil {
                         panic(err)
                 }
                 
                 //保存到你的网络服务器上的文件
                 png.Encode(imgFile, img)
                 fmt.Println("screen-capture.png file saved")
         }
         
         // 注意：由于一些奇怪的原因，这部分必须在r.Method == "POST "之外。
         // 为了使流媒体/下载 直接下载到浏览器的工作
         streamBytes, _ := ioutil.ReadFile("./screen-capture.png")
         //输出到浏览器
         w.Header().Set("Content-Disposition", "attachment; filename='screen-capture.png'")
         w.Header().Set("Content-Type", "image/png")
         w.Header().Set("Content-Transfer-Encoding", "binary")
         w.Header().Set("Content-Description", "File Transfer")
         
         //w.Header().Set("Content-Length", string(len(streamBytes)))
         
         //将产生警告信息：http: invalid Content-Length of "뚴"
         // 把它抽出到浏览器中!
         b := bytes.NewBuffer(streamBytes)
         if _, err := b.WriteTo(w); err != nil {
                 fmt.Fprintf(w, "%s", err)
         }
 }
 func ScreenCapture(w http.ResponseWriter, r *http.Request) {
         html := `<html><body>
         <script type="text/javascript" src="https://code.jquery.com/jquery-1.11.3.js"></script>
         <script type="text/javascript" src="https://socketloop.com/public/tutorial/html2canvas.js"></script>
         <style type="text/css">
         #buttons-area
         {
          margin:30px auto;
          padding:5px;
          border:solid #cdcdcd 1px;
          width:700px;
          text-color:white;
          background:#f4f4f4;
         }
         #target-area
         {
          margin:50px auto;
          padding:15px;
          border:solid #cdcdcd 1px;
          width:700px;
          background:blue;
         }
         button
         {
          text-align:center;
          width:48%;
          height:80px;
          border:1px solid #c3d825;
          font-family:Verdana, Geneva, sans-serif;
          border-radius:3px;
          background:#c3d825;
          color:#355105;
          text-shadow: 0 1pm #ecffba;
          font-weight:bolder;
          font-size:18px;
          border-radius:10px;
         }
         </style>
         <div id="target-area"></div>
         <div id="buttons-area">
         <h1>Take a screenshot with JQuery and Golang example.</h1>
         <button type="button" onclick="captureDiv()">capture div and replace</button>
         <button type="button" onclick="capturePage()">capture entire page & save</button>
         </div>
         </body></html>
         <script type="text/javascript">
           function captureDiv() {
             html2canvas([document.getElementById('buttons-area')], {
                  onrendered: function(canvas)
                  {
                     var imgBase64 = canvas.toDataURL() // already base64
                     var img2html = '<img src=' + imgBase64 + '>'
                     document.getElementById('target-area').innerHTML = img2html;
                  }
             });
           }
           function capturePage() {
               html2canvas(document.body, {
                   onrendered: function(canvas)
                   {
                     var imgBase64 = canvas.toDataURL() // already base64
                     var img2html = '<img src=' + imgBase64 + '>'
                     // 太大，无法放入目标区域的div中。
                     // 因此，我们将其保存到文件中
                     //document.getElementById('target-area').innerHTML = img2html;
                     //发布屏幕捕捉base64图像数据的保存功能
                     $.post("/save", {data: imgBase64}, function () {
                      window.location.href = "save"});
                   }
              });
           }
         </script>`
         w.Write([]byte(html))
 }
 func main() {
         mux := http.NewServeMux()
         mux.HandleFunc("/", ScreenCapture)
         mux.HandleFunc("/save", SaveFile)
         http.ListenAndServe(":8080", mux)
 }

希望这对您有所帮助

参考资料：

https://github.com/niklasvh/html2canvas

  • 上一条：
    人生感悟分享：讲一个大学毕业生到社畜老狗的蜕变心路历程
    下一条：
    在go语言中如何通过名称获得结构字段和值代码示例

    • 昵称：

    邮箱：

    0条评论 (评论内容有缓存机制,请悉知!)
    Top

    Copyright·© 2019 侯体宗版权所有· 粤ICP备20027696号 PHP交流群

    侯体宗的博客