feat: buffer send/recv

src/main/java/com/codemp/intellij/actions/ConnectAction.java

@@ -10,26 +10,30 @@ import org.jetbrains.annotations.NotNull;
 import java.io.IOException;
 
 public class ConnectAction extends AnAction {
-	static {
+	public ConnectAction() {
+		super();
 		/*try {
 			NativeUtils.loadLibraryFromJar("/resources/libHelloJNI.so");
 		} catch(IOException e) {
 			throw new RuntimeException(e);
 		}*/
-		System.load("/home/zaaarf/dev/irl/rust/codemp/client/intellij/target/debug/libcodemp_intellij.so");
-		//System.load("O:/dev/IRL/Rust/codemp/client/intellij/target/debug/codemp_intellij.dll");
+		//System.load("/home/zaaarf/dev/irl/rust/codemp/client/intellij/target/debug/libcodemp_intellij.so");
+		System.load("O:/dev/IRL/Rust/codemp/client/intellij/target/debug/codemp_intellij.dll");
+	}
+
+	public void connect(String url) throws Exception {
+		CodeMPHandler.connect(url);
+		//Messages.showInfoMessage(String.format("Connected to %s!", url), "CodeMP");
+		System.out.printf("Connected to %s!\n", url);
 	}
 
 	@Override
 	public void actionPerformed(@NotNull AnActionEvent e) {
-		String url = Messages.showInputDialog("URL to CodeMP instance:", "CodeMP Connect", Messages.getQuestionIcon());
-		if(url == null || url.isBlank())
-			url = "http://alemi.dev:50051";
+		String url = Messages.showInputDialog("URL to CodeMP instance:", "CodeMP Connect",
+			Messages.getQuestionIcon());
 
 		try {
-			CodeMPHandler.connect(url);
-			//Messages.showInfoMessage(String.format("Connected to %s!", url), "CodeMP");
-			System.out.printf("Connected to %s!\n", url);
+			this.connect(url);
 		} catch(Exception ex) {
 			Messages.showErrorDialog(String.format("Failed to connect to %s: %s!", url, ex.getMessage()), "CodeMP");
 		}

src/main/java/com/codemp/intellij/actions/FastForwardAction.java

@@ -0,0 +1,22 @@
+package com.codemp.intellij.actions;
+
+import com.codemp.intellij.actions.buffer.BufferAttachAction;
+import com.intellij.openapi.actionSystem.AnAction;
+import com.intellij.openapi.actionSystem.AnActionEvent;
+import org.jetbrains.annotations.NotNull;
+
+public class FastForwardAction extends AnAction {
+	@Override
+	public void actionPerformed(@NotNull AnActionEvent e) {
+		ConnectAction connectAction = new ConnectAction();
+		JoinAction joinAction = new JoinAction();
+		BufferAttachAction attachAction = new BufferAttachAction();
+		try {
+			connectAction.connect("http://alemi.dev:50051");
+			joinAction.join(e, "default");
+			attachAction.attach(e, "test");
+		} catch(Exception ex) {
+			throw new RuntimeException(ex);
+		}
+	}
+}

src/main/java/com/codemp/intellij/actions/JoinAction.java

@@ -36,22 +36,13 @@ public class JoinAction extends AnAction {
 		null, JBColor.BLUE, null, null, Font.PLAIN
 	);
 
-	@Override
-	public void actionPerformed(@NotNull AnActionEvent e) {
-		String session = Messages.showInputDialog(
-			"Session to connect to:",
-			"CodeMP Join",
-			Messages.getQuestionIcon());
-		if(session == null || session.isBlank())
-			session = "default";
-
-		try {
+	public void join(AnActionEvent e, String session) throws Exception {
 		CursorHandler cursorHandler = CodeMPHandler.join(session);
 		EditorFactory.getInstance()
 			.getEventMulticaster()
 			.addCaretListener(new CursorEventListener());
 		//Messages.showInfoMessage(String.format("Joined %s!", session), "CodeMP");
-			System.out.printf(String.format("Joined %s!", session));
+		System.out.printf(String.format("Joined %s!\n", session));
 		Editor editor = FileEditorManager.getInstance(Objects.requireNonNull(e.getProject()))
 			.getSelectedTextEditor();
 		assert editor != null;
@@ -93,7 +84,17 @@ public class JoinAction extends AnAction {
 				}
 			}
 		});
+	}
 
+	@Override
+	public void actionPerformed(@NotNull AnActionEvent e) {
+		String session = Messages.showInputDialog(
+			"Session to connect to:",
+			"CodeMP Join",
+			Messages.getQuestionIcon());
+
+		try {
+			this.join(e, session);
 		} catch(Exception ex) {
 			Messages.showErrorDialog(String.format(
 					"Failed to join session %s: %s!",

src/main/java/com/codemp/intellij/actions/buffer/BufferAttachAction.java

@@ -0,0 +1,77 @@
+package com.codemp.intellij.actions.buffer;
+
+import com.codemp.intellij.jni.BufferHandler;
+import com.codemp.intellij.jni.CodeMPHandler;
+import com.codemp.intellij.jni.TextChangeWrapper;
+import com.codemp.intellij.listeners.BufferEventListener;
+import com.intellij.openapi.actionSystem.AnAction;
+import com.intellij.openapi.actionSystem.AnActionEvent;
+import com.intellij.openapi.application.ApplicationManager;
+import com.intellij.openapi.editor.Document;
+import com.intellij.openapi.editor.Editor;
+import com.intellij.openapi.fileEditor.FileEditorManager;
+import com.intellij.openapi.progress.ProgressIndicator;
+import com.intellij.openapi.progress.ProgressManager;
+import com.intellij.openapi.progress.Task;
+import com.intellij.openapi.ui.Messages;
+import org.jetbrains.annotations.NotNull;
+
+import javax.print.Doc;
+import java.util.Objects;
+
+public class BufferAttachAction extends AnAction {
+
+	public void attach(AnActionEvent e, String buffer) throws Exception {
+		BufferHandler bufferHandler = CodeMPHandler.attach(buffer);
+		//Messages.showInfoMessage(String.format("Connected to %s!", url), "CodeMP");
+
+		//register buffer change listener
+		//TODO "get" the Document corresponding to buffer, for now use the current one
+		BufferEventListener listener = new BufferEventListener(buffer);
+		assert e.getProject() != null;
+		Editor editor = FileEditorManager.getInstance(e.getProject()).getSelectedTextEditor();
+		assert editor != null;
+		Document document = editor.getDocument();
+		document.addDocumentListener(listener);
+
+		ProgressManager.getInstance().run(new Task.Backgroundable(e.getProject(), "Awaiting CodeMP buffer events") {
+			@Override
+			public void run(@NotNull ProgressIndicator indicator) {
+				while(true) {
+					try {
+						TextChangeWrapper event = bufferHandler.recv();
+						ApplicationManager.getApplication().invokeLaterOnWriteThread(() ->
+							ApplicationManager.getApplication().runWriteAction(() -> {
+								System.out.printf("Received text change %s from offset %d to %d!\n",
+									event.getContent(), event.getStart(), event.getEnd());
+								document.replaceString( //TODO this doesn't work
+									(int) event.getStart(),
+									(int) event.getEnd(),
+									event.getContent()
+								);
+							}));
+					} catch(Exception ex) {
+						throw new RuntimeException(ex);
+					}
+				}
+			}
+		});
+		System.out.printf("Created buffer %s!\n", buffer);
+	}
+
+	@Override
+	public void actionPerformed(@NotNull AnActionEvent e) {
+		String buffer = Messages.showInputDialog(
+			"Buffer name:",
+			"Attach to CodeMP Buffer",
+			Messages.getQuestionIcon());
+
+		try {
+			this.attach(e, buffer);
+		} catch(Exception ex) {
+			Messages.showErrorDialog(String.format(
+				"Failed to attach to buffer %s: %s!",
+				buffer, ex.getMessage()), "Attach to CodeMP Buffer");
+		}
+	}
+}

src/main/java/com/codemp/intellij/listeners/BufferEventListener.java

@@ -0,0 +1,37 @@
+package com.codemp.intellij.listeners;
+
+import com.codemp.intellij.jni.BufferHandler;
+import com.codemp.intellij.jni.CodeMPHandler;
+import com.intellij.openapi.editor.event.DocumentEvent;
+import com.intellij.openapi.editor.event.DocumentListener;
+import org.jetbrains.annotations.NotNull;
+
+public class BufferEventListener implements DocumentListener {
+
+	private final String bufferName;
+
+	public BufferEventListener(String bufferName) {
+		this.bufferName = bufferName;
+	}
+
+	@Override
+	public void documentChanged(@NotNull DocumentEvent event) {
+		try {
+			int changeOffset = event.getOffset();
+			CharSequence newFragment = event.getNewFragment();
+			BufferHandler bufferHandler = CodeMPHandler.getBuffer(this.bufferName);
+			bufferHandler.send(changeOffset, changeOffset + event.getOldFragment().length(),
+				newFragment.toString());
+		} catch(Exception e) {
+			throw new RuntimeException(e);
+		}
+	}
+}
+/*
+ABCD
+ABAACD
+
+getOffset() -> B + 1
+
+
+ */

src/main/java/com/codemp/intellij/listeners/CursorEventListener.java

@@ -2,7 +2,6 @@ package com.codemp.intellij.listeners;
 
 import com.codemp.intellij.jni.CodeMPHandler;
 import com.intellij.openapi.editor.Caret;
-import com.intellij.openapi.editor.EditorFactory;
 import com.intellij.openapi.editor.VisualPosition;
 import com.intellij.openapi.editor.event.CaretEvent;
 import com.intellij.openapi.editor.event.CaretListener;
@@ -20,7 +19,8 @@ public class CursorEventListener implements CaretListener {
 		try {
 			VisualPosition startPos = caret.getSelectionStartPosition();
 			VisualPosition endPos = caret.getSelectionEndPosition();
-			System.out.printf("start %dx %dy end %dx %dy", startPos.line, startPos.column, endPos.line, endPos.column);
+			System.out.printf("start %dx %dy end %dx %dy",
+				startPos.line, startPos.column, endPos.line, endPos.column);
 			CodeMPHandler.getCursor().send(
 				"", startPos.line, startPos.column, endPos.line, endPos.column
 			);

src/main/resources/META-INF/plugin.xml

@@ -25,17 +25,14 @@
     <actions>
         <group id="codemp" text="CodeMP" popup="true">
             <add-to-group group-id="ToolsMenu" anchor="first"/>
-            <action id="codemp.connect" class="com.codemp.intellij.actions.ConnectAction" text="Connect...">
-                <add-to-group group-id="ToolsMenu" anchor="first"/>
-            </action>
-            <action id="codemp.join" class="com.codemp.intellij.actions.JoinAction" text="Join...">
-                <add-to-group group-id="ToolsMenu" anchor="first"/>
-            </action>
+            <action id="codemp.fast-forward" class="com.codemp.intellij.actions.FastForwardAction" text="Pls hurry"/>
+            <action id="codemp.connect" class="com.codemp.intellij.actions.ConnectAction" text="Connect..."/>
+            <action id="codemp.join" class="com.codemp.intellij.actions.JoinAction" text="Join..."/>
             <group id="codemp.buffer" text="Buffer" popup="true">
                 <action id="codemp.buffer.create" class="com.codemp.intellij.actions.buffer.BufferCreateAction"
-                        text="Create Buffer">
-                    <add-to-group group-id="ToolsMenu" anchor="first"/>
-                </action>
+                        text="Create Buffer"/>
+                <action id="codemp.buffer.attach" class="com.codemp.intellij.actions.buffer.BufferAttachAction"
+                        text="Attach to Buffer"/>
             </group>
         </group>
 	</actions>

src/main/rust/lib.rs

@@ -93,33 +93,33 @@ impl CursorEventWrapper {
 	}
 
 	#[generate_interface]
-	fn get_user(&self) -> Result<&str, String> {
-		Ok(&self.user)
+	fn get_user(&self) -> &str {
+		&self.user
 	}
 
 	#[generate_interface]
-	fn get_buffer(&self) -> Result<&str, String> {
-		Ok(&self.buffer)
+	fn get_buffer(&self) -> &str {
+		&self.buffer
 	}
 
 	#[generate_interface]
-	fn get_start_row(&self) -> Result<i32, String> {
-		Ok(self.start_row)
+	fn get_start_row(&self) -> i32 {
+		self.start_row
 	}
 
 	#[generate_interface]
-	fn get_start_col(&self) -> Result<i32, String> {
-		Ok(self.start_col)
+	fn get_start_col(&self) -> i32 {
+		self.start_col
 	}
 
 	#[generate_interface]
-	fn get_end_row(&self) -> Result<i32, String> {
-		Ok(self.end_row)
+	fn get_end_row(&self) -> i32 {
+		self.end_row
 	}
 
 	#[generate_interface]
-	fn get_end_col(&self) -> Result<i32, String> {
-		Ok(self.end_col)
+	fn get_end_col(&self) -> i32 {
+		self.end_col
 	}
 }
 
@@ -139,8 +139,7 @@ impl CursorHandler {
 	fn recv(&self) -> Result<CursorEventWrapper, String>  {
 		match self.cursor.blocking_recv(CODEMP_INSTANCE.rt()) {
 			Err(err) => Err(ErrorWrapper::from(err).get_error_message()),
-			Ok(event) => {
-				Ok(CursorEventWrapper {
+			Ok(event) => Ok(CursorEventWrapper {
 				user: event.user,
 				buffer: event.position.as_ref().unwrap().buffer.clone(),
 				start_row: event.position.as_ref().unwrap().start().row,
@@ -150,7 +149,6 @@ impl CursorHandler {
 			})
 		}
 	}
-	}
 
 	#[generate_interface]
 	fn send(&self, buffer: String, start_row: i32, start_col: i32, end_row: i32, end_col: i32) -> Result<(), String> {
@@ -162,6 +160,35 @@ impl CursorHandler {
 	}
 }
 
+#[generate_interface_doc]
+struct TextChangeWrapper {
+	start: usize,
+	end: usize, //not inclusive
+	content: String
+}
+
+impl TextChangeWrapper {
+	#[generate_interface(constructor)]
+	fn new() -> TextChangeWrapper {
+		panic!("Default constructor for TextChangeWrapper should never be called!")
+	}
+
+	#[generate_interface]
+	fn get_start(&self) -> usize {
+		self.start
+	}
+
+	#[generate_interface]
+	fn get_end(&self) -> usize {
+		self.end
+	}
+
+	#[generate_interface]
+	fn get_content(&self) -> &str {
+		&self.content
+	}
+}
+
 #[generate_interface_doc]
 struct BufferHandler {
 	#[allow(unused)]
@@ -173,4 +200,25 @@ impl BufferHandler {
 	fn new() -> BufferHandler {
 		panic!("Default constructor for BufferHandler should never be called!")
 	}
+
+	#[generate_interface]
+	fn recv(&self) -> Result<TextChangeWrapper, String> {
+		match self.buffer.blocking_recv(CODEMP_INSTANCE.rt()) {
+			Err(err) => Err(ErrorWrapper::from(err).get_error_message()),
+			Ok(change) => Ok(TextChangeWrapper {
+				start: change.span.start,
+				end: change.span.end,
+				content: change.content.clone()
+			})
+		}
+	}
+
+	#[generate_interface]
+	fn send(&self, start_offset: usize, end_offset: usize, content: String) -> Result<(), String> {
+		match self.buffer.delta(start_offset, &content, end_offset) {
+			None => Err("Cannot send a no-op".to_string()),
+			Some(op_seq) => self.buffer.send(op_seq)
+				.map_err(|err| ErrorWrapper::from(err).get_error_message())
+		}
+	}
 }