Support primary clipboard

helix-term/src/commands.rs

@@ -258,12 +258,17 @@ impl Command {
         yank, "Yank selection",
         yank_joined_to_clipboard, "Join and yank selections to clipboard",
         yank_main_selection_to_clipboard, "Yank main selection to clipboard",
+        yank_joined_to_primary_selection, "Join and yank selections to primary selection",
+        yank_main_selection_to_primary_selection, "Yank main selection to primary selection",
         replace_with_yanked, "Replace with yanked text",
         replace_selections_with_clipboard, "Replace selections by clipboard content",
+        replace_selections_with_primary_selection, "Replace selections by primary selection content",
         paste_after, "Paste after selection",
         paste_before, "Paste before selection",
         paste_clipboard_after, "Paste clipboard after selections",
         paste_clipboard_before, "Paste clipboard before selections",
+        paste_primary_selection_after, "Paste primary selection after selections",
+        paste_primary_selection_before, "Paste primary selection before selections",
         indent, "Indent selection",
         unindent, "Unindent selection",
         format_selections, "Format selection",
@@ -1642,6 +1647,26 @@ mod cmd {
         yank_joined_to_clipboard_impl(&mut cx.editor, separator)
     }
 
+    fn yank_main_selection_to_primary_selection(
+        cx: &mut compositor::Context,
+        _args: &[&str],
+        _event: PromptEvent,
+    ) -> anyhow::Result<()> {
+        yank_main_selection_to_primary_selection_impl(&mut cx.editor)
+    }
+
+    fn yank_joined_to_primary_selection(
+        cx: &mut compositor::Context,
+        args: &[&str],
+        _event: PromptEvent,
+    ) -> anyhow::Result<()> {
+        let (_, doc) = current!(cx.editor);
+        let separator = args
+            .first()
+            .copied()
+            .unwrap_or_else(|| doc.line_ending.as_str());
+        yank_joined_to_primary_selection_impl(&mut cx.editor, separator)
+    }
     fn paste_clipboard_after(
         cx: &mut compositor::Context,
         _args: &[&str],
@@ -1681,13 +1706,55 @@ mod cmd {
         }
     }
 
+    fn paste_primary_selection_after(
+        cx: &mut compositor::Context,
+        _args: &[&str],
+        _event: PromptEvent,
+    ) -> anyhow::Result<()> {
+        paste_primary_selection_impl(&mut cx.editor, Paste::After)
+    }
+
+    fn paste_primary_selection_before(
+        cx: &mut compositor::Context,
+        _args: &[&str],
+        _event: PromptEvent,
+    ) -> anyhow::Result<()> {
+        paste_primary_selection_impl(&mut cx.editor, Paste::After)
+    }
+
+    fn replace_selections_with_primary_selection(
+        cx: &mut compositor::Context,
+        _args: &[&str],
+        _event: PromptEvent,
+    ) -> anyhow::Result<()> {
+        let (view, doc) = current!(cx.editor);
+
+        match cx.editor.primary_selection_provider.get_contents() {
+            Ok(contents) => {
+                let selection = doc.selection(view.id);
+                let transaction =
+                    Transaction::change_by_selection(doc.text(), selection, |range| {
+                        (range.from(), range.to(), Some(contents.as_str().into()))
+                    });
+
+                doc.apply(&transaction, view.id);
+                doc.append_changes_to_history(view.id);
+                Ok(())
+            }
+            Err(e) => Err(e.context("Couldn't get system primary selection contents")),
+        }
+    }
+
     fn show_clipboard_provider(
         cx: &mut compositor::Context,
         _args: &[&str],
         _event: PromptEvent,
     ) -> anyhow::Result<()> {
-        cx.editor
-            .set_status(cx.editor.clipboard_provider.name().into());
+        cx.editor.set_status(format!(
+            "Clipboard: {}; Primary selection: {}",
+            cx.editor.clipboard_provider.name(),
+            cx.editor.primary_selection_provider.name()
+        ));
         Ok(())
     }
 
@@ -1888,6 +1955,20 @@ mod cmd {
             fun: yank_joined_to_clipboard,
             completer: None,
         },
+        TypableCommand {
+            name: "primary-selection-yank",
+            alias: None,
+            doc: "Yank main selection into system primary selection.",
+            fun: yank_main_selection_to_primary_selection,
+            completer: None,
+        },
+        TypableCommand {
+            name: "primary-selection-yank-join",
+            alias: None,
+            doc: "Yank joined selections into system primary selection. A separator can be provided as first argument. Default value is newline.", // FIXME: current UI can't display long doc.
+            fun: yank_joined_to_primary_selection,
+            completer: None,
+        },
         TypableCommand {
             name: "clipboard-paste-after",
             alias: None,
@@ -1909,6 +1990,27 @@ mod cmd {
             fun: replace_selections_with_clipboard,
             completer: None,
         },
+        TypableCommand {
+            name: "primary-selection-paste-after",
+            alias: None,
+            doc: "Paste primary selection after selections.",
+            fun: paste_primary_selection_after,
+            completer: None,
+        },
+        TypableCommand {
+            name: "primary-selection-paste-before",
+            alias: None,
+            doc: "Paste primary selection before selections.",
+            fun: paste_primary_selection_before,
+            completer: None,
+        },
+        TypableCommand {
+            name: "primary-selection-paste-replace",
+            alias: None,
+            doc: "Replace selections with content of system primary selection.",
+            fun: replace_selections_with_primary_selection,
+            completer: None,
+        },
         TypableCommand {
             name: "show-clipboard-provider",
             alias: None,
@@ -3181,6 +3283,62 @@ fn yank_main_selection_to_clipboard(cx: &mut Context) {
     let _ = yank_main_selection_to_clipboard_impl(&mut cx.editor);
 }
 
+fn yank_joined_to_primary_selection_impl(
+    editor: &mut Editor,
+    separator: &str,
+) -> anyhow::Result<()> {
+    let (view, doc) = current!(editor);
+    let text = doc.text().slice(..);
+
+    let values: Vec<String> = doc
+        .selection(view.id)
+        .fragments(text)
+        .map(Cow::into_owned)
+        .collect();
+
+    let msg = format!(
+        "joined and yanked {} selection(s) to system primary selection",
+        values.len(),
+    );
+
+    let joined = values.join(separator);
+
+    editor
+        .primary_selection_provider
+        .set_contents(joined)
+        .context("Couldn't set system primary selection content")?;
+
+    editor.set_status(msg);
+
+    Ok(())
+}
+
+fn yank_joined_to_primary_selection(cx: &mut Context) {
+    let line_ending = current!(cx.editor).1.line_ending;
+    let _ = yank_joined_to_primary_selection_impl(&mut cx.editor, line_ending.as_str());
+}
+
+fn yank_main_selection_to_primary_selection_impl(editor: &mut Editor) -> anyhow::Result<()> {
+    let (view, doc) = current!(editor);
+    let text = doc.text().slice(..);
+
+    let value = doc.selection(view.id).primary().fragment(text);
+
+    if let Err(e) = editor
+        .primary_selection_provider
+        .set_contents(value.into_owned())
+    {
+        bail!("Couldn't set system primary selection content: {:?}", e);
+    }
+
+    editor.set_status("yanked main selection to system primary selection".to_owned());
+    Ok(())
+}
+
+fn yank_main_selection_to_primary_selection(cx: &mut Context) {
+    let _ = yank_main_selection_to_primary_selection_impl(&mut cx.editor);
+}
+
 #[derive(Copy, Clone)]
 enum Paste {
     Before,
@@ -3256,6 +3414,32 @@ fn paste_clipboard_before(cx: &mut Context) {
     let _ = paste_clipboard_impl(&mut cx.editor, Paste::Before);
 }
 
+fn paste_primary_selection_impl(editor: &mut Editor, action: Paste) -> anyhow::Result<()> {
+    let (view, doc) = current!(editor);
+
+    match editor
+        .primary_selection_provider
+        .get_contents()
+        .map(|contents| paste_impl(&[contents], doc, view, action))
+    {
+        Ok(Some(transaction)) => {
+            doc.apply(&transaction, view.id);
+            doc.append_changes_to_history(view.id);
+            Ok(())
+        }
+        Ok(None) => Ok(()),
+        Err(e) => Err(e.context("Couldn't get system primary clipboard contents")),
+    }
+}
+
+fn paste_primary_selection_after(cx: &mut Context) {
+    let _ = paste_primary_selection_impl(&mut cx.editor, Paste::After);
+}
+
+fn paste_primary_selection_before(cx: &mut Context) {
+    let _ = paste_primary_selection_impl(&mut cx.editor, Paste::Before);
+}
+
 fn replace_with_yanked(cx: &mut Context) {
     let reg_name = cx.selected_register.name();
     let (view, doc) = current!(cx.editor);
@@ -3300,6 +3484,28 @@ fn replace_selections_with_clipboard(cx: &mut Context) {
     let _ = replace_selections_with_clipboard_impl(&mut cx.editor);
 }
 
+fn replace_selections_with_primary_selection_impl(editor: &mut Editor) -> anyhow::Result<()> {
+    let (view, doc) = current!(editor);
+
+    match editor.primary_selection_provider.get_contents() {
+        Ok(contents) => {
+            let selection = doc.selection(view.id);
+            let transaction = Transaction::change_by_selection(doc.text(), selection, |range| {
+                (range.from(), range.to(), Some(contents.as_str().into()))
+            });
+
+            doc.apply(&transaction, view.id);
+            doc.append_changes_to_history(view.id);
+            Ok(())
+        }
+        Err(e) => Err(e.context("Couldn't get system primary selection contents")),
+    }
+}
+
+fn replace_selections_with_primary_selection(cx: &mut Context) {
+    let _ = replace_selections_with_primary_selection_impl(&mut cx.editor);
+}
+
 fn paste_after(cx: &mut Context) {
     let reg_name = cx.selected_register.name();
     let (view, doc) = current!(cx.editor);

helix-term/src/ui/editor.rs

@@ -856,6 +856,69 @@ impl Component for EditorView {
                 doc.set_selection(view.id, selection);
                 EventResult::Consumed(None)
             }
+
+            Event::Mouse(MouseEvent {
+                kind: MouseEventKind::Up(MouseButton::Left),
+                ..
+            }) => {
+                let (view, doc) = current!(cx.editor);
+                let range = doc.selection(view.id).primary();
+
+                if range.to() - range.from() <= 1 {
+                    return EventResult::Ignored;
+                }
+
+                let mut cxt = commands::Context {
+                    selected_register: helix_view::RegisterSelection::default(),
+                    editor: &mut cx.editor,
+                    count: None,
+                    callback: None,
+                    on_next_key_callback: None,
+                    jobs: cx.jobs,
+                };
+
+                commands::Command::yank_main_selection_to_primary_selection.execute(&mut cxt);
+
+                EventResult::Consumed(None)
+            }
+
+            Event::Mouse(MouseEvent {
+                kind: MouseEventKind::Up(MouseButton::Middle),
+                row,
+                column,
+                ..
+            }) => {
+                let editor = &mut cx.editor;
+
+                let result = editor.tree.views().find_map(|(view, _focus)| {
+                    view.pos_at_screen_coords(&editor.documents[view.doc], row, column)
+                        .map(|pos| (pos, view.id))
+                });
+
+                if let Some((pos, view_id)) = result {
+                    let doc = &mut editor.documents[editor.tree.get(view_id).doc];
+
+                    doc.set_selection(view_id, Selection::point(pos));
+
+                    editor.tree.focus = view_id;
+
+                    let mut cxt = commands::Context {
+                        selected_register: helix_view::RegisterSelection::default(),
+                        editor: &mut cx.editor,
+                        count: None,
+                        callback: None,
+                        on_next_key_callback: None,
+                        jobs: cx.jobs,
+                    };
+
+                    commands::Command::paste_primary_selection_before.execute(&mut cxt);
+
+                    return EventResult::Consumed(None);
+                }
+
+                EventResult::Ignored
+            }
+
             Event::Mouse(_) => EventResult::Ignored,
         }
     }