diff options
| -rw-r--r-- | find_session.v | 194 |
1 files changed, 166 insertions, 28 deletions
diff --git a/find_session.v b/find_session.v index aebf2b4..c1250a7 100644 --- a/find_session.v +++ b/find_session.v @@ -2,8 +2,27 @@ import json import net.http import os import time +import term import term.ui as tui +struct StatesResponse { + states []State +} + +struct State { + state_id int + state_name string +} + +struct DistrictResponse { + districts []District +} + +struct District { + district_id int + district_name string +} + struct App { mut: tui &tui.Context = 0 @@ -16,8 +35,10 @@ struct VaccineAvailability { mut: date string available_sessions []Sessions - state_id int = 17 - district_id int = 296 + state_id int + district_id int + state State + district District } struct SessionResponse { @@ -74,7 +95,7 @@ fn frame(x voidptr) { } fn (mut app App) draw_header() { - app.tui.draw_text(3,2,"Vaccine Availability on ${app.data.date} | Total Centers : ${app.data.available_sessions.len}") + app.tui.draw_text(3,2, term.bold(term.bright_magenta("Vaccine Availability on ${app.data.date} | ${app.data.state.state_name},${app.data.district.district_name} | Total Centers : ${app.data.available_sessions.len}"))) app.tui.horizontal_separator(3) } @@ -83,15 +104,21 @@ fn (mut app App) draw_vaccine_availability() { start_x := 3 mut start_y := 5 if available_sessions.len == 0 { - app.tui.draw_text(app.tui.window_width/2 - 10,app.tui.window_height/2,"No Vaccine Available") + app.tui.draw_text(app.tui.window_width/2 - 10,app.tui.window_height/2, term.black(term.bg_white("No Vaccine Available"))) } else { for session in available_sessions { fee := match session.fee_type{ - 'Free' {'Free'} - 'Paid' { session.fee } + 'Free' {term.green('Free')} + 'Paid' {term.blue(session.fee) } else { 'N/A' } } - app.tui.draw_text(start_x,start_y,"${session.address} | Vaccine: ${session.vaccine} | Total Available: ${session.available_capacity} | Fee: ${fee}") + vaccine := match session.vaccine { + 'COVISHIELD' {term.bold(term.green(session.vaccine))} + 'COVAXIN' {term.bold(term.yellow(session.vaccine))} + else {session.vaccine} + } + capacity := term.green(session.available_capacity.str()) + app.tui.draw_text(start_x,start_y,"${session.name} | Vaccine: ${vaccine} | Total Available: ${capacity} | Fee: ${fee}") start_y += 2 } } @@ -102,29 +129,57 @@ fn init(x voidptr) { app.find_session(app.data.district_id, app.data.date) } - fn main() { - mut app := &App{} - app.tui = tui.init( - user_data: app - event_fn: event - frame_fn: frame - hide_cursor: false, - init_fn: init - ) - mut date_of_vaccine := time.now().add_days(app.idx).get_fmt_date_str(.hyphen,.ddmmyyyy) - if os.args.len ==2 { - date_of_vaccine = os.args[1] - } + setup_states() +} - println("Searching Vaccine Availability on $date_of_vaccine") - app.data.date = date_of_vaccine - app.tui.set_window_title("COWIN Vaccine Finder") - app.tui.run() ? +fn get_states() ?[]State { + fetch_config := http.FetchConfig{ + header: http.new_header( + key : .accept_language, + value: 'en-US' + ) + } + + url := 'https://cdn-api.co-vin.in/api/v2/admin/location/states' + resp := http.fetch(url, fetch_config) or { + term.clear() + println(term.fail_message('Failed to fetch States from the server')) + return error('Failed to fetch States from the server') + } + + states_data := json.decode(StatesResponse, resp.text) or { + term.clear() + println(term.fail_message('Failed to decode data from server')) + return error('Failed to decode data from server') + } + return states_data.states +} + +fn get_districts(state_id int) ?[]District { + fetch_config := http.FetchConfig{ + header: http.new_header( + key : .accept_language, + value: 'en-US' + ) + } + + url := 'https://cdn-api.co-vin.in/api/v2/admin/location/districts/$state_id' + resp := http.fetch(url, fetch_config) or { + term.clear() + println(term.fail_message('Failed to fetch Districts from the server')) + return error('Failed to fetch Districts from the server') + } + + district_data := json.decode(DistrictResponse, resp.text) or { + term.clear() + println(term.fail_message('Failed to decode data from server')) + return error('Failed to Decode data from the server') + } + return district_data.districts } fn (mut app App) find_session(district_id int,date string) { - app.is_loading = true fetch_config := http.FetchConfig{ header: http.new_header( key : .accept_language, @@ -140,14 +195,97 @@ fn (mut app App) find_session(district_id int,date string) { } session_data := json.decode(SessionResponse, resp.text) or { - println('failed to decode session json') + println(term.fail_message('Failed to decode data')) return } app.data.available_sessions = session_data.sessions.filter( it.available_capacity > 0 && it.min_age_limit<45) - app.is_loading = false } -fn setup_termui() { +fn setup_termui(state State, district District) { + mut app := &App{} + app.tui = tui.init( + user_data: app + event_fn: event + frame_fn: frame + hide_cursor: false, + init_fn: init + ) + mut date_of_vaccine := time.now().add_days(app.idx).get_fmt_date_str(.hyphen,.ddmmyyyy) + if os.args.len ==2 { + date_of_vaccine = os.args[1] + } + println("Searching Vaccine Availability on $date_of_vaccine") + app.data.date = date_of_vaccine + app.data.state = state + app.data.district = district + app.data.state_id = state.state_id + app.data.district_id = district.district_id + app.tui.set_window_title("COWIN Vaccine Finder") + app.tui.run() or {} +} + +fn setup_states() { + term.clear() + width, height := term.get_terminal_size() + println(term.header('Available State', '=')) + start_x := width/2 - 10 + mut start_y := 5 + mut selected_state := 0 + term.set_cursor_position(x: start_x, y: start_y) + + states := get_states() or { + return + } + + for i in 0 .. states.len { + println(term.bright_green('$i . ${states[i].state_name}')) + start_y += 1 + term.set_cursor_position(x: start_x, y: start_y) + } + term.set_cursor_position(x: 0, y: height) + + for { + if var := os.input_opt('Please select a State: ') { + if var.int() < 0 || var.int() >= states.len { + continue + } + setup_districts(states[var.int()]) + break + } + println('') + break + } + +} + +fn setup_districts(state State) { + term.clear() + width, height := term.get_terminal_size() + println(term.header('Available States', '=')) + start_x := width/2 - 10 + mut start_y := 5 + term.set_cursor_position(x: start_x, y: start_y) + districts := get_districts(state.state_id) or { + return + } + + for i in 0 .. districts.len { + println(term.bright_green('$i . ${districts[i].district_name}')) + start_y += 1 + term.set_cursor_position(x: start_x, y: start_y) + } + term.set_cursor_position(x: 0, y: height) + for { + if var := os.input_opt('Please select a District: ') { + if var.int() < 0 || var.int() >= districts.len { + continue + } + setup_termui(state,districts[var.int()]) + break + } + println('') + break + } }
\ No newline at end of file |