add netease login plugin

This commit is contained in:
Aynakeya
2022-07-13 18:59:04 -07:00
parent 6f5cfc9028
commit 41e2a4775a
15 changed files with 399 additions and 135 deletions

View File

@@ -8,7 +8,7 @@ import (
"AynaLivePlayer/plugin/diange"
"AynaLivePlayer/plugin/qiege"
"AynaLivePlayer/plugin/textinfo"
"AynaLivePlayer/plugin/webinfo"
"AynaLivePlayer/plugin/wylogin"
"fmt"
"github.com/mitchellh/panicwrap"
"os"
@@ -25,7 +25,8 @@ func init() {
}
}
var plugins = []controller.Plugin{diange.NewDiange(), qiege.NewQiege(), textinfo.NewTextInfo(), webinfo.NewWebInfo()}
var plugins = []controller.Plugin{diange.NewDiange(), qiege.NewQiege(), textinfo.NewTextInfo(),
wylogin.NewWYLogin()}
func main() {
fmt.Printf("BiliAudioBot Revive %s\n", config.VERSION)

View File

@@ -1,124 +0,0 @@
package main
import (
"encoding/json"
"fmt"
"github.com/gorilla/websocket"
"net/http"
"sync"
"time"
)
var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
CheckOrigin: func(r *http.Request) bool {
return true
},
}
type WebInfo struct {
A string
B string
}
type WebInfoServer struct {
ServeMux http.ServeMux
Clients map[*Client]int
lock sync.Mutex
}
type Client struct {
conn *websocket.Conn
Data chan []byte
Close chan byte
}
func NewWebInfoServer() *WebInfoServer {
server := &WebInfoServer{
Clients: map[*Client]int{},
}
server.ServeMux.Handle("/", http.FileServer(http.Dir("./assets/webinfo")))
server.ServeMux.HandleFunc("/ws/info", server.handleInfo)
return server
}
func (s *WebInfoServer) handleInfo(w http.ResponseWriter, r *http.Request) {
fmt.Println("connection start")
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
fmt.Println("upgrade error", err)
return
}
client := &Client{
conn: conn,
Data: make(chan []byte, 16),
Close: make(chan byte, 1),
}
s.addClient(client)
defer s.removeClient(client)
go func() {
for {
_, _, err := client.conn.ReadMessage()
if err != nil {
client.Close <- 1
}
}
}()
for {
fmt.Println("waiting for message")
select {
case data := <-client.Data:
writer, err := client.conn.NextWriter(websocket.TextMessage)
if err != nil {
fmt.Println("get writer error", err)
return
}
if _, err = writer.Write(data); err != nil {
fmt.Println("send error:", err)
return
}
if err = writer.Close(); err != nil {
fmt.Println("can't close writer")
return
}
case _ = <-client.Close:
fmt.Println("client close", client.conn.Close())
return
}
}
}
func (s *WebInfoServer) sendInfo(info *WebInfo) {
for client := range s.Clients {
d, _ := json.Marshal(info)
client.Data <- d
}
}
func (s *WebInfoServer) addClient(c *Client) {
s.lock.Lock()
s.Clients[c] = 1
s.lock.Unlock()
}
func (s *WebInfoServer) removeClient(c *Client) {
s.lock.Lock()
close(c.Data)
delete(s.Clients, c)
s.lock.Unlock()
}
var info WebInfo = WebInfo{A: "asdf", B: "ffff"}
func main() {
server := NewWebInfoServer()
go func() {
for {
time.Sleep(time.Second * 5)
server.sendInfo(&info)
}
}()
http.ListenAndServe("localhost:8080", &server.ServeMux)
}

58
app/wylogin/main.go Normal file
View File

@@ -0,0 +1,58 @@
package main
import (
"fmt"
neteaseApi "github.com/XiaoMengXinX/Music163Api-Go/api"
"github.com/skip2/go-qrcode"
"net/http"
//neteaseTypes "github.com/XiaoMengXinX/Music163Api-Go/types"
neteaseUtil "github.com/XiaoMengXinX/Music163Api-Go/utils"
)
var reqData = neteaseUtil.RequestData{
Cookies: []*http.Cookie{},
Headers: neteaseUtil.Headers{
{
"X-Real-IP",
"118.88.88.88",
},
},
}
func IsLogin() {
}
func main() {
status, err := neteaseApi.GetLoginStatus(reqData)
if err != nil {
return
}
fmt.Println(status.Profile.UserId)
fmt.Println(status.Account.Id)
unikey, err := neteaseApi.GetQrUnikey(reqData)
if err != nil {
return
}
fmt.Println(unikey.Unikey)
qrcode.WriteFile(fmt.Sprintf("https://music.163.com/login?codekey=%s", unikey.Unikey), qrcode.Medium, 256, "qrcode.png")
//wait user input
var input string
_, _ = fmt.Scanln(&input)
login, h, err := neteaseApi.CheckQrLogin(reqData, unikey.Unikey)
if err != nil {
return
}
for _, c := range (&http.Response{Header: h}).Cookies() {
fmt.Println(c)
}
fmt.Println(login.Nickname, login.Message, login.Code)
reqData.Cookies = (&http.Response{Header: h}).Cookies()
status, err = neteaseApi.GetLoginStatus(reqData)
if err != nil {
return
}
fmt.Println(status.Profile.UserId)
fmt.Println(status.Account.Id)
}