将 QR 推送到客户端

huangapple
go评论121阅读模式
英文:

Pushing QR to client

问题

尝试将登录界面制作成类似WhatsApp的样式，其中二维码被推送到客户端进行扫描，所以我编写了以下服务器代码：

  1. package main
  3. import (
  4. 	"context"
  5. 	"fmt"
  6. 	"net/http"
  7. 	"os"
  8. 	"os/signal"
  9. 	"syscall"
  11. 	_ "github.com/mattn/go-sqlite3"
  12. 	"github.com/skip2/go-qrcode"
  13. 	"google.golang.org/protobuf/proto"
  15. 	"go.mau.fi/whatsmeow"
  16. 	"go.mau.fi/whatsmeow/store"
  17. 	"go.mau.fi/whatsmeow/store/sqlstore"
  18. 	"go.mau.fi/whatsmeow/types/events"
  19. 	waLog "go.mau.fi/whatsmeow/util/log"
  20. )
  22. func eventHandler(evt interface{}) {
  23. 	switch v := evt.(type) {
  24. 	case *events.Message:
  25. 		fmt.Println("Received a message!", v.Message.GetConversation())
  26. 	}
  27. }
  29. func main() {
  30. 	passer := &DataPasser{logs: make(chan string)}
  32. 	http.HandleFunc("/", passer.handleHello)
  33. 	go http.ListenAndServe(":9999", nil)
  35. 	store.DeviceProps.Os = proto.String("WhatsApp GO")
  36. 	dbLog := waLog.Stdout("Database", "DEBUG", true)
  37. 	// 确保添加适当的数据库连接器导入，例如 github.com/mattn/go-sqlite3 用于 SQLite
  38. 	container, err := sqlstore.New("sqlite3", "file:datastore.db?_foreign_keys=on", dbLog)
  39. 	if err != nil {
  40. 		panic(err)
  41. 	}
  42. 	// 如果您想要多个会话，请记住它们的 JID 并使用 .GetDevice(jid) 或 .GetAllDevices() 代替。
  43. 	deviceStore, err := container.GetFirstDevice()
  44. 	if err != nil {
  45. 		panic(err)
  46. 	}
  47. 	clientLog := waLog.Stdout("Client", "DEBUG", true)
  48. 	client := whatsmeow.NewClient(deviceStore, clientLog)
  49. 	client.AddEventHandler(eventHandler)
  51. 	if client.Store.ID == nil {
  52. 		// 未存储 ID，新登录
  53. 		qrChan, _ := client.GetQRChannel(context.Background())
  54. 		err = client.Connect()
  55. 		if err != nil {
  56. 			panic(err)
  57. 		}
  59. 		for evt := range qrChan {
  60. 			switch evt.Event {
  61. 			case "success":
  62. 				{
  63. 					passer.logs <- "success"
  64. 					fmt.Println("Login event: success")
  65. 				}
  66. 			case "timeout":
  67. 				{
  68. 					passer.logs <- "timeout"
  69. 					fmt.Println("Login event: timeout")
  70. 				}
  71. 			case "code":
  72. 				{
  73. 					passer.logs <- "new code"
  74. 					fmt.Println("new code recieved")
  75. 					img, err := qrcode.Encode(evt.Code, qrcode.Medium, 200) // evt.Code
  76. 					if err != nil {
  77. 						fmt.Println("error when write qrImage", err.Error())
  78. 					}
  79. 					passer.logs <- string(img)
  80. 				}
  81. 			}
  82. 		}
  83. 	} else {
  84. 		// 已登录，只需连接
  85. 		passer.logs <- "Already logged"
  86. 		fmt.Println("Already logged")
  87. 		err = client.Connect()
  88. 		if err != nil {
  89. 			panic(err)
  90. 		}
  91. 	}
  93. 	// 监听 Ctrl+C（您也可以执行其他阻止程序退出的操作）
  94. 	c := make(chan os.Signal, 1)
  95. 	signal.Notify(c, os.Interrupt, syscall.SIGTERM)
  96. 	<-c
  98. 	client.Disconnect()
  99. }

