<?xml version="1.0" encoding="UTF-8"?> <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd"> <html> <head> <!-- Generated by HsColour, http://www.cs.york.ac.uk/fp/darcs/hscolour/ --> <title>XMonad/Prompt/Ssh.hs</title> <link type='text/css' rel='stylesheet' href='hscolour.css' /> </head> <body> <pre><a name="line-1"></a><span class='hs-comment'>-----------------------------------------------------------------------------</span> <a name="line-2"></a><span class='hs-comment'>-- |</span> <a name="line-3"></a><span class='hs-comment'>-- Module : XMonad.Prompt.Ssh</span> <a name="line-4"></a><span class='hs-comment'>-- Copyright : (C) 2007 Andrea Rossato</span> <a name="line-5"></a><span class='hs-comment'>-- License : BSD3</span> <a name="line-6"></a><span class='hs-comment'>--</span> <a name="line-7"></a><span class='hs-comment'>-- Maintainer : andrea.rossato@unibz.it</span> <a name="line-8"></a><span class='hs-comment'>-- Stability : unstable</span> <a name="line-9"></a><span class='hs-comment'>-- Portability : unportable</span> <a name="line-10"></a><span class='hs-comment'>--</span> <a name="line-11"></a><span class='hs-comment'>-- A ssh prompt for XMonad</span> <a name="line-12"></a><span class='hs-comment'>--</span> <a name="line-13"></a><span class='hs-comment'>-----------------------------------------------------------------------------</span> <a name="line-14"></a> <a name="line-15"></a><span class='hs-keyword'>module</span> <span class='hs-conid'>XMonad</span><span class='hs-varop'>.</span><span class='hs-conid'>Prompt</span><span class='hs-varop'>.</span><span class='hs-conid'>Ssh</span> <a name="line-16"></a> <span class='hs-layout'>(</span> <span class='hs-comment'>-- * Usage</span> <a name="line-17"></a> <span class='hs-comment'>-- $usage</span> <a name="line-18"></a> <span class='hs-varid'>sshPrompt</span> <a name="line-19"></a> <span class='hs-layout'>)</span> <span class='hs-keyword'>where</span> <a name="line-20"></a> <a name="line-21"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>XMonad</span> <a name="line-22"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>XMonad</span><span class='hs-varop'>.</span><span class='hs-conid'>Util</span><span class='hs-varop'>.</span><span class='hs-conid'>Run</span> <a name="line-23"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>XMonad</span><span class='hs-varop'>.</span><span class='hs-conid'>Prompt</span> <a name="line-24"></a> <a name="line-25"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>System</span><span class='hs-varop'>.</span><span class='hs-conid'>Directory</span> <a name="line-26"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>System</span><span class='hs-varop'>.</span><span class='hs-conid'>Environment</span> <a name="line-27"></a> <a name="line-28"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Control</span><span class='hs-varop'>.</span><span class='hs-conid'>Monad</span> <a name="line-29"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>List</span> <a name="line-30"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Maybe</span> <a name="line-31"></a> <a name="line-32"></a><span class='hs-comment'>-- $usage</span> <a name="line-33"></a><span class='hs-comment'>-- 1. In your @~\/.xmonad\/xmonad.hs@:</span> <a name="line-34"></a><span class='hs-comment'>--</span> <a name="line-35"></a><span class='hs-comment'>-- > import XMonad.Prompt</span> <a name="line-36"></a><span class='hs-comment'>-- > import XMonad.Prompt.Ssh</span> <a name="line-37"></a><span class='hs-comment'>--</span> <a name="line-38"></a><span class='hs-comment'>-- 2. In your keybindings add something like:</span> <a name="line-39"></a><span class='hs-comment'>--</span> <a name="line-40"></a><span class='hs-comment'>-- > , ((modm .|. controlMask, xK_s), sshPrompt defaultXPConfig)</span> <a name="line-41"></a><span class='hs-comment'>--</span> <a name="line-42"></a><span class='hs-comment'>-- Keep in mind, that if you want to use the completion you have to</span> <a name="line-43"></a><span class='hs-comment'>-- disable the "HashKnownHosts" option in your ssh_config</span> <a name="line-44"></a><span class='hs-comment'>--</span> <a name="line-45"></a><span class='hs-comment'>-- For detailed instruction on editing the key binding see</span> <a name="line-46"></a><span class='hs-comment'>-- "XMonad.Doc.Extending#Editing_key_bindings".</span> <a name="line-47"></a> <a name="line-48"></a><a name="Ssh"></a><span class='hs-keyword'>data</span> <span class='hs-conid'>Ssh</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Ssh</span> <a name="line-49"></a> <a name="line-50"></a><span class='hs-keyword'>instance</span> <span class='hs-conid'>XPrompt</span> <span class='hs-conid'>Ssh</span> <span class='hs-keyword'>where</span> <a name="line-51"></a> <span class='hs-varid'>showXPrompt</span> <span class='hs-conid'>Ssh</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"SSH to: "</span> <a name="line-52"></a> <span class='hs-varid'>commandToComplete</span> <span class='hs-keyword'>_</span> <span class='hs-varid'>c</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>c</span> <a name="line-53"></a> <span class='hs-varid'>nextCompletion</span> <span class='hs-keyword'>_</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>getNextCompletion</span> <a name="line-54"></a> <a name="line-55"></a><a name="sshPrompt"></a><span class='hs-definition'>sshPrompt</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>XPConfig</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>X</span> <span class='hs-conid'>()</span> <a name="line-56"></a><span class='hs-definition'>sshPrompt</span> <span class='hs-varid'>c</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-57"></a> <span class='hs-varid'>sc</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>io</span> <span class='hs-varid'>sshComplList</span> <a name="line-58"></a> <span class='hs-varid'>mkXPrompt</span> <span class='hs-conid'>Ssh</span> <span class='hs-varid'>c</span> <span class='hs-layout'>(</span><span class='hs-varid'>mkComplFunFromList</span> <span class='hs-varid'>sc</span><span class='hs-layout'>)</span> <span class='hs-varid'>ssh</span> <a name="line-59"></a> <a name="line-60"></a><a name="ssh"></a><span class='hs-definition'>ssh</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>String</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>X</span> <span class='hs-conid'>()</span> <a name="line-61"></a><span class='hs-definition'>ssh</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>runInTerm</span> <span class='hs-str'>""</span> <span class='hs-varop'>.</span> <span class='hs-layout'>(</span><span class='hs-str'>"ssh "</span> <span class='hs-varop'>++</span> <span class='hs-layout'>)</span> <a name="line-62"></a> <a name="line-63"></a><a name="sshComplList"></a><span class='hs-definition'>sshComplList</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>IO</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>String</span><span class='hs-keyglyph'>]</span> <a name="line-64"></a><span class='hs-definition'>sshComplList</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>uniqSort</span> <span class='hs-varop'>`fmap`</span> <span class='hs-varid'>liftM2</span> <span class='hs-layout'>(</span><span class='hs-varop'>++</span><span class='hs-layout'>)</span> <span class='hs-varid'>sshComplListLocal</span> <span class='hs-varid'>sshComplListGlobal</span> <a name="line-65"></a> <a name="line-66"></a><a name="sshComplListLocal"></a><span class='hs-definition'>sshComplListLocal</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>IO</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>String</span><span class='hs-keyglyph'>]</span> <a name="line-67"></a><span class='hs-definition'>sshComplListLocal</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-68"></a> <span class='hs-varid'>h</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>getEnv</span> <span class='hs-str'>"HOME"</span> <a name="line-69"></a> <span class='hs-varid'>sshComplListFile</span> <span class='hs-varop'>$</span> <span class='hs-varid'>h</span> <span class='hs-varop'>++</span> <span class='hs-str'>"/.ssh/known_hosts"</span> <a name="line-70"></a> <a name="line-71"></a><a name="sshComplListGlobal"></a><span class='hs-definition'>sshComplListGlobal</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>IO</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>String</span><span class='hs-keyglyph'>]</span> <a name="line-72"></a><span class='hs-definition'>sshComplListGlobal</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-73"></a> <span class='hs-varid'>env</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>getEnv</span> <span class='hs-str'>"SSH_KNOWN_HOSTS"</span> <span class='hs-varop'>`catch`</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>\</span><span class='hs-keyword'>_</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>return</span> <span class='hs-str'>"/nonexistent"</span><span class='hs-layout'>)</span> <a name="line-74"></a> <span class='hs-varid'>fs</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mapM</span> <span class='hs-varid'>fileExists</span> <span class='hs-keyglyph'>[</span> <span class='hs-varid'>env</span> <a name="line-75"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"/usr/local/etc/ssh/ssh_known_hosts"</span> <a name="line-76"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"/usr/local/etc/ssh_known_hosts"</span> <a name="line-77"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"/etc/ssh/ssh_known_hosts"</span> <a name="line-78"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"/etc/ssh_known_hosts"</span> <a name="line-79"></a> <span class='hs-keyglyph'>]</span> <a name="line-80"></a> <span class='hs-keyword'>case</span> <span class='hs-varid'>catMaybes</span> <span class='hs-varid'>fs</span> <span class='hs-keyword'>of</span> <a name="line-81"></a> <span class='hs-conid'>[]</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>return</span> <span class='hs-conid'>[]</span> <a name="line-82"></a> <span class='hs-layout'>(</span><span class='hs-varid'>f</span><span class='hs-conop'>:</span><span class='hs-keyword'>_</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>sshComplListFile'</span> <span class='hs-varid'>f</span> <a name="line-83"></a> <a name="line-84"></a><a name="sshComplListFile"></a><span class='hs-definition'>sshComplListFile</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>String</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>IO</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>String</span><span class='hs-keyglyph'>]</span> <a name="line-85"></a><span class='hs-definition'>sshComplListFile</span> <span class='hs-varid'>kh</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-86"></a> <span class='hs-varid'>f</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>doesFileExist</span> <span class='hs-varid'>kh</span> <a name="line-87"></a> <span class='hs-keyword'>if</span> <span class='hs-varid'>f</span> <span class='hs-keyword'>then</span> <span class='hs-varid'>sshComplListFile'</span> <span class='hs-varid'>kh</span> <a name="line-88"></a> <span class='hs-keyword'>else</span> <span class='hs-varid'>return</span> <span class='hs-conid'>[]</span> <a name="line-89"></a> <a name="line-90"></a><a name="sshComplListFile'"></a><span class='hs-definition'>sshComplListFile'</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>String</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>IO</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>String</span><span class='hs-keyglyph'>]</span> <a name="line-91"></a><span class='hs-definition'>sshComplListFile'</span> <span class='hs-varid'>kh</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-92"></a> <span class='hs-varid'>l</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>readFile</span> <span class='hs-varid'>kh</span> <a name="line-93"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>map</span> <span class='hs-layout'>(</span><span class='hs-varid'>getWithPort</span> <span class='hs-varop'>.</span> <span class='hs-varid'>takeWhile</span> <span class='hs-layout'>(</span><span class='hs-varop'>/=</span> <span class='hs-chr'>','</span><span class='hs-layout'>)</span> <span class='hs-varop'>.</span> <span class='hs-varid'>concat</span> <span class='hs-varop'>.</span> <span class='hs-varid'>take</span> <span class='hs-num'>1</span> <span class='hs-varop'>.</span> <span class='hs-varid'>words</span><span class='hs-layout'>)</span> <a name="line-94"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>filter</span> <span class='hs-varid'>nonComment</span> <a name="line-95"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>lines</span> <span class='hs-varid'>l</span> <a name="line-96"></a> <a name="line-97"></a><a name="fileExists"></a><span class='hs-definition'>fileExists</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>String</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>IO</span> <span class='hs-layout'>(</span><span class='hs-conid'>Maybe</span> <span class='hs-conid'>String</span><span class='hs-layout'>)</span> <a name="line-98"></a><span class='hs-definition'>fileExists</span> <span class='hs-varid'>kh</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-99"></a> <span class='hs-varid'>f</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>doesFileExist</span> <span class='hs-varid'>kh</span> <a name="line-100"></a> <span class='hs-keyword'>if</span> <span class='hs-varid'>f</span> <span class='hs-keyword'>then</span> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-conid'>Just</span> <span class='hs-varid'>kh</span> <a name="line-101"></a> <span class='hs-keyword'>else</span> <span class='hs-varid'>return</span> <span class='hs-conid'>Nothing</span> <a name="line-102"></a> <a name="line-103"></a><a name="nonComment"></a><span class='hs-definition'>nonComment</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>String</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Bool</span> <a name="line-104"></a><span class='hs-definition'>nonComment</span> <span class='hs-conid'>[]</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>False</span> <a name="line-105"></a><span class='hs-definition'>nonComment</span> <span class='hs-layout'>(</span><span class='hs-chr'>'#'</span><span class='hs-conop'>:</span><span class='hs-keyword'>_</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>False</span> <a name="line-106"></a><span class='hs-definition'>nonComment</span> <span class='hs-layout'>(</span><span class='hs-chr'>'|'</span><span class='hs-conop'>:</span><span class='hs-keyword'>_</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>False</span> <span class='hs-comment'>-- hashed, undecodeable</span> <a name="line-107"></a><span class='hs-definition'>nonComment</span> <span class='hs-keyword'>_</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>True</span> <a name="line-108"></a> <a name="line-109"></a><a name="getWithPort"></a><span class='hs-definition'>getWithPort</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>String</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>String</span> <a name="line-110"></a><span class='hs-definition'>getWithPort</span> <span class='hs-layout'>(</span><span class='hs-chr'>'['</span><span class='hs-conop'>:</span><span class='hs-varid'>str</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>host</span> <span class='hs-varop'>++</span> <span class='hs-str'>" -p "</span> <span class='hs-varop'>++</span> <span class='hs-varid'>port</span> <a name="line-111"></a> <span class='hs-keyword'>where</span> <span class='hs-layout'>(</span><span class='hs-varid'>host</span><span class='hs-layout'>,</span><span class='hs-varid'>p</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>break</span> <span class='hs-layout'>(</span><span class='hs-varop'>==</span><span class='hs-chr'>']'</span><span class='hs-layout'>)</span> <span class='hs-varid'>str</span> <a name="line-112"></a> <span class='hs-varid'>port</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>case</span> <span class='hs-varid'>p</span> <span class='hs-keyword'>of</span> <a name="line-113"></a> <span class='hs-chr'>']'</span><span class='hs-conop'>:</span><span class='hs-chr'>':'</span><span class='hs-conop'>:</span><span class='hs-varid'>x</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>x</span> <a name="line-114"></a> <span class='hs-keyword'>_</span> <span class='hs-keyglyph'>-></span> <span class='hs-str'>"22"</span> <a name="line-115"></a><span class='hs-definition'>getWithPort</span> <span class='hs-varid'>str</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>str</span> </pre></body> </html>