Skip to content

Commit

Permalink
Allow specifying file start position
Browse files Browse the repository at this point in the history
Like helix-term/src/commands.rs:3426:15
  • Loading branch information
pickfire committed Nov 13, 2021
1 parent 6d4409c commit cffd0f8
Show file tree
Hide file tree
Showing 4 changed files with 73 additions and 16 deletions.
30 changes: 25 additions & 5 deletions helix-term/src/application.rs
Original file line number Diff line number Diff line change
@@ -1,8 +1,15 @@
use helix_core::{merge_toml_values, syntax};
use helix_core::{line_ending::line_end_char_index, merge_toml_values, syntax, Selection};
use helix_lsp::{lsp, util::lsp_pos_to_pos, LspProgressMap};
use helix_view::{theme, Editor};

use crate::{args::Args, compositor::Compositor, config::Config, job::Jobs, ui};
use crate::{
args::Args,
commands::{align_view, Align},
compositor::Compositor,
config::Config,
job::Jobs,
ui,
};

use log::{error, warn};

Expand Down Expand Up @@ -116,24 +123,37 @@ impl Application {
// Unset path to prevent accidentally saving to the original tutor file.
doc_mut!(editor).set_path(None)?;
} else if !args.files.is_empty() {
let first = &args.files[0]; // we know it's not empty
let first = &args.files[0].0; // we know it's not empty
if first.is_dir() {
std::env::set_current_dir(&first)?;
editor.new_file(Action::VerticalSplit);
compositor.push(Box::new(ui::file_picker(".".into())));
} else {
let nr_of_files = args.files.len();
editor.open(first.to_path_buf(), Action::VerticalSplit)?;
for file in args.files {
for (file, row, col) in args.files {
if file.is_dir() {
return Err(anyhow::anyhow!(
"expected a path to file, found a directory. (to open a directory pass it as first argument)"
));
} else {
editor.open(file.to_path_buf(), Action::Load)?;
let doc_id = editor.open(file, Action::Load)?;
// with Action::Load all documents have the same view
let view_id = editor.tree.focus;
let doc = editor.document_mut(doc_id).unwrap();
let row = row.min(doc.text().len_lines() - 1);
let pos = Selection::point(
(doc.text().line_to_char(row) + col)
.min(line_end_char_index(&doc.text().slice(..), row)),
);
doc.set_selection(view_id, pos);
}
}
editor.set_status(format!("Loaded {} files.", nr_of_files));
// align the view to center after all files are loaded,
// does not affect views without pos since it is at the top
let (view, doc) = current!(editor);
align_view(doc, view, Align::Center);
}
} else if stdin().is_tty() {
editor.new_file(Action::VerticalSplit);
Expand Down
36 changes: 32 additions & 4 deletions helix-term/src/args.rs
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ pub struct Args {
pub display_version: bool,
pub load_tutor: bool,
pub verbosity: u64,
pub files: Vec<PathBuf>,
pub files: Vec<(PathBuf, usize, usize)>,
}

impl Args {
Expand Down Expand Up @@ -41,15 +41,43 @@ impl Args {
}
}
}
arg => args.files.push(PathBuf::from(arg)),
arg => args.files.push(parse_file(arg)),
}
}

// push the remaining args, if any to the files
for filename in iter {
args.files.push(PathBuf::from(filename));
for arg in iter {
args.files.push(parse_file(arg));
}

Ok(args)
}
}

/// Parse arg into [`PathBuf`] and position.
pub(crate) fn parse_file(s: &str) -> (PathBuf, usize, usize) {
split_path_row_col(s)
.or_else(|| split_path_row(s))
.unwrap_or_else(|| (PathBuf::from(s), 0, 0))
}

/// Split file.rs:10:2 into [`PathBuf`], row and col.
///
/// Does not validate if file.rs is a file or directory.
fn split_path_row_col(s: &str) -> Option<(PathBuf, usize, usize)> {
let mut s = s.rsplitn(3, ':');
let col: usize = s.next()?.parse().ok()?;
let row: usize = s.next()?.parse().ok()?;
let path = s.next()?.into();
Some((path, row.saturating_sub(1), col.saturating_sub(1)))
}

/// Split file.rs:10 into [`PathBuf`] and row.
///
/// Does not validate if file.rs is a file or directory.
fn split_path_row(s: &str) -> Option<(PathBuf, usize, usize)> {
let mut s = s.rsplitn(2, ':');
let row: usize = s.next()?.parse().ok()?;
let path = s.next()?.into();
Some((path, row.saturating_sub(1), 0))
}
21 changes: 15 additions & 6 deletions helix-term/src/commands.rs
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ use insert::*;
use movement::Movement;

use crate::{
args,
compositor::{self, Component, Compositor},
ui::{self, FilePicker, Picker, Popup, Prompt, PromptEvent},
};
Expand Down Expand Up @@ -109,13 +110,13 @@ impl<'a> Context<'a> {
}
}

enum Align {
pub(crate) enum Align {
Top,
Center,
Bottom,
}

fn align_view(doc: &Document, view: &mut View, align: Align) {
pub(crate) fn align_view(doc: &Document, view: &mut View, align: Align) {
let pos = doc
.selection(view.id)
.primary()
Expand Down Expand Up @@ -1682,10 +1683,18 @@ mod cmd {
_event: PromptEvent,
) -> anyhow::Result<()> {
use helix_core::path::expand_tilde;
let path = args.get(0).context("wrong argument count")?;
let _ = cx
.editor
.open(expand_tilde(Path::new(path)), Action::Replace)?;
let arg = args.get(0).context("wrong argument count")?;
let (path, row, col) = args::parse_file(arg);
let _ = cx.editor.open(expand_tilde(&path), Action::Replace)?;
let (view, doc) = current!(cx.editor);
let row = row.min(doc.text().len_lines() - 1);
let pos = Selection::point(
(doc.text().line_to_char(row) + col)
.min(line_end_char_index(&doc.text().slice(..), row)),
);
doc.set_selection(view.id, pos);
// does not affect opening a buffer without pos
align_view(doc, view, Align::Center);
Ok(())
}

Expand Down
2 changes: 1 addition & 1 deletion helix-term/src/main.rs
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ USAGE:
hx [FLAGS] [files]...
ARGS:
<files>... Sets the input file to use
<files>... Sets the input file to use or with pos file[:row[:col]]
FLAGS:
-h, --help Prints help information
Expand Down

0 comments on commit cffd0f8

Please sign in to comment.