使用以下 API：

  1. package main
  3. import (
  4. 	"bytes"
  5. 	"fmt"
  6. 	"log"
  7. 	"net/http"
  8. 	"strconv"
  9. 	"sync"
  10. )
  12. var (
  13. 	mux sync.Mutex
  14. )
  16. type Result struct {
  17. 	ResultType, Result string
  18. }
  20. type DataPasser struct {
  21. 	logs chan string
  22. }
  24. func (p *DataPasser) handleHello(w http.ResponseWriter, r *http.Request) {
  25. 	setupCORS(&w, r)
  26. 	w.Header().Set("Content-Type", "text/event-stream")
  27. 	flusher, ok := w.(http.Flusher)
  28. 	if !ok {
  29. 		http.Error(w, "Internal error", 500)
  30. 		return
  31. 	}
  32. 	flusher.Flush()
  33. 	done := r.Context().Done()
  34. 	defer fmt.Println("EXIT")
  35. 	for {
  36. 		select {
  37. 		case <-done:
  38. 			// 客户端断开连接
  39. 			return
  40. 		case m := <-p.logs:
  41. 			fmt.Println(m)
  42. 			if m == "new code" || m == "We are logging in" || m == "Already logged" || m == "timeout" || m == "success" {
  43. 				if _, err := fmt.Fprintf(w, "data: %s\n\n", m); err != nil {
  44. 					// 写入连接失败。后续写入可能会失败。
  45. 					return
  46. 				}
  47. 				flusher.Flush()
  48. 			} else {
  49. 				mux.Lock()
  51. 				buffer := bytes.NewBuffer([]byte(m))
  53. 				/*	img, _, err := image.Decode(buffer)
  54. 					if err != nil {
  55. 						fmt.Println("err: ", err)
  56. 					}
  58. 					if err := jpeg.Encode(buffer, img, nil); err != nil {
  59. 						log.Println("unable to encode image.")
  60. 					} */
  62. 				w.Header().Set("Content-Type", "image/jpeg")
  63. 				w.Header().Set("Content-Length", strconv.Itoa(len(buffer.Bytes())))
  64. 				if _, err := w.Write(buffer.Bytes()); err != nil {
  65. 					log.Println("unable to write image.")
  66. 				}
  67. 				mux.Unlock()
  68. 				flusher.Flush()
  69. 			}
  70. 		}
  71. 	}
  72. }
  74. func setupCORS(w *http.ResponseWriter, req *http.Request) {
  75. 	(*w).Header().Set("Cache-Control", "no-cache")
  76. 	(*w).Header().Set("Connection", "keep-alive")
  77. 	(*w).Header().Set("Access-Control-Allow-Origin", "*")
  78. 	(*w).Header().Set("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE")
  79. 	(*w).Header().Set("Access-Control-Allow-Headers", "Accept, Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token, Authorization")
  80. }

但在客户端中无法读取它，我尝试了以下内容，但迷失了：

  1. <html>
  2. <head></head>
  3. <body>
  4. note: <span id="content"></span><br>
  5. <img id="photo" style="display: block;-webkit-user-select: none;">
  6. </body>
  7. <script>
  8. /*  (function(){
  9. document.getElementById("content").innerHTML='<object type="text/html" data="http://localhost:9999/"></object>';
  10. })(); 
  11. */
  12. const myRequest = new Request('http://127.0.0.1:9999/', {
  13. method: 'GET',
  14. headers: new Headers(),
  15. type: "arraybuffer",
  16. mode: 'cors',
  17. cache: 'default',
  18. });  
  19. var source = new EventSource("http://127.0.0.1:9999/");
  20. source.onmessage = function (event) {
  21. console.log(event)
  22. var counter = event.data; // JSON.parse(event.data);
  23. document.getElementById("content").innerHTML = counter;
  24. }
  25. fetch(myRequest).then(response => {
  26. console.log(response)
  27. console.log(response.headers)
  28. const contentType = response.headers.get("content-type");
  29. if (contentType && contentType.indexOf("application/json") !== -1) {
  30. return response.json().then(data => {
  31. var obj = JSON.parse(str);
  32. console.log(obj)
  33. // Process your data as a JavaScript object
  34. });
  35. }  if (contentType && contentType.indexOf("image/jpeg") !== -1) {
  36. console.log("Image received")
  37. return response.blob().then(data => {
  38. var reader = new FileReader();
  39. reader.readAsDataURL(blob); 
  40. reader.onloadend = function() {
  41. var imageUrl = reader.result;                
  42. var img = document.querySelector("#photo");
  43. img.src = imageUrl;
  44. }
  45. });
  46. } else if (contentType && contentType.indexOf("text/event-stream") !== -1) {
  47. return response.text().then(text => {
  48. console.log(text)
  49. var source = new EventSource("http://localhost:9999/");
  50. source.onmessage = function (event) {
  51. var response = event.data // JSON.parse(event.data);
  52. document.getElementById("content").innerHTML = counter;
  53. } 
  54. // Process your text as a String
  55. });
  56. } else if (contentType && contentType.indexOf("text/html") !== -1) {
  57. return response.text().then(text => {
  58. console.log(text)
  59. var source = new EventSource("http://localhost:9999/");
  60. source.onmessage = function (event) {
  61. var response = event.data // JSON.parse(event.data);
  62. document.getElementById("content").innerHTML = counter;
  63. } 
  64. // Process your text as a String
  65. });
  66. } 
  67. });
  68. </script>
  69. </html>

如果我运行并查看显示的页面，我会看到：

将 QR 推送到客户端

有什么帮助吗？

英文:

