1use gpui::SharedString;
 2
 3use crate::context::{Context, ContextId, ContextKind};
 4
 5pub struct ContextStore {
 6    context: Vec<Context>,
 7    next_context_id: ContextId,
 8}
 9
10impl ContextStore {
11    pub fn new() -> Self {
12        Self {
13            context: Vec::new(),
14            next_context_id: ContextId(0),
15        }
16    }
17
18    pub fn context(&self) -> &Vec<Context> {
19        &self.context
20    }
21
22    pub fn drain(&mut self) -> Vec<Context> {
23        self.context.drain(..).collect()
24    }
25
26    pub fn clear(&mut self) {
27        self.context.clear();
28    }
29
30    pub fn insert_context(
31        &mut self,
32        kind: ContextKind,
33        name: impl Into<SharedString>,
34        text: impl Into<SharedString>,
35    ) {
36        self.context.push(Context {
37            id: self.next_context_id.post_inc(),
38            name: name.into(),
39            kind,
40            text: text.into(),
41        });
42    }
43
44    pub fn remove_context(&mut self, id: &ContextId) {
45        self.context.retain(|context| context.id != *id);
46    }
47}