ocs 0.1.2

A session picker for OpenCode
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89

use super::*;

pub(crate) struct SessionItem {
  pub(crate) data_dir: PathBuf,
  pub(crate) id: String,
  pub(crate) preview: OnceLock<String>,
  pub(crate) project: String,
  pub(crate) search_text: String,
  pub(crate) title: String,
}

impl SessionItem {
  pub(crate) fn new(storage: &Storage, session: &Session) -> Self {
    let project = Path::new(&session.directory)
      .file_name()
      .and_then(|name| name.to_str())
      .unwrap_or(&session.directory);

    Self {
      data_dir: storage.data_dir.clone(),
      id: session.id.clone(),
      preview: OnceLock::new(),
      project: project.into(),
      search_text: session.search_text(),
      title: session.title.clone(),
    }
  }
}

impl SkimItem for SessionItem {
  fn display(&self, _context: DisplayContext) -> Line<'_> {
    Line::from(vec![
      Span::raw(self.title.as_str()),
      Span::raw(" "),
      Span::styled(self.project.as_str(), Style::new().fg(DARK_GRAY)),
    ])
  }

  fn output(&self) -> Cow<'_, str> {
    Cow::Borrowed(&self.id)
  }

  fn preview(&self, _context: PreviewContext) -> ItemPreview {
    ItemPreview::AnsiText(
      self
        .preview
        .get_or_init(|| {
          Storage::new(self.data_dir.clone())
            .delete_session(&self.id)
            .map_or_else(
              |error| format!("could not load preview: {error}"),
              |session| session.preview(),
            )
        })
        .clone(),
    )
  }

  fn text(&self) -> Cow<'_, str> {
    Cow::Borrowed(&self.search_text)
  }
}

#[cfg(test)]
mod tests {
  use super::*;

  #[test]
  fn displays_the_project_name() {
    let storage = Storage::new("/tmp/foo".into());

    let session = Session {
      directory: "/tmp/bar".into(),
      id: "ses_foo".into(),
      messages: Vec::new(),
      time: Time::default(),
      title: "foo".into(),
    };

    let item = SessionItem::new(&storage, &session);

    let display = item.display(DisplayContext::default());

    assert_eq!(display.spans[0].content, "foo");
    assert_eq!(display.spans[1].content, " ");
    assert_eq!(display.spans[2].content, "bar");
    assert_eq!(display.spans[2].style.fg, Some(DARK_GRAY));
  }
}