存储多个订阅

Author: cht

src/state.rs

@@ -25,14 +25,15 @@ pub fn run_app(terminal: &mut Terminal<MyBackend>) -> io::Result<()> {
     let mut appsub = AppSub::default();
     let informations = utils::start();
     if !informations.is_empty() {
-        appsub.messages = informations
+        appsub.subs = informations
             .iter()
             .map(|amessage| spider::remove_quotation(amessage.ps.clone()))
             .collect();
         appsub.stateoflist = true;
         appsub.state.select(Some(0));
         appsub.informations = informations;
     }
+    appsub.subscription = utils::get_subs();
     appsub.settings_input[0] = utils::start_v2core();
     let appbar = AppBar::new();
     let mut local_page = Page::SubScribe;

src/subscribe/appsub.rs

@@ -2,8 +2,7 @@ use super::app::*;
 use super::render::ui;
 use super::state::subscribe_state;
 use crate::spider;
-use crate::state::MyBackend;
-use crate::state::IFEXIT;
+use crate::state::{MyBackend, IFEXIT};
 use std::io;
 use tui::widgets::ListState;
 use tui::Terminal;
@@ -18,25 +17,29 @@ pub(crate) enum InputMode {
 /// App holds the state of the application
 pub struct AppSub {
     /// Current value of the input box
+    // search bar
     pub input: String,
+    // settings , include coresetting and subscribe setting
     pub settings_input: Vec<String>,
     /// Current input mode
     pub(crate) input_mode: InputMode,
-    /// History of recorded messages
-    pub messages: Vec<String>,
+    // History of recorded subs
+    // subs's names
+    pub subs: Vec<String>,
     pub state: ListState,
     pub index_subscription: ListState,
     pub index_settings: usize,
     pub stateoflist: bool,
     pub show_popup: bool,
+    // subscribes's information
     pub informations: Vec<spider::Information>,
     pub subscription: Vec<String>,
 }
 impl AppSub {
     pub fn next(&mut self) {
         let i = match self.state.selected() {
             Some(i) => {
-                if i >= self.messages.len() - 1 {
+                if i >= self.subs.len() - 1 {
                     0
                 } else {
                     i + 1
@@ -51,7 +54,7 @@ impl AppSub {
         let i = match self.state.selected() {
             Some(i) => {
                 if i == 0 {
-                    self.messages.len() - 1
+                    self.subs.len() - 1
                 } else {
                     i - 1
                 }
@@ -110,7 +113,7 @@ impl Default for AppSub {
             input: String::new(),
             settings_input: vec![String::new(), String::new()],
             input_mode: InputMode::Normal,
-            messages: Vec::new(),
+            subs: Vec::new(),
             state: ListState::default(),
             index_subscription: ListState::default(),
             index_settings: 0,

src/subscribe/render.rs

@@ -88,16 +88,16 @@ pub(crate) fn ui<B: Backend>(f: &mut Frame<B>, app: &mut AppSub) {
         .constraints([Constraint::Percentage(40), Constraint::Percentage(60)].as_ref())
         .split(chunks[2]);
 
-    let messages: Vec<ListItem> = app
-        .messages
+    let subs: Vec<ListItem> = app
+        .subs
         .iter()
         .enumerate()
         .map(|(i, m)| {
             let content = vec![Spans::from(Span::raw(format!("{}: {}", i, m)))];
             ListItem::new(content)
         })
         .collect();
-    let messages = List::new(messages)
+    let subs = List::new(subs)
         .block(Block::default().borders(Borders::ALL).title("List"))
         .highlight_style(
             Style::default()
@@ -106,7 +106,7 @@ pub(crate) fn ui<B: Backend>(f: &mut Frame<B>, app: &mut AppSub) {
         )
         .highlight_symbol(">> ");
     // popup wiget
-    f.render_stateful_widget(messages, bottom_chunks[0], &mut app.state);
+    f.render_stateful_widget(subs, bottom_chunks[0], &mut app.state);
     //let block : Box<dyn Widget> = {
     if let Some(a) = app.state.selected() {
         let list = app.informations[a].information_to_list();

src/subscribe/state.rs

@@ -37,15 +37,15 @@ pub(crate) fn subscribe_state(app: &mut AppSub) -> io::Result<IFEXIT> {
                         storge.push('[');
                         storge.push('\n');
                         if !list[0].is_empty() {
-                            //app.messages = list[0].clone();
+                            //app.subs = list[0].clone();
                             app.stateoflist = true;
                             app.state.select(Some(0));
                             for alist in &list[0] {
                                 let information = spider::Information::new(alist.to_string());
                                 app.informations.push(information.clone());
                                 storge.push_str(information.get_the_json_node().as_str());
                             }
-                            app.messages = app
+                            app.subs = app
                                 .informations
                                 .iter()
                                 .map(|ainformation| {
@@ -61,7 +61,7 @@ pub(crate) fn subscribe_state(app: &mut AppSub) -> io::Result<IFEXIT> {
                             .unwrap_or_else(|err| panic!("err {}", err));
                     }
 
-                    //app.messages.push(app.input.drain(..).collect());
+                    //app.subs.push(app.input.drain(..).collect());
                 }
                 KeyCode::Char(c) => {
                     app.input.push(c);
@@ -133,6 +133,22 @@ pub(crate) fn subscribe_state(app: &mut AppSub) -> io::Result<IFEXIT> {
                         ),
                     )
                     .unwrap_or_else(|err| panic!("{}", err));
+                    let mut subscribe_json: String = "[\n".to_string();
+                    for asub in &app.subscription {
+                        subscribe_json.push_str(&format!(
+                            "
+    {{
+        \"url\": \"{}\"
+    }},\n",
+                            asub
+                        ));
+                    }
+                    subscribe_json.pop();
+                    subscribe_json.pop();
+                    subscribe_json.push_str("\n]");
+                    utils::create_json_file(utils::Save::Subscribes, subscribe_json)
+                        .unwrap_or_else(|err| panic!("{}", err));
+                    //    .collect();
                 }
                 _ => {}
             },
@@ -175,6 +191,16 @@ pub(crate) fn subscribe_state(app: &mut AppSub) -> io::Result<IFEXIT> {
             InputMode::SubscriptView => match key.code {
                 KeyCode::Up => app.previous_sub(),
                 KeyCode::Down => app.next_sub(),
+                KeyCode::Char('d') => {
+                    app.subscription.remove(app.index_subscription.selected().unwrap());
+                    if app.subscription.is_empty() {
+                        app.unselect_sub();
+                        app.input_mode = InputMode::Popup;
+                    } else {
+                        app.index_subscription.select(Some(0));
+                    }
+                    //app.settings_input[app.index_settings].push(c);
+                }
                 KeyCode::Esc => {
                     app.index_settings = 0;
                     app.unselect_sub();

src/utils.rs

@@ -10,6 +10,7 @@ pub enum Save {
     Storage,
     Running,
     V2ray,
+    Subscribes,
 }
 fn create_storage_before() {
     let home = env::var("HOME").unwrap();
@@ -21,6 +22,7 @@ pub fn create_json_file(save: Save, input: String) -> Result<()> {
         Save::Storage => format!("{}/.config/tv2ray/storage.json", home),
         Save::Running => format!("{}/.config/tv2ray/running.json", home),
         Save::V2ray => format!("{}/.config/tv2ray/v2core.json", home),
+        Save::Subscribes => format!("{}/.config/tv2ray/subscribes.json", home),
     };
     let path = Path::new(location.as_str());
     let mut file = File::create(&path)?;
@@ -34,6 +36,7 @@ fn get_json(save: Save) -> Result<String> {
         Save::Storage => format!("{}/.config/tv2ray/storage.json", home),
         Save::Running => format!("{}/.config/tv2ray/running.json", home),
         Save::V2ray => format!("{}/.config/tv2ray/v2core.json", home),
+        Save::Subscribes => format!("{}/.config/tv2ray/subscribes.json", home),
     };
     let mut file = File::open(location)?;
     let mut output = String::new();
@@ -54,6 +57,28 @@ pub fn start_v2core() -> String {
     let message_pre = v["v2core"].to_string();
     crate::spider::remove_quotation(message_pre)
 }
+pub fn get_subs() -> Vec<String> {
+    create_storage_before();
+    let messages = match get_json(Save::Subscribes) {
+        Ok(output) => output,
+        Err(_) => {
+            create_json_file(Save::Storage, "[]".to_string())
+                .unwrap_or_else(|err| panic!("{}", err));
+            "[]".to_string()
+        }
+    };
+    let mut subscribes = Vec::new();
+    let v: Value = serde_json::from_str(messages.as_str()).unwrap();
+    let mut index = 0;
+    while v[index] != Value::Null {
+        let sub = v[index]["url"].to_string();
+        let length = sub.len();
+        let sub = (&sub[1..length - 1]).to_string();
+        subscribes.push(sub);
+        index += 1;
+    }
+    subscribes
+}
 pub fn start() -> Vec<Information> {
     create_storage_before();
     let messages = match get_json(Save::Storage) {
@@ -69,8 +94,8 @@ pub fn start() -> Vec<Information> {
     let mut index = 0;
     while v[index] != Value::Null {
         let the_url = v[index]["url"].to_string();
-        let lenghth = the_url.len();
-        let instore = &the_url[1..lenghth - 1];
+        let length = the_url.len();
+        let instore = &the_url[1..length - 1];
         informations.push(Information {
             urls: instore.to_string(),
             func: v[index]["func"].to_string(),