diff --git a/background/boss.js b/background/boss.js new file mode 100644 index 0000000..c50112e --- /dev/null +++ b/background/boss.js @@ -0,0 +1,14 @@ +console.log("login start") + + +console.log(window.location.hostname, window.location.pathname); + +if(window.location.hostname === "signup.zhipin.com") { + console.log("登录操作", document.getElementsByClassName("link-signin")); + document.getElementsByClassName("link-signin")[1].click(); + + + // https://www.zhipin.com/web/geek/recommend +} else if ( window.location.pathname == "web/geek/recommend" ) { + +} \ No newline at end of file diff --git a/background/worker.js b/background/worker.js index a188ff8..f012156 100644 --- a/background/worker.js +++ b/background/worker.js @@ -3,7 +3,13 @@ console.log("background"); chrome.runtime.onMessage.addListener(function (msg, sender, sendResponse) { console.log(msg, sender) if(msg.name === "boss") { // boss - chrome.tabs.cre // 创建一个tab 然后注入脚本 + + // 创建一个tab 然后注入脚本 + chrome.tabs.create({ url: "https://signup.zhipin.com/" }, function (tab) { + chrome.tabs.executeScript(tab.id, { runAt: "document_end", file: "background/boss.js" }); + }); + + } // sendResponse({}); diff --git a/content/inject.js b/content/inject.js index 1756d26..68d9b38 100644 --- a/content/inject.js +++ b/content/inject.js @@ -8,15 +8,8 @@ if( window.location.href === "http://localhost:12203/start") { }; ws.onmessage = function(evt) { - console.log("Received Message: ", JSON.parse(evt.data)); - - // if (window.location.hostname !== "www.zhipin.com") { - // var handler = window.open("https://www.zhipin.com/") - // } chrome.runtime.sendMessage(JSON.parse(evt.data)) - - }; ws.onclose = function(evt) { @@ -29,34 +22,7 @@ if( window.location.href === "http://localhost:12203/start") { } - -// setInterval(function name() { -// chrome.runtime.sendMessage({say: "Hello"}) - -// fetch("http://localhost:12203/api").then((response)=>{ -// console.log(response); -// console.log(response.json()) -// }) -// }, 10000) - -// if(href.startsWith(Host)) { -// setInterval(function(){ -// GetTask(); -// }, 1); -// } else { -// setTimeout(function(){ -// chrome.runtime.sendMessage({ type: BackgroundMsgType.CLOSETAB }) -// }, 15000) -// } - -// function GetTask() { -// try { -// chrome.runtime.sendMessage({ type: BackgroundMsgType.GETTASK }, function(){}); -// } catch (error) { -// console.log(error); -// } -// }; - + diff --git a/server/main.go b/server/main.go index 5206e1b..66b9686 100644 --- a/server/main.go +++ b/server/main.go @@ -9,12 +9,6 @@ import ( "github.com/gorilla/websocket" ) -var upGrader = websocket.Upgrader{ - CheckOrigin: func(r *http.Request) bool { - return true - }, -} - var RemoteAddress = ":12203" func init() { @@ -34,6 +28,12 @@ func main() { eg.POST("queue", PostPushQueue) eg.GET("task", func(c *gin.Context) { + + var upGrader = websocket.Upgrader{ + CheckOrigin: func(r *http.Request) bool { + return true + }, + } //升级get请求为webSocket协议 ws, err := upGrader.Upgrade(c.Writer, c.Request, nil) if err != nil { @@ -54,7 +54,7 @@ func main() { //读取ws中的数据 log.Println( - "123", + "start task", ) task := taskQueue.PopBlock() diff --git a/server/task_test.go b/server/task_test.go index 22888f5..0f44e64 100644 --- a/server/task_test.go +++ b/server/task_test.go @@ -87,3 +87,21 @@ func TestTaskPostPushQueue(t *testing.T) { } log.Println(string(resp.Content())) } + +func TestBoss(t *testing.T) { + temp := requests.NewSession().Post("http://localhost" + RemoteAddress + "/queue") + task := &Task{ + Name: "boss", + Type: 1, + CreateAt: time.Now(), + Info: gin.H{ + "data": "boss", + }, + } + temp.SetBodyAuto(task) + resp, err := temp.Execute() + if err != nil { + t.Error(err) + } + log.Println(string(resp.Content())) +}