Trying to make login screen as WhatsApp, where the QR is pushed to the client for scanning, so I wrote the below server code:

  1. package main
  3. import (
  4. 	&quot;context&quot;
  5. 	&quot;fmt&quot;
  6. 	&quot;net/http&quot;
  7. 	&quot;os&quot;
  8. 	&quot;os/signal&quot;
  9. 	&quot;syscall&quot;
  11. 	_ &quot;github.com/mattn/go-sqlite3&quot;
  12. 	&quot;github.com/skip2/go-qrcode&quot;
  13. 	&quot;google.golang.org/protobuf/proto&quot;
  15. 	&quot;go.mau.fi/whatsmeow&quot;
  16. 	&quot;go.mau.fi/whatsmeow/store&quot;
  17. 	&quot;go.mau.fi/whatsmeow/store/sqlstore&quot;
  18. 	&quot;go.mau.fi/whatsmeow/types/events&quot;
  19. 	waLog &quot;go.mau.fi/whatsmeow/util/log&quot;
  20. )
  22. func eventHandler(evt interface{}) {
  23. 	switch v := evt.(type) {
  24. 	case *events.Message:
  25. 		fmt.Println(&quot;Received a message!&quot;, v.Message.GetConversation())
  26. 	}
  27. }
  29. func main() {
  30. 	passer := &amp;DataPasser{logs: make(chan string)}
  32. 	http.HandleFunc(&quot;/&quot;, passer.handleHello)
  33. 	go http.ListenAndServe(&quot;:9999&quot;, nil)
  35. 	store.DeviceProps.Os = proto.String(&quot;WhatsApp GO&quot;)
  36. 	dbLog := waLog.Stdout(&quot;Database&quot;, &quot;DEBUG&quot;, true)
  37. 	// Make sure you add appropriate DB connector imports, e.g. github.com/mattn/go-sqlite3 for SQLite
  38. 	container, err := sqlstore.New(&quot;sqlite3&quot;, &quot;file:datastore.db?_foreign_keys=on&quot;, dbLog)
  39. 	if err != nil {
  40. 		panic(err)
  41. 	}
  42. 	// If you want multiple sessions, remember their JIDs and use .GetDevice(jid) or .GetAllDevices() instead.
  43. 	deviceStore, err := container.GetFirstDevice()
  44. 	if err != nil {
  45. 		panic(err)
  46. 	}
  47. 	clientLog := waLog.Stdout(&quot;Client&quot;, &quot;DEBUG&quot;, true)
  48. 	client := whatsmeow.NewClient(deviceStore, clientLog)
  49. 	client.AddEventHandler(eventHandler)
  51. 	if client.Store.ID == nil {
  52. 		// No ID stored, new login
  53. 		qrChan, _ := client.GetQRChannel(context.Background())
  54. 		err = client.Connect()
  55. 		if err != nil {
  56. 			panic(err)
  57. 		}
  59. 		for evt := range qrChan {
  60. 			switch evt.Event {
  61. 			case &quot;success&quot;:
  62. 				{
  63. 					passer.logs &lt;- &quot;success&quot;
  64. 					fmt.Println(&quot;Login event: success&quot;)
  65. 				}
  66. 			case &quot;timeout&quot;:
  67. 				{
  68. 					passer.logs &lt;- &quot;timeout&quot;
  69. 					fmt.Println(&quot;Login event: timeout&quot;)
  70. 				}
  71. 			case &quot;code&quot;:
  72. 				{
  73. 					passer.logs &lt;- &quot;new code&quot;
  74. 					fmt.Println(&quot;new code recieved&quot;)
  75. 					img, err := qrcode.Encode(evt.Code, qrcode.Medium, 200) // evt.Code
  76. 					if err != nil {
  77. 						fmt.Println(&quot;error when write qrImage&quot;, err.Error())
  78. 					}
  79. 					passer.logs &lt;- string(img)
  80. 				}
  81. 			}
  82. 		}
  83. 	} else {
  84. 		// Already logged in, just connect
  85. 		passer.logs &lt;- &quot;Already logged&quot;
  86. 		fmt.Println(&quot;Already logged&quot;)
  87. 		err = client.Connect()
  88. 		if err != nil {
  89. 			panic(err)
  90. 		}
  91. 	}
  93. 	// Listen to Ctrl+C (you can also do something else that prevents the program from exiting)
  94. 	c := make(chan os.Signal, 1)
  95. 	signal.Notify(c, os.Interrupt, syscall.SIGTERM)
  96. 	&lt;-c
  98. 	client.Disconnect()
  99. }

