mrfluffy/kami
1
0
Fork 0
mirror of https://github.com/mrfluffy-dev/kami.git synced 2026-01-17 12:50:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

added basic image support

Browse Source
This commit is contained in:
Zastian Pretorius
2023-01-11 00:50:19 +00:00
parent 26a8cb5e77
commit 4d0112ed86
4 changed files with 649 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
src/anime/anime.rs
View File

@@ -2,7 +2,7 @@ use crate::{
get_an_progress, get_anime_id, get_user_anime_progress, update_anime_progress,
write_an_progress,
};
use crate::{get_anime_link, get_animes};
use crate::{get_anime_link, get_animes, get_image};
use crate::{open_cast, open_video};
use crossterm::{
@@ -20,6 +20,7 @@ use tui::{
Frame, Terminal,
};
use unicode_width::UnicodeWidthStr;
use viuer::{print_from_file, terminal_size, Config};
use super::scraper::get_anime_info;
@@ -83,7 +84,7 @@ impl<T> StatefulList<T> {
struct App {
/// Current value of the input box
input: String,
animes: (Vec<String>, Vec<String>),
animes: (Vec<String>, Vec<String>, Vec<String>),
/// Current input mode
input_mode: InputMode,
/// History of recorded messages
@@ -102,7 +103,7 @@ impl<'a> App {
fn default() -> App {
App {
input: String::new(),
animes: (Vec::new(), Vec::new()),
animes: (Vec::new(), Vec::new(), Vec::new()),
input_mode: InputMode::Normal,
messages: StatefulList::with_items(Vec::new()),
title: String::new(),
@@ -134,7 +135,19 @@ pub fn anime_ui(
app.token = token;
app.provider = provider;
app.cast = cast;
let res = run_app(&mut terminal, app);
let (term_width, _term_height) = terminal_size();
let width = 50;
let height = 24;
let conf = Config {
// set dimensions
x: (term_width - 33),
y: 2,
width: Some((width) as u32),
height: Some((height) as u32),
restore_cursor: true,
..Default::default()
};
let res = run_app(&mut terminal, app, conf);
// restore terminal
disable_raw_mode()?;
@@ -152,8 +165,15 @@ pub fn anime_ui(
Ok(())
}
fn run_app<B: Backend>(terminal: &mut Terminal<B>, mut app: App) -> io::Result<()> {
fn run_app<B: Backend>(terminal: &mut Terminal<B>, mut app: App, conf: Config) -> io::Result<()> {
let mut ep_select = false;
fn change_image(conf: &Config, app: &App) {
let selected = app.messages.state.selected();
let image_url = app.animes.2[selected.unwrap()].clone();
get_image(&image_url);
print_from_file("/home/mrfluffy/.config/kami/temp.png", &conf)
.expect("Image printing failed.");
}
loop {
terminal.draw(|f| ui(f, &mut app))?;
@@ -168,10 +188,22 @@ fn run_app<B: Backend>(terminal: &mut Terminal<B>, mut app: App) -> io::Result<(
}
KeyCode::Left => app.messages.unselect(),
KeyCode::Char('h') => app.messages.unselect(),
KeyCode::Down => app.messages.next(),
KeyCode::Char('j') => app.messages.next(),
KeyCode::Up => app.messages.previous(),
KeyCode::Char('k') => app.messages.previous(),
KeyCode::Down => {
app.messages.next();
change_image(&conf, &app);
}
KeyCode::Char('j') => {
app.messages.next();
change_image(&conf, &app)
}
KeyCode::Up => {
app.messages.previous();
change_image(&conf, &app);
}
KeyCode::Char('k') => {
app.messages.previous();
change_image(&conf, &app)
}
//if KeyCode::Enter => {
KeyCode::Enter => {
if ep_select == false {
@@ -286,12 +318,18 @@ fn ui<B: Backend>(f: &mut Frame<B>, app: &mut App) {
.as_ref(),
)
.split(f.size());
let block = Block::default()
.borders(Borders::ALL)
.title("kami")
.border_type(BorderType::Rounded);
f.render_widget(block, f.size());
let top_chunks = Layout::default()
.direction(Direction::Horizontal)
.constraints([Constraint::Percentage(50), Constraint::Percentage(50)].as_ref())
.split(chunks[0]);
let (msg, style) = match app.input_mode {
InputMode::Normal => (
vec![
@@ -325,7 +363,12 @@ fn ui<B: Backend>(f: &mut Frame<B>, app: &mut App) {
})
.collect();
let messages = List::new(messages)
.block(Block::default().borders(Borders::ALL).title("list"))
.block(
Block::default()
.borders(Borders::ALL)
.title("list")
.border_type(BorderType::Rounded),
)
.style(Style::default().fg(Color::White))
.highlight_style(
Style::default()
@@ -333,7 +376,12 @@ fn ui<B: Backend>(f: &mut Frame<B>, app: &mut App) {
.add_modifier(Modifier::BOLD),
)
.highlight_symbol(">>");
f.render_stateful_widget(messages, chunks[0], &mut app.messages.state);
f.render_stateful_widget(messages, top_chunks[0], &mut app.messages.state);
let block = Block::default()
.borders(Borders::ALL)
.title("info")
.border_type(BorderType::Rounded);
f.render_widget(block, top_chunks[1]);
let mut text = Text::from(Spans::from(msg));
text.patch_style(style);

21
src/anime/scraper.rs
View File

@@ -1,6 +1,9 @@
use isahc::config::Configurable;
use isahc::{ReadResponseExt, Request, RequestExt};
use regex::Regex;
use std::fs::File;
use std::io::prelude::*;
//use serde_json::json;
pub fn get_anime_html(url: &str) -> String {
@@ -30,7 +33,7 @@ pub fn get_post(id: &str) -> String {
resp
}
pub fn get_animes(query: String) -> (Vec<String>, Vec<String>) {
pub fn get_animes(query: String) -> (Vec<String>, Vec<String>, Vec<String>) {
let query = query.replace(" ", "+");
let html = get_anime_html(&format!("https://yugen.to/search/?q={}", query));
let re = Regex::new(r#"href="(/anime[^"]*)""#).unwrap();
@@ -43,7 +46,12 @@ pub fn get_animes(query: String) -> (Vec<String>, Vec<String>) {
for cap in re.captures_iter(&html) {
animes_names.push(cap[1].to_string());
}
(animes_links, animes_names)
let re = Regex::new(r#"data-src="([^"]*)"#).unwrap();
let mut animes_images = Vec::new();
for cap in re.captures_iter(&html) {
animes_images.push(cap[1].to_string());
}
(animes_links, animes_names, animes_images)
}
pub fn get_anime_info(url: &str) -> (i32, u16) {
@@ -77,3 +85,12 @@ pub fn get_anime_link(url: &str, episode: u64) -> String {
//return the link
link.to_string()
}
pub fn get_image(url: &str) {
let url = url;
let mut response = isahc::get(url).unwrap();
let mut buffer = Vec::new();
response.copy_to(&mut buffer).unwrap();
let mut file = File::create("/home/mrfluffy/.config/kami/temp.png").unwrap();
file.write_all(&buffer).unwrap();
}