Merge pull request #40 from heroesofcode/feature/organize-code-main

Separate block into another file

Author: joaolfp

src/block.rs

@@ -0,0 +1,72 @@
+use ratatui::{
+	layout::Alignment,
+	style::{Color, Style},
+	widgets::{Block, Borders, Paragraph, Wrap}
+};
+
+pub fn add_info_text() -> Paragraph<'static> {
+	let text = "🌐 🛢️ Command Line Tools to check for SQL Injection vulnerability.\n👨‍💻 https://github.com/heroesofcode/inject-sql";
+
+	let info_text = Paragraph::new(text)
+		.block(Block::default().borders(Borders::ALL))
+		.style(Style::default().fg(Color::Green));
+
+	return info_text;
+}
+
+pub fn add_url_block(url: &str) -> Paragraph<'_> {
+	let url_block = Paragraph::new(&url[..])
+		.block(
+			Block::default()
+				.title("Enter the URL")
+				.borders(Borders::ALL),
+		)
+		.style(Style::default().fg(Color::White))
+		.wrap(Wrap { trim: true });
+
+	return url_block;
+}
+
+pub fn add_type_payload_text() -> Paragraph<'static> {
+	let text = "\n1 - classical 1\n2 - classical 2\n3 - time-based\n4 - blind 1\n5 - blind 2\n6 - boolean 1\n7 - boolean 2\n8 - Get Database\n";
+
+	let type_payload_text = Paragraph::new(text).style(Style::default().fg(Color::Green));
+
+	return type_payload_text;
+}
+
+pub fn add_payload_block(payload_type: &str) -> Paragraph<'_> {
+	let payload_block = Paragraph::new(&payload_type[..])
+		.block(
+			Block::default()
+				.title("Enter the payload type")
+				.borders(Borders::all()),
+		)
+		.style(Style::default().fg(Color::White))
+		.wrap(Wrap { trim: true });
+
+	return payload_block;
+}
+
+pub fn add_result_block(result_text: &str) -> Paragraph<'_> {
+	let result_block = Paragraph::new(&result_text[..])
+		.alignment(Alignment::Center)
+		.block(
+			Block::default()
+				.title("Result")
+				.borders(Borders::TOP)
+				.style(Style::default().fg(Color::Yellow)),
+		)
+		.wrap(Wrap { trim: true });
+
+	return result_block;
+}
+
+pub fn add_help_text() -> Paragraph<'static> {
+	let text = "Use TAB to switch between fields. Press ENTER to validate.";
+
+	let help_text = Paragraph::new(text)
+		.style(Style::default().fg(Color::White));
+
+	return help_text;
+}

src/main.rs

@@ -4,9 +4,7 @@ use tokio;
 
 use ratatui::{
 	backend::CrosstermBackend,
-	layout::{Alignment, Constraint, Direction, Layout},
-	style::{Color, Style},
-	widgets::{Block, Borders, Paragraph, Wrap},
+	layout::{Constraint, Direction, Layout},
 	Terminal,
 };
 
@@ -19,7 +17,10 @@ use crossterm::{
 use std::io::stdout;
 
 mod response;
+mod block;
+
 use response::validation_exist_sql_injection;
+use block::*;
 
 #[tokio::main]
 async fn main() -> Result<(), io::Error> {
@@ -115,73 +116,6 @@ async fn run_app<B: ratatui::backend::Backend>(terminal: &mut Terminal<B>) -> io
 	}
 }
 
-fn add_info_text() -> Paragraph<'static> {
-	let text = "🌐 🛢️ Command Line Tools to check for SQL Injection vulnerability.\n👨‍💻 https://github.com/heroesofcode/inject-sql";
-
-	let info_text = Paragraph::new(text)
-		.block(Block::default().borders(Borders::ALL))
-		.style(Style::default().fg(Color::Green));
-
-	return info_text;
-}
-
-fn add_url_block(url: &str) -> Paragraph<'_> {
-	let url_block = Paragraph::new(&url[..])
-		.block(
-			Block::default()
-				.title("Enter the URL")
-				.borders(Borders::ALL),
-		)
-		.style(Style::default().fg(Color::White))
-		.wrap(Wrap { trim: true });
-
-	return url_block;
-}
-
-fn add_type_payload_text() -> Paragraph<'static> {
-	let text = "\n1 - classical 1\n2 - classical 2\n3 - time-based\n4 - blind 1\n5 - blind 2\n6 - boolean 1\n7 - boolean 2\n8 - Get Database\n";
-
-	let type_payload_text = Paragraph::new(text).style(Style::default().fg(Color::Green));
-
-	return type_payload_text;
-}
-
-fn add_payload_block(payload_type: &str) -> Paragraph<'_> {
-	let payload_block = Paragraph::new(&payload_type[..])
-		.block(
-			Block::default()
-				.title("Enter the payload type")
-				.borders(Borders::all()),
-		)
-		.style(Style::default().fg(Color::White))
-		.wrap(Wrap { trim: true });
-
-	return payload_block;
-}
-
-fn add_result_block(result_text: &str) -> Paragraph<'_> {
-	let result_block = Paragraph::new(&result_text[..])
-		.alignment(Alignment::Center)
-		.block(
-			Block::default()
-				.title("Result")
-				.borders(Borders::TOP)
-				.style(Style::default().fg(Color::Yellow)),
-		)
-		.wrap(Wrap { trim: true });
-
-	return result_block;
-}
-
-fn add_help_text() -> Paragraph<'static> {
-	let text = "Use TAB to switch between fields. Press ENTER to validate.";
-
-	let help_text = Paragraph::new(text)
-		.style(Style::default().fg(Color::White));
-
-	return help_text;
-}
-
 async fn show_result(url: &str, payload_type: &str) -> String {
     if !url.is_empty() && !payload_type.is_empty() {
         match validation_exist_sql_injection(&url, &payload_type).await {