With the below api:

  1. package main
  3. import (
  4. 	&quot;bytes&quot;
  5. 	&quot;fmt&quot;
  6. 	&quot;log&quot;
  7. 	&quot;net/http&quot;
  8. 	&quot;strconv&quot;
  9. 	&quot;sync&quot;
  10. )
  12. var (
  13. 	mux sync.Mutex
  14. )
  16. type Result struct {
  17. 	ResultType, Result string
  18. }
  20. type DataPasser struct {
  21. 	logs chan string
  22. }
  24. func (p *DataPasser) handleHello(w http.ResponseWriter, r *http.Request) {
  25. 	setupCORS(&amp;w, r)
  26. 	w.Header().Set(&quot;Content-Type&quot;, &quot;text/event-stream&quot;)
  27. 	flusher, ok := w.(http.Flusher)
  28. 	if !ok {
  29. 		http.Error(w, &quot;Internal error&quot;, 500)
  30. 		return
  31. 	}
  32. 	flusher.Flush()
  33. 	done := r.Context().Done()
  34. 	defer fmt.Println(&quot;EXIT&quot;)
  35. 	for {
  36. 		select {
  37. 		case &lt;-done:
  38. 			// the client disconnected
  39. 			return
  40. 		case m := &lt;-p.logs:
  41. 			fmt.Println(m)
  42. 			if m == &quot;new code&quot; || m == &quot;We are logging in&quot; || m == &quot;Already logged&quot; || m == &quot;timeout&quot; || m == &quot;success&quot; {
  43. 				if _, err := fmt.Fprintf(w, &quot;data: %s\n\n&quot;, m); err != nil {
  44. 					// Write to connection failed. Subsequent writes will probably fail.
  45. 					return
  46. 				}
  47. 				flusher.Flush()
  48. 			} else {
  49. 				mux.Lock()
  51. 				buffer := bytes.NewBuffer([]byte(m))
  53. 				/*	img, _, err := image.Decode(buffer)
  54. 					if err != nil {
  55. 						fmt.Println(&quot;err: &quot;, err)
  56. 					}
  58. 					if err := jpeg.Encode(buffer, img, nil); err != nil {
  59. 						log.Println(&quot;unable to encode image.&quot;)
  60. 					} */
  62. 				w.Header().Set(&quot;Content-Type&quot;, &quot;image/jpeg&quot;)
  63. 				w.Header().Set(&quot;Content-Length&quot;, strconv.Itoa(len(buffer.Bytes())))
  64. 				if _, err := w.Write(buffer.Bytes()); err != nil {
  65. 					log.Println(&quot;unable to write image.&quot;)
  66. 				}
  67. 				mux.Unlock()
  68. 				flusher.Flush()
  69. 			}
  70. 		}
  71. 	}
  72. }
  74. func setupCORS(w *http.ResponseWriter, req *http.Request) {
  75. 	(*w).Header().Set(&quot;Cache-Control&quot;, &quot;no-cache&quot;)
  76. 	(*w).Header().Set(&quot;Connection&quot;, &quot;keep-alive&quot;)
  77. 	(*w).Header().Set(&quot;Access-Control-Allow-Origin&quot;, &quot;*&quot;)
  78. 	(*w).Header().Set(&quot;Access-Control-Allow-Methods&quot;, &quot;POST, GET, OPTIONS, PUT, DELETE&quot;)
  79. 	(*w).Header().Set(&quot;Access-Control-Allow-Headers&quot;, &quot;Accept, Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token, Authorization&quot;)
  80. }

