File Coverage

blib/lib/Net/OpenVPN/TrayIcon.pm

Criterion	Covered	Total	%
statement	10	12	83.3
branch			n/a
condition			n/a
subroutine	4	4	100.0
pod			n/a
total	14	16	87.5

package Net::OpenVPN::TrayIcon;

use 5.010;
use warnings;
use Moo;
use Gtk2 '-init';
use Gtk2::TrayIcon;
use MIME::Base64;
use Time::HiRes 'usleep';
use Data::Section -setup;
use POSIX ':sys_wait_h';

has icons    => ( is => 'rw' );
has config   => ( is => 'rw' );
has dispatch => ( is => 'rw' );
has gtk_icon => ( is => 'rw' );
has gtk_menu => ( is => 'rw' );
has gtk_tray_icon => ( is => 'rw' );
has gtk_tooltip => ( is => 'rw' );

our $this;

sub _sig_usr1_handler {
    my ($sig) = @_;
    $this->dispatch->{set_icon}->('active') if $sig ~~ 'USR1';
    return 1;
}

sub _sig_usr2_handler {
    my ($sig) = @_;
    $this->dispatch->{set_icon}->('inactive') if $sig ~~ 'USR2';
    return 1;
}

sub BUILD {
    my ($self) = @_;

    if ($self->_first_run){
        my %icons;
        $icons{default} = ${$self->section_data('icon_default')};
        $icons{active} = ${$self->section_data('icon_active')};
        $icons{inactive} = ${$self->section_data('icon_inactive')};

        my $default_config = ${$self->section_data('default_config')};

        mkdir $ENV{HOME} . '/.ovpntray' or die "Could not create config directory: $!";

        open my $fh, '>', $ENV{HOME} . '/.ovpntray/config';
        print $fh $default_config;
        close $fh;

        for (keys %icons){
            open my $bfh, '>', $ENV{HOME} . '/.ovpntray/icon_' . $_ . '.png';
            binmode $bfh;
            print $bfh decode_base64($icons{$_});
            close $bfh;
        }
    }

    $self->_build_config;
    $self->_build_dispatch_table;
    $self->_build_icons;
    $self->_build_tray;
    $self->_build_this;

    $SIG{USR1} = \&_sig_usr1_handler;
    $SIG{USR2} = \&_sig_usr2_handler;

    return 1;
}

sub _build_this {
    my ($self) = @_;
    $this = $self;

    return 1;
}

sub run {
    my ($self) = @_;

    Gtk2->main;
}

sub click {
    my ($self, $gtk_evnt_box, $gtk_evnt_button) = @_;

    given ($gtk_evnt_button->button){
        when (1){   # left click
            $self->menu;
        }
        when (2){   # middle click
            $self->menu;
        }
        when (3){   # right click
            $self->menu;
        }
        default {
        }
    }

    return 1;
}

sub config_view {
    my ($self) = @_;

    # main config window
    my $gtk_window = Gtk2::Window->new('toplevel');
    $gtk_window->set_title('Config');
    $gtk_window->set_position('center');

    # vbox container (rows, columns, homogeneous)
    my $gtk_table = Gtk2::Table->new( 3, 1, 0 );

    # notebook
    my $gtk_notebook = Gtk2::Notebook->new;
    $gtk_notebook->set_tab_pos('top');

    # first page
    my $gtk_vbox_page_1 = Gtk2::VBox->new( 0, 1 );
}

sub menu {
    my ($self) = @_;

    my $gtk_menu = Gtk2::Menu->new();
    my $gtk_menu_start = Gtk2::ImageMenuItem->new_with_label('start VPN');
    $gtk_menu_start->signal_connect(
        activate => sub {
            system 'sudo ' . $self->config->{start_cmnd};
            $self->dispatch->{set_icon}->('default');
            $self->dispatch->{set_tip}->('OpenVPN checking connection..');
            $self->dispatch->{active_if_running}->();
        },
    );

    my $gtk_menu_stop = Gtk2::ImageMenuItem->new_with_label('stop VPN');
    $gtk_menu_stop->signal_connect(
        activate => sub {
            system 'sudo ' . $self->config->{stop_cmnd};
            $self->dispatch->{set_icon}->('default');
            $self->dispatch->{set_tip}->('OpenVPN checking connection..');
            $self->dispatch->{inactive_if_not_running}->();
        },
    );

    $gtk_menu->add($gtk_menu_start);
    $gtk_menu->add($gtk_menu_stop);
    $gtk_menu->show_all;

    $gtk_menu->popup( undef, undef, undef, undef, 0, 0 );
    $self->gtk_menu($gtk_menu);

    return 1;
}

