Start work on a login command

Co-Authored-By: Nathan Sobo <nathan@zed.dev>
2021-06-07 18:15:11 -07:00 · 2021-06-07 18:15:11 -07:00 · db3e01a12c
commit db3e01a12c
parent 3a932cc9bf
9 changed files with 178 additions and 12 deletions
--- a/zed/src/lib.rs
+++ b/zed/src/lib.rs
@ -1,3 +1,8 @@
+use anyhow::{anyhow, Context};
+use gpui::MutableAppContext;
+use smol::io::{AsyncBufReadExt, AsyncWriteExt};
+use url::Url;
+
 pub mod assets;
 pub mod editor;
 pub mod file_finder;
@ -18,3 +23,83 @@ pub struct AppState {
    pub settings: postage::watch::Receiver<settings::Settings>,
    pub language_registry: std::sync::Arc<language::LanguageRegistry>,
 }
+
+pub fn init(cx: &mut MutableAppContext) {
+    cx.add_global_action("app:authenticate", authenticate);
+    cx.add_global_action("app:quit", quit);
+}
+
+fn authenticate(_: &(), cx: &mut MutableAppContext) {
+    let zed_url = std::env::var("ZED_SERVER_URL").unwrap_or("https://zed.dev".to_string());
+    let platform = cx.platform().clone();
+
+    dbg!(&zed_url);
+
+    let task = cx.background_executor().spawn(async move {
+        let listener = smol::net::TcpListener::bind("127.0.0.1:0").await?;
+        let port = listener.local_addr()?.port();
+
+        platform.open_url(&format!(
+            "{}/sign_in?native_app_port={}&native_app_public_key=unused-for-now",
+            zed_url, port,
+        ));
+
+        let (mut stream, _) = listener.accept().await?;
+        let mut reader = smol::io::BufReader::new(&mut stream);
+        let mut line = String::new();
+        reader.read_line(&mut line).await?;
+
+        let mut parts = line.split(" ");
+        if parts.next() == Some("GET") {
+            if let Some(path) = parts.next() {
+                let url = Url::parse(&format!("http://example.com{}", path))
+                    .context("failed to parse login notification url")?;
+                let mut access_token = None;
+                let mut public_key = None;
+                for (key, value) in url.query_pairs() {
+                    if key == "access_token" {
+                        access_token = Some(value);
+                    } else if key == "public_key" {
+                        public_key = Some(value);
+                    }
+                }
+                stream
+                    .write_all(LOGIN_RESPONSE.as_bytes())
+                    .await
+                    .context("failed to write login response")?;
+                stream.flush().await.context("failed to flush tcp stream")?;
+
+                eprintln!(
+                    "logged in. access_token: {:?}, public_key: {:?}",
+                    access_token, public_key
+                );
+
+                platform.activate(true);
+                return Ok(());
+            }
+        }
+        Err(anyhow!("failed to parse http request from zed web app"))
+    });
+
+    cx.spawn(|_| async move {
+        if let Err(e) = task.await {
+            log::error!("failed to login {:?}", e)
+        }
+    })
+    .detach();
+}
+
+fn quit(_: &(), cx: &mut MutableAppContext) {
+    cx.platform().quit();
+}
+
+const LOGIN_RESPONSE: &'static str = "
+HTTP/1.1 200 OK\r
+Content-Length: 64\r
+Content-Type: text/html\r
+\r
+<!DOCTYPE html>
+<html>
+<script>window.close();</script>
+</html>
+";