But stuck to read it in the client, I tried something as below, but got lost:

  1. &lt;html&gt;
  2. &lt;head&gt;&lt;/head&gt;
  3. &lt;body&gt;
  4. note: &lt;span id=&quot;content&quot;&gt;&lt;/span&gt;&lt;br&gt;
  5. &lt;img id=&quot;photo&quot; style=&quot;display: block;-webkit-user-select: none;&quot;&gt;
  6. &lt;/body&gt;
  7. &lt;script&gt;
  8. /*  (function(){
  9. document.getElementById(&quot;content&quot;).innerHTML=&#39;&lt;object type=&quot;text/html&quot; data=&quot;http://localhost:9999/&quot; &gt;&lt;/object&gt;&#39;;
  10. })(); 
  11. */
  12. const myRequest = new Request(&#39;http://127.0.0.1:9999/&#39;, {
  13. method: &#39;GET&#39;,
  14. headers: new Headers(),
  15. type: &quot;arraybuffer&quot;,
  16. mode: &#39;cors&#39;,
  17. cache: &#39;default&#39;,
  18. });  
  19. var source = new EventSource(&quot;http://127.0.0.1:9999/&quot;);
  20. source.onmessage = function (event) {
  21. console.log(event)
  22. var counter = event.data; // JSON.parse(event.data);
  23. document.getElementById(&quot;content&quot;).innerHTML = counter;
  24. }
  25. fetch(myRequest).then(response =&gt; {
  26. console.log(response)
  27. console.log(response.headers)
  28. const contentType = response.headers.get(&quot;content-type&quot;);
  29. if (contentType &amp;&amp; contentType.indexOf(&quot;application/json&quot;) !== -1) {
  30. return response.json().then(data =&gt; {
  31. var obj = JSON.parse(str);
  32. console.log(obj)
  33. // Process your data as a JavaScript object
  34. });
  35. }  if (contentType &amp;&amp; contentType.indexOf(&quot;image/jpeg&quot;) !== -1) {
  36. console.log(&quot;Image received&quot;)
  37. return response.blob().then(data =&gt; {
  38. var reader = new FileReader();
  39. reader.readAsDataURL(blob); 
  40. reader.onloadend = function() {
  41. var imageUrl = reader.result;                
  42. var img = document.querySelector(&quot;#photo&quot;);
  43. img.src = imageUrl;
  44. }
  45. });
  46. } else if (contentType &amp;&amp; contentType.indexOf(&quot;text/event-stream&quot;) !== -1) {
  47. return response.text().then(text =&gt; {
  48. console.log(text)
  49. var source = new EventSource(&quot;http://localhost:9999/&quot;);
  50. source.onmessage = function (event) {
  51. var response = event.data // JSON.parse(event.data);
  52. document.getElementById(&quot;content&quot;).innerHTML = counter;
  53. } 
  54. // Process your text as a String
  55. });
  56. } else if (contentType &amp;&amp; contentType.indexOf(&quot;text/html&quot;) !== -1) {
  57. return response.text().then(text =&gt; {
  58. console.log(text)
  59. var source = new EventSource(&quot;http://localhost:9999/&quot;);
  60. source.onmessage = function (event) {
  61. var response = event.data // JSON.parse(event.data);
  62. document.getElementById(&quot;content&quot;).innerHTML = counter;
  63. } 
  64. // Process your text as a String
  65. });
  66. } 
  67. });
  68. &lt;/script&gt;
  69. &lt;/html&gt;

If I run and see the displayed page, I see:

将 QR 推送到客户端

Any help?

答案1

得分: 0

我通过将QR字符串推送到服务器，并使用qrcodejs将其转换为QR码来解决了这个问题。如果有兴趣，以下是完整的代码：

  1. // main.go
  2. package main
  4. import (
  5. 	"context"
  6. 	"fmt"
  7. 	"net/http"
  8. 	"os"
  9. 	"os/signal"
  10. 	"syscall"
  12. 	_ "github.com/mattn/go-sqlite3"
  13. 	"google.golang.org/protobuf/proto"
  15. 	"go.mau.fi/whatsmeow"
  16. 	"go.mau.fi/whatsmeow/store"
  17. 	"go.mau.fi/whatsmeow/store/sqlstore"
  18. 	"go.mau.fi/whatsmeow/types/events"
  19. 	waLog "go.mau.fi/whatsmeow/util/log"
  20. )
  22. func eventHandler(evt interface{}) {
  23. 	switch v := evt.(type) {
  24. 	case *events.Message:
  25. 		fmt.Println("Received a message!", v.Message.GetConversation())
  26. 	}
  27. }
  29. func main() {
  30. 	passer := &DataPasser{logs: make(chan string)}
  32. 	http.HandleFunc("/sse/dashboard", passer.handleHello)
  33. 	go http.ListenAndServe(":1234", nil)
  34. 	/*
  35. 		ticker := time.NewTicker(500 * time.Millisecond)
  36. 		defer ticker.Stop()
  37. 		done := make(chan bool)
  38. 		go func() {
  39. 			for {
  40. 				select {
  41. 				case <-done:
  42. 					return
  43. 				case <-ticker.C:
  44. 					//	fmt.Println("Tick at", t)
  45. 					// passer.logs <- buffer.String()
  46. 				}
  47. 			}
  48. 		}()
  49. 	*/
  50. 	store.DeviceProps.Os = proto.String("WhatsApp GO")
  51. 	dbLog := waLog.Stdout("Database", "DEBUG", true)
  52. 	// Make sure you add appropriate DB connector imports, e.g. github.com/mattn/go-sqlite3 for SQLite
  53. 	container, err := sqlstore.New("sqlite3", "file:datastore.db?_foreign_keys=on", dbLog)
  54. 	if err != nil {
  55. 		panic(err)
  56. 	}
  57. 	// If you want multiple sessions, remember their JIDs and use .GetDevice(jid) or .GetAllDevices() instead.
  58. 	deviceStore, err := container.GetFirstDevice()
  59. 	if err != nil {
  60. 		panic(err)
  61. 	}
  62. 	clientLog := waLog.Stdout("Client", "DEBUG", true)
  63. 	client := whatsmeow.NewClient(deviceStore, clientLog)
  64. 	client.AddEventHandler(eventHandler)
  66. 	if client.Store.ID == nil {
  67. 		// No ID stored, new login
  68. 		qrChan, _ := client.GetQRChannel(context.Background())
  69. 		err = client.Connect()
  70. 		if err != nil {
  71. 			panic(err)
  72. 		}
  74. 		for evt := range qrChan {
  75. 			switch evt.Event {
  76. 			case "success":
  77. 				{
  78. 					passer.logs <- "success"
  79. 					fmt.Println("Login event: success")
  80. 				}
  81. 			case "timeout":
  82. 				{
  83. 					passer.logs <- "timeout"
  84. 					fmt.Println("Login event: timeout")
  85. 				}
  86. 			case "code":
  87. 				{
  88. 					fmt.Println("new code recieved")
  89. 					fmt.Println(evt.Code)
  90. 					passer.logs <- evt.Code
  91. 				}
  92. 			}
  93. 		}
  94. 	} else {
  95. 		// Already logged in, just connect
  96. 		passer.logs <- "Already logged"
  97. 		fmt.Println("Already logged")
  98. 		err = client.Connect()
  99. 		if err != nil {
  100. 			panic(err)
  101. 		}
  102. 	}
  104. 	// Listen to Ctrl+C (you can also do something else that prevents the program from exiting)
  105. 	c := make(chan os.Signal, 1)
  106. 	signal.Notify(c, os.Interrupt, syscall.SIGTERM)
  107. 	<-c
  109. 	client.Disconnect()
  110. }

和

  1. // api.go
  2. package main
  4. import (
  5. 	"fmt"
  6. 	"net/http"
  7. 	"sync"
  8. )
  10. var (
  11. 	mux sync.Mutex
  12. )
  14. type DataPasser struct {
  15. 	logs chan string
  16. }
  18. func (p *DataPasser) handleHello(w http.ResponseWriter, r *http.Request) {
  19. 	fmt.Println("from here")
  20. 	flusher, ok := w.(http.Flusher)
  21. 	if !ok {
  22. 		http.Error(w, "Internal error", 500)
  23. 		return
  24. 	}
  25. 	w.Header().Set("Content-Type", "text/event-stream")
  26. 	w.Header().Set("Cache-Control", "no-cache")
  27. 	w.Header().Set("Connection", "keep-alive")
  28. 	setupCORS(&w, r)
  30. 	for {
  31. 		select {
  32. 		case c := <-p.logs:
  33. 			fmt.Println("recieved")
  34. 			mux.Lock()
  35. 			//counter++
  36. 			//c := counter
  37. 			mux.Unlock()
  38. 			fmt.Fprintf(w, "data: %v\n\n", c)
  39. 			flusher.Flush()
  40. 		case <-r.Context().Done():
  41. 			fmt.Println("Connection closed")
  42. 			return
  43. 		}
  44. 	}
  45. }
  47. func setupCORS(w *http.ResponseWriter, req *http.Request) {
  48. 	(*w).Header().Set("Cache-Control", "no-cache")
  49. 	(*w).Header().Set("Access-Control-Allow-Origin", "*")
  50. 	(*w).Header().Set("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE")
  51. 	(*w).Header().Set("Access-Control-Allow-Headers", "Accept, Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token, Authorization")
  52. }

和客户端代码：

  1. <html lang="en">
  2. <head>
  3. <meta charset="utf-8">
  4. <meta name="viewport" content="width=device-width, initial-scale=1">
  5. <script src="https://cdn.rawgit.com/davidshimjs/qrcodejs/gh-pages/qrcode.min.js"></script>
  6. </head>
  7. <body>
  8. <h5>Server message/Code: <span id="message"></span></h5>
  9. <div id="qr"></div>
  10. <script>
  11. var source = new EventSource("http://localhost:1234/sse/dashboard");
  12. source.onmessage = function (event) {
  13. var message = event.data
  14. document.querySelector('#message').innerHTML = message;
  15. if (new String(message).valueOf() == "success" || new String(message).valueOf() == "timeout"
  16. || new String(message).valueOf() == "Already logged") {
  17. document.querySelector('#qr').innerHTML = "";
  18. } else {
  19. var qrcode = new QRCode("qr", {
  20. text: message,
  21. width: 128,
  22. height: 128,
  23. colorDark : "#000000",
  24. colorLight : "#ffffff",
  25. correctLevel : QRCode.CorrectLevel.M
  26. });
  27. }
  28. }
  29. /*
  30. var qrcode = new QRCode(
  31. "qr",
  32. [
  33. "BEGIN:VCARD",
  34. "VERSION:2.1",
  35. "N:Doe;John;;Dr;",
  36. "FN:Dr. John Doe",
  37. "EMAIL:johndoe@hotmail.com",
  38. "TEL;TYPE=cell:(123) 555-5832",
  39. "END:VCARD"
  40. ].join("\r\n")
  41. ); */
  42. </script>
  43. </body>
  44. </html>
英文:

I solved it buy pushing the QR string t o the server, and convert it to QR using qrcodejs. My full code if any is interested is:

  1. // main.go
  2. package main
  4. import (
  5. 	&quot;context&quot;
  6. 	&quot;fmt&quot;
  7. 	&quot;net/http&quot;
  8. 	&quot;os&quot;
  9. 	&quot;os/signal&quot;
  10. 	&quot;syscall&quot;
  12. 	_ &quot;github.com/mattn/go-sqlite3&quot;
  13. 	&quot;google.golang.org/protobuf/proto&quot;
  15. 	&quot;go.mau.fi/whatsmeow&quot;
  16. 	&quot;go.mau.fi/whatsmeow/store&quot;
  17. 	&quot;go.mau.fi/whatsmeow/store/sqlstore&quot;
  18. 	&quot;go.mau.fi/whatsmeow/types/events&quot;
  19. 	waLog &quot;go.mau.fi/whatsmeow/util/log&quot;
  20. )
  22. func eventHandler(evt interface{}) {
  23. 	switch v := evt.(type) {
  24. 	case *events.Message:
  25. 		fmt.Println(&quot;Received a message!&quot;, v.Message.GetConversation())
  26. 	}
  27. }
  29. func main() {
  30. 	passer := &amp;DataPasser{logs: make(chan string)}
  32. 	http.HandleFunc(&quot;/sse/dashboard&quot;, passer.handleHello)
  33. 	go http.ListenAndServe(&quot;:1234&quot;, nil)
  34. 	/*
  35. 		ticker := time.NewTicker(500 * time.Millisecond)
  36. 		defer ticker.Stop()
  37. 		done := make(chan bool)
  38. 		go func() {
  39. 			for {
  40. 				select {
  41. 				case &lt;-done:
  42. 					return
  43. 				case &lt;-ticker.C:
  44. 					//	fmt.Println(&quot;Tick at&quot;, t)
  45. 					// passer.logs &lt;- buffer.String()
  46. 				}
  47. 			}
  48. 		}()
  49. 	*/
  50. 	store.DeviceProps.Os = proto.String(&quot;WhatsApp GO&quot;)
  51. 	dbLog := waLog.Stdout(&quot;Database&quot;, &quot;DEBUG&quot;, true)
  52. 	// Make sure you add appropriate DB connector imports, e.g. github.com/mattn/go-sqlite3 for SQLite
  53. 	container, err := sqlstore.New(&quot;sqlite3&quot;, &quot;file:datastore.db?_foreign_keys=on&quot;, dbLog)
  54. 	if err != nil {
  55. 		panic(err)
  56. 	}
  57. 	// If you want multiple sessions, remember their JIDs and use .GetDevice(jid) or .GetAllDevices() instead.
  58. 	deviceStore, err := container.GetFirstDevice()
  59. 	if err != nil {
  60. 		panic(err)
  61. 	}
  62. 	clientLog := waLog.Stdout(&quot;Client&quot;, &quot;DEBUG&quot;, true)
  63. 	client := whatsmeow.NewClient(deviceStore, clientLog)
  64. 	client.AddEventHandler(eventHandler)
  66. 	if client.Store.ID == nil {
  67. 		// No ID stored, new login
  68. 		qrChan, _ := client.GetQRChannel(context.Background())
  69. 		err = client.Connect()
  70. 		if err != nil {
  71. 			panic(err)
  72. 		}
  74. 		for evt := range qrChan {
  75. 			switch evt.Event {
  76. 			case &quot;success&quot;:
  77. 				{
  78. 					passer.logs &lt;- &quot;success&quot;
  79. 					fmt.Println(&quot;Login event: success&quot;)
  80. 				}
  81. 			case &quot;timeout&quot;:
  82. 				{
  83. 					passer.logs &lt;- &quot;timeout&quot;
  84. 					fmt.Println(&quot;Login event: timeout&quot;)
  85. 				}
  86. 			case &quot;code&quot;:
  87. 				{
  88. 					fmt.Println(&quot;new code recieved&quot;)
  89. 					fmt.Println(evt.Code)
  90. 					passer.logs &lt;- evt.Code
  91. 				}
  92. 			}
  93. 		}
  94. 	} else {
  95. 		// Already logged in, just connect
  96. 		passer.logs &lt;- &quot;Already logged&quot;
  97. 		fmt.Println(&quot;Already logged&quot;)
  98. 		err = client.Connect()
  99. 		if err != nil {
  100. 			panic(err)
  101. 		}
  102. 	}
  104. 	// Listen to Ctrl+C (you can also do something else that prevents the program from exiting)
  105. 	c := make(chan os.Signal, 1)
  106. 	signal.Notify(c, os.Interrupt, syscall.SIGTERM)
  107. 	&lt;-c
  109. 	client.Disconnect()
  110. }

And

  1. // api.go
  2. package main
  4. import (
  5. 	&quot;fmt&quot;
  6. 	&quot;net/http&quot;
  7. 	&quot;sync&quot;
  8. )
  10. var (
  11. 	mux sync.Mutex
  12. )
  14. type DataPasser struct {
  15. 	logs chan string
  16. }
  18. func (p *DataPasser) handleHello(w http.ResponseWriter, r *http.Request) {
  19. 	fmt.Println(&quot;from here&quot;)
  20. 	flusher, ok := w.(http.Flusher)
  21. 	if !ok {
  22. 		http.Error(w, &quot;Internal error&quot;, 500)
  23. 		return
  24. 	}
  25. 	w.Header().Set(&quot;Content-Type&quot;, &quot;text/event-stream&quot;)
  26. 	w.Header().Set(&quot;Cache-Control&quot;, &quot;no-cache&quot;)
  27. 	w.Header().Set(&quot;Connection&quot;, &quot;keep-alive&quot;)
  28. 	setupCORS(&amp;w, r)
  30. 	for {
  31. 		select {
  32. 		case c := &lt;-p.logs:
  33. 			fmt.Println(&quot;recieved&quot;)
  34. 			mux.Lock()
  35. 			//counter++
  36. 			//c := counter
  37. 			mux.Unlock()
  38. 			fmt.Fprintf(w, &quot;data: %v\n\n&quot;, c)
  39. 			flusher.Flush()
  40. 		case &lt;-r.Context().Done():
  41. 			fmt.Println(&quot;Connection closed&quot;)
  42. 			return
  43. 		}
  44. 	}
  45. }
  47. func setupCORS(w *http.ResponseWriter, req *http.Request) {
  48. 	(*w).Header().Set(&quot;Cache-Control&quot;, &quot;no-cache&quot;)
  49. 	(*w).Header().Set(&quot;Access-Control-Allow-Origin&quot;, &quot;*&quot;)
  50. 	(*w).Header().Set(&quot;Access-Control-Allow-Methods&quot;, &quot;POST, GET, OPTIONS, PUT, DELETE&quot;)
  51. 	(*w).Header().Set(&quot;Access-Control-Allow-Headers&quot;, &quot;Accept, Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token, Authorization&quot;)
  52. }

And the client side is:

  1. &lt;html lang=&quot;en&quot;&gt;
  2. &lt;head&gt;
  3. &lt;meta charset=&quot;utf-8&quot;&gt;
  4. &lt;meta name=&quot;viewport&quot; content=&quot;width=device-width, initial-scale=1&quot;&gt;
  5. &lt;script src=&quot;https://cdn.rawgit.com/davidshimjs/qrcodejs/gh-pages/qrcode.min.js&quot;&gt;&lt;/script&gt;
  6. &lt;/head&gt;
  7. &lt;body&gt;
  8. &lt;h5&gt;Server message/Code: &lt;span id=&quot;message&quot;&gt;&lt;/span&gt;&lt;/h5&gt;
  9. &lt;div id=&quot;qr&quot;&gt;&lt;/div&gt;
  10. &lt;script&gt;
  11. var source = new EventSource(&quot;http://localhost:1234/sse/dashboard&quot;);
  12. source.onmessage = function (event) {
  13. var message = event.data
  14. document.querySelector(&#39;#message&#39;).innerHTML = message;
  15. if (new String(message).valueOf() == &quot;success&quot; || new String(message).valueOf() == &quot;timeout&quot;
  16. || new String(message).valueOf() == &quot;Already logged&quot;) {
  17. document.querySelector(&#39;#qr&#39;).innerHTML = &quot;&quot;;
  18. } else {
  19. var qrcode = new QRCode(&quot;qr&quot;, {
  20. text: message,
  21. width: 128,
  22. height: 128,
  23. colorDark : &quot;#000000&quot;,
  24. colorLight : &quot;#ffffff&quot;,
  25. correctLevel : QRCode.CorrectLevel.M
  26. });
  27. }
  28. }
  29. /*
  30. var qrcode = new QRCode(
  31. &quot;qr&quot;,
  32. [
  33. &quot;BEGIN:VCARD&quot;,
  34. &quot;VERSION:2.1&quot;,
  35. &quot;N:Doe;John;;Dr;&quot;,
  36. &quot;FN:Dr. John Doe&quot;,
  37. &quot;EMAIL:johndoe@hotmail.com&quot;,
  38. &quot;TEL;TYPE=cell:(123) 555-5832&quot;,
  39. &quot;END:VCARD&quot;
  40. ].join(&quot;\r\n&quot;)
  41. ); */
  42. &lt;/script&gt;
  43. &lt;/body&gt;
  44. &lt;/html&gt;

huangapple
  • 本文由 发表于 2022年6月13日 05:47:41
  • 转载请务必保留本文链接：https://go.coder-hub.com/72596131.html
  • go
  • javascript
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定