1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
import eduvpncommon.main as eduvpn
import webbrowser
import json
import sys
# Asks the user for a profile index
# It loops up until a valid input is given
def ask_profile_input(total: int) -> int:
profile_index = None
while profile_index is None:
try:
profile_index = int(
input("Please select a profile by inputting a number (e.g. 1): ")
)
if (profile_index > total) or (profile_index < 1):
print("Invalid profile range")
profile_index = None
except ValueError:
print("Please enter a valid input")
# The profile is one based, move to zero based input
return profile_index - 1
# Sets up the callbacks using the provided class
def setup_callbacks(_eduvpn: eduvpn.EduVPN) -> None:
# The callback that starst OAuth
# It needs to open the URL in the web browser
@_eduvpn.event.on("OAuth_Started", eduvpn.StateType.Enter)
def oauth_initialized(old_state: str, url: str) -> None:
print(f"Got OAuth URL {url}, old state: {old_state}")
webbrowser.open(url)
@_eduvpn.event.on("Ask_Location", eduvpn.StateType.Enter)
def ask_location(old_state: str, locations: str):
print("Locations: ", locations)
_eduvpn.set_secure_location("NL")
# The callback which asks the user for a profile
@_eduvpn.event.on("Ask_Profile", eduvpn.StateType.Enter)
def ask_profile(old_state: str, profiles: str):
print(
"Multiple profiles found, you need to select a profile, old state: {old_state}"
)
# Parse the profiles as JSON
data = json.loads(profiles)
# Get a lits of profiles
profile_strings = [x["profile_id"] for x in data["info"]["profile_list"]]
total_profiles = len(profile_strings)
# Create a list of the strings to standard output
for idx, profile in enumerate(profile_strings):
print(f"{idx+1}. {profile}")
# Get the profile index from the user
profile_index = ask_profile_input(total_profiles)
# Set the profile with the index
_eduvpn.set_profile(profile_strings[profile_index])
# The main entry point
if __name__ == "__main__":
_eduvpn = eduvpn.EduVPN("org.eduvpn.app.linux", "configs")
setup_callbacks(_eduvpn)
# Register with the eduVPN-common library
try:
_eduvpn.register(debug=True)
except Exception as e:
print("Failed registering:", e)
server = input(
"Which server (Custom/Institute Access) do you want to connect to? (e.g. https://eduvpn.example.com): "
)
# Ensure we have a valid http prefix
if not server.startswith("http"):
# https by default
server = "https://" + server
# Get a Wireguard/OpenVPN config
try:
config, config_type = _eduvpn.get_config_custom_server(server)
print(f"Got a config with type: {config_type} and contents:\n{config}")
except Exception as e:
print("Failed to connect:", e)
# Save and exit
_eduvpn.deregister()
sys.exit(1)
# Set the internal FSM state to connected
try:
_eduvpn.set_connected()
except Exception as e:
print("Failed to set connected:", e)
# Save and exit
_eduvpn.deregister()
|
