mariobassem
c27fcc6983
- Updated WireGuard client to use `wg-quick` instead of `sudo wg-quick`. - Improved error handling in WireGuard client. - Added `sudo` to `wg show` command for proper permissions. - Updated example `wireguard.vsh` script to reflect changes. - Added a new `wg0.conf` file for the WireGuard configuration. - Resolved the issue where the script wasn't finding the configuration file. Co-authored-by: mahmmoud.hassanein <mahmmoud.hassanein@gmail.com>
103 lines
2.1 KiB
V
103 lines
2.1 KiB
V
module wireguard
|
|
|
|
import freeflowuniverse.herolib.core.base
|
|
import freeflowuniverse.herolib.core.playbook
|
|
|
|
__global (
|
|
wireguard_global map[string]&WireGuard
|
|
wireguard_default string
|
|
)
|
|
|
|
/////////FACTORY
|
|
|
|
@[params]
|
|
pub struct ArgsGet {
|
|
pub mut:
|
|
name string
|
|
}
|
|
|
|
fn args_get(args_ ArgsGet) ArgsGet {
|
|
mut args := args_
|
|
if args.name == '' {
|
|
args.name = wireguard_default
|
|
}
|
|
if args.name == '' {
|
|
args.name = 'wireguard'
|
|
}
|
|
return args
|
|
}
|
|
|
|
pub fn get(args_ ArgsGet) !&WireGuard {
|
|
mut args := args_get(args_)
|
|
if args.name !in wireguard_global {
|
|
if args.name == 'wireguard' {
|
|
if !config_exists(args) {
|
|
if default {
|
|
println('When saving')
|
|
config_save(args)!
|
|
}
|
|
}
|
|
config_load(args)!
|
|
}
|
|
}
|
|
return wireguard_global[args.name] or {
|
|
println(wireguard_global)
|
|
panic('could not get config for wireguard with name:${args.name}')
|
|
}
|
|
}
|
|
|
|
fn config_exists(args_ ArgsGet) bool {
|
|
mut args := args_get(args_)
|
|
mut context := base.context() or { panic('bug') }
|
|
return context.hero_config_exists('wireguard', args.name)
|
|
}
|
|
|
|
fn config_load(args_ ArgsGet) ! {
|
|
mut args := args_get(args_)
|
|
mut context := base.context()!
|
|
mut heroscript := context.hero_config_get('wireguard', args.name)!
|
|
play(heroscript: heroscript)!
|
|
}
|
|
|
|
fn config_save(args_ ArgsGet) ! {
|
|
mut args := args_get(args_)
|
|
mut context := base.context()!
|
|
context.hero_config_set('wireguard', args.name, heroscript_default()!)!
|
|
}
|
|
|
|
fn set(o WireGuard) ! {
|
|
mut o2 := obj_init(o)!
|
|
wireguard_global[o.name] = &o2
|
|
wireguard_default = o.name
|
|
}
|
|
|
|
@[params]
|
|
pub struct PlayArgs {
|
|
pub mut:
|
|
heroscript string // if filled in then plbook will be made out of it
|
|
plbook ?playbook.PlayBook
|
|
reset bool
|
|
}
|
|
|
|
pub fn play(args_ PlayArgs) ! {
|
|
mut args := args_
|
|
|
|
if args.heroscript == '' {
|
|
args.heroscript = heroscript_default()!
|
|
}
|
|
mut plbook := args.plbook or { playbook.new(text: args.heroscript)! }
|
|
|
|
mut install_actions := plbook.find(filter: 'wireguard.configure')!
|
|
if install_actions.len > 0 {
|
|
for install_action in install_actions {
|
|
mut p := install_action.params
|
|
cfg_play(p)!
|
|
}
|
|
}
|
|
}
|
|
|
|
// switch instance to be used for wireguard
|
|
pub fn switch(name string) {
|
|
wireguard_default = name
|
|
}
|