sub _build_dispatch_table {
    my ($self) = @_;

    my %table = (
        set_icon => sub {
            my ($type) = @_;
            $self->gtk_icon->set_from_pixbuf($self->icons->{$type});
            return 1;
        },
        set_tip => sub {
            my ($tip) = @_;
            $self->gtk_tooltip->set_tip( $self->gtk_tray_icon, $tip);
            return 1;
        },
        active_if_running => sub {
            my $parent_pid = $$;
            my $child_pid = fork;

            if ($child_pid == 0){ # the child
                if ($self->_vpn_is_running(200)){
                    if ($self->_check_if_connected(10000)){
                        kill USR1 => $parent_pid;
                    }
                }
                else {
                    kill USR2 => $parent_pid;
                }

                POSIX::_exit(0);
            }
            else {
                # keep going
            }
        },
        inactive_if_not_running => sub {
            my $parent_pid = $$;
            my $child_pid = fork;

            if ($child_pid == 0){ # the child
                if ($self->_vpn_is_not_running(200)){
                    kill USR2 => $parent_pid;
                }
                else {
                    kill USR1 => $parent_pid;
                }

                POSIX::_exit(0);
            }
            else {
                # keep going
            }
        },

    );

    $self->dispatch({ %table });

    return 1;
}

sub _build_tray {
    my ($self) = @_;

    my $gtk_tooltip = Gtk2::Tooltips->new;
    my $gtk_trayicon = Gtk2::TrayIcon->new('VPN Tray');
    my $gtk_icon;
    if ($self->_vpn_is_running(10)){
        $gtk_icon = Gtk2::Image->new_from_pixbuf($self->icons->{'active'});
        $gtk_tooltip->set_tip( $gtk_trayicon, 'OpenVPN started');
    }
    else {
        $gtk_icon = Gtk2::Image->new_from_pixbuf($self->icons->{'inactive'});
        $gtk_tooltip->set_tip( $gtk_trayicon, 'OpenVPN stopped');
    }
    my $gtk_eventbox = Gtk2::EventBox->new;
    $gtk_eventbox->add($gtk_icon);


    $gtk_trayicon->add($gtk_eventbox);

    $self->gtk_icon($gtk_icon);
    $self->gtk_tooltip($gtk_tooltip);
    $self->gtk_tray_icon($gtk_trayicon);

    my $click_callback = sub {
        $self->click(@_);
    };

    $gtk_eventbox->signal_connect( 'button_press_event', $click_callback );

    $gtk_trayicon->show_all;

    return 1;
}

sub _vpn_is_running {
    my ($self, $timeout) = @_;
    return $self->_check_if_running(1, $timeout);
}

sub _vpn_is_not_running {
    my ($self, $timeout) = @_;
    return $self->_check_if_running(0, $timeout);
}

sub _check_if_running {
    my ($self, $condition, $timeout) = @_;

    while ($timeout){
        my $ovpn_pid = qx[pgrep openvpn];
        if ($condition){
            if ($ovpn_pid){
                return 1;
            }
        }
        else {
            if (!$ovpn_pid){
                return 1;
            }
        }
        Time::HiRes::usleep 100;
        $timeout--;
    }

    return 0;
}

sub _check_if_connected { # check if tunnel_interface appears in routes
    my ($self, $timeout) = @_;

    my $tunnel = $self->config->{tunnel_interface};
    while ($timeout){
        my ($routes) = qx[/sbin/ip r]; # first line -> default route

        return 1 if $routes ~~ /default\svia\s\d{0,3}\.\d{0,3}\.\d{0,3}\.\d{0,3}\sdev\s$tunnel/;

        Time::HiRes::usleep 100;
        $timeout--;
    }

    return 0;
}

sub _first_run {
    my ($self) = @_;
    return -f $ENV{HOME} . '/.ovpntray/config' ? 0 : 1;
}

sub _build_config {
    my ($self) = @_;

    my %config;
    open my $fh, '<', $ENV{HOME} . '/.ovpntray/config';
    while (<$fh>){
        my ($key, $value) = split ':', $_;
        chomp $key;
        chomp $value;
        $config{$key} = $value;
    }
    close $fh;

    $self->config({%config});

    return 1;
}

sub _build_icons {
    my ($self) = @_;

    my %icons;
    $icons{'default'} = Gtk2::Gdk::Pixbuf->new_from_file_at_scale(
        $ENV{HOME} . '/.ovpntray/' . $self->config->{icon_file},
        $self->config->{icon_size}, # width
        $self->config->{icon_size}, # height
        1,  # keep aspect
    );
    $icons{'active'} = Gtk2::Gdk::Pixbuf->new_from_file_at_scale(
        $ENV{HOME} . '/.ovpntray/' . $self->config->{icon_file_active},
        $self->config->{icon_size}, # width
        $self->config->{icon_size}, # height
        1,  # keep aspect
    );
    $icons{'inactive'} = Gtk2::Gdk::Pixbuf->new_from_file_at_scale(
        $ENV{HOME} . '/.ovpntray/' . $self->config->{icon_file_inactive},
        $self->config->{icon_size}, # width
        $self->config->{icon_size}, # height
        1,  # keep aspect
    );

    $self->icons({%icons});

    return 1;
}

1;

__DATA__