158 lines
4.7 KiB
Haskell
158 lines
4.7 KiB
Haskell
--------------------------------------------------------------------------------
|
|
-- rofi-evpn - a prompt to dicsonnect/connect with express VPN
|
|
--
|
|
|
|
module Main (main) where
|
|
|
|
import Data.Maybe
|
|
import RIO
|
|
import qualified RIO.Text as T
|
|
import Rofi.Command
|
|
import System.Environment
|
|
import System.Process
|
|
|
|
main :: IO ()
|
|
main = getArgs >>= runPrompt
|
|
|
|
runPrompt :: [String] -> IO ()
|
|
runPrompt args = do
|
|
servers <- getServers
|
|
maybe (return ()) run servers
|
|
where
|
|
run (VPNStatus connected servers) = do
|
|
let d = getDisconnectAction <$> connected
|
|
let cs = fmap (getConnectAction connected) servers
|
|
runRofi (RofiVPNConf $ fmap T.pack args) $
|
|
emptyMenu
|
|
{ groups =
|
|
[ untitledGroup $ toRofiActions $ maybeToList d
|
|
, untitledGroup $ toRofiActions cs
|
|
]
|
|
, prompt = Just "Select Action"
|
|
}
|
|
|
|
newtype RofiVPNConf = RofiVPNConf [T.Text]
|
|
|
|
instance HasRofiConf RofiVPNConf where
|
|
defArgs (RofiVPNConf as) = as
|
|
|
|
type VPNAction = RofiAction RofiVPNConf
|
|
|
|
type VPNServer = (T.Text, T.Text)
|
|
|
|
data VPNStatus = VPNStatus (Maybe T.Text) [VPNServer] deriving (Show)
|
|
|
|
getServers :: IO (Maybe VPNStatus)
|
|
getServers = do
|
|
running <- daemonIsRunning
|
|
if running
|
|
then Just <$> getStatus
|
|
else notify IconError "ExpressVPN daemon not running" >> return Nothing
|
|
|
|
getStatus :: IO VPNStatus
|
|
getStatus = do
|
|
connected <- getConnectedServer
|
|
VPNStatus connected <$> getAvailableServers
|
|
|
|
getConnectedServer :: IO (Maybe T.Text)
|
|
getConnectedServer = (procStatus =<<) <$> readCmdSuccess eVPN ["status"] ""
|
|
where
|
|
procStatus = listToMaybe . mapMaybe procLine . T.lines
|
|
procLine l = case T.words l of
|
|
-- the output is green...
|
|
("\ESC[1;32;49mConnected" : "to" : server) -> Just $ T.unwords server
|
|
_ -> Nothing
|
|
|
|
getAvailableServers :: IO [VPNServer]
|
|
getAvailableServers = procOut =<< readCmdSuccess eVPN ["ls"] ""
|
|
where
|
|
procOut Nothing = do
|
|
notify IconError "failed to get list of servers"
|
|
return []
|
|
-- ASSUME the output has a useless header that ends in a line that starts
|
|
-- with "-----", after which is the stuff we care about, which is followed
|
|
-- by a blank line, after which there is more stuff I don't care about
|
|
procOut (Just ls) =
|
|
return $
|
|
mapMaybe (matchLine . T.split (== '\t')) $
|
|
takeWhile (/= "") $
|
|
drop 1
|
|
-- super lame way of matching lines that start with "-----"
|
|
$
|
|
dropWhile (not . T.isPrefixOf "-----") $
|
|
T.lines ls
|
|
-- The output of this command is very strange; it is delimited (kinda) by
|
|
-- tabs but some lines are long enough that they don't have a tab. In
|
|
-- whatever case, splitting by tabs leads to variable length lists, and the
|
|
-- id is always at the front and the location is always at the end. These
|
|
-- should handle all cases.
|
|
matchLine [i, _, l] = Just (i, l)
|
|
matchLine [i, _, _, l] = Just (i, l)
|
|
matchLine [i, _, _, _, l] = Just (i, l)
|
|
matchLine _ = Nothing
|
|
|
|
daemonIsRunning :: IO Bool
|
|
daemonIsRunning = isJust <$> readCmdSuccess "pgrep" [eVPND] ""
|
|
|
|
getDisconnectAction :: T.Text -> VPNAction
|
|
getDisconnectAction server =
|
|
(T.append "Disconnect from " server, io $ void $ disconnect server)
|
|
|
|
getConnectAction :: Maybe T.Text -> VPNServer -> VPNAction
|
|
getConnectAction connected server =
|
|
(formatServerLine server, io $ go connected)
|
|
where
|
|
go (Just c) = do
|
|
success <- disconnect c
|
|
when success con
|
|
go _ = con
|
|
con = connect server
|
|
|
|
formatServerLine :: VPNServer -> T.Text
|
|
formatServerLine (sid, sname) = T.concat [pad sid, " | ", sname]
|
|
where
|
|
pad s = T.append s $ T.replicate (10 - T.length s) " "
|
|
|
|
eVPN :: T.Text
|
|
eVPN = "expressvpn"
|
|
|
|
eVPND :: T.Text
|
|
eVPND = "expressvpnd"
|
|
|
|
connect :: VPNServer -> IO ()
|
|
connect (sid, sname) = do
|
|
res <- readCmdSuccess' eVPN ["connect", sid]
|
|
notifyIf
|
|
res
|
|
(T.append "connected to " sname)
|
|
(T.append "failed to connect to " sname)
|
|
|
|
disconnect :: T.Text -> IO Bool
|
|
disconnect server = do
|
|
res <- readCmdSuccess' eVPN ["disconnect"]
|
|
notifyIf
|
|
res
|
|
(T.append "disconnected from " server)
|
|
(T.append "failed to disconnect from " server)
|
|
return res
|
|
|
|
readCmdSuccess' :: T.Text -> [T.Text] -> IO Bool
|
|
readCmdSuccess' cmd args = isJust <$> readCmdSuccess cmd args ""
|
|
|
|
-- TODO not DRY
|
|
data NotifyIcon = IconError | IconInfo
|
|
|
|
instance Show NotifyIcon where
|
|
show IconError = "dialog-error-symbolic"
|
|
show IconInfo = "dialog-information-symbolic"
|
|
|
|
notifyIf :: Bool -> T.Text -> T.Text -> IO ()
|
|
notifyIf True s _ = notify IconInfo s
|
|
notifyIf False _ s = notify IconError s
|
|
|
|
notify :: NotifyIcon -> T.Text -> IO ()
|
|
notify icon body = void $ spawnProcess "notify-send" $ args ++ [T.unpack body]
|
|
where
|
|
args = ["-i", show icon, summary]
|
|
summary = "ExpressVPN"
|