require 'Gelo'

Tor = {}
Tor.executable = 'tor.exe'
Tor.server = '127.0.0.1'
Tor.port = '9050'
Tor.level = '5'
Tor.proxyserver = 'socks='..Tor.server..':'..Tor.port
Tor.msgenabled = 'Tor Proxy Enabled.\n\nChromium plugin support disabled for greater anonymity.'
Tor.msgdisabled = 'Tor Proxy Disabled.\n\nChromium plugin support re-enabled.'
Tor.msgpartialanon = 'Warning: Tor Proxy is disabled for web browsing.\n\nThe request will be sent via Tor network, but the response will be rendered using your normal connection.'

function Tor:AddButton(state)
 browser.addhtml_navbar([[
<div class="button" style="behavior: ~ popup-menu;foreground-image: url(Syhunt.scx#Images\icon_tor_]]..state..[[.png);" titleid="for-tor">
 <menu.popup>
 <li>Proxy Switcher
  <menu>
  <li onclick="Syhunt:SetProxy('Direct')">Direct Connection</li>
  <li onclick="Syhunt:SetProxy('Tor') Syhunt:RunTor()">Tor (Anonymous Browsing)</li>
  </menu>
 </li>
 <li onclick="Syhunt:RunTor() Syhunt:SendReqViaTor()">Send request via Tor</li>
 <li onclick="browser.newtab('https://check.torproject.org/')">Are you using Tor?</li>
 <li>Help
  <menu>
  <li onclick="browser.newtab('https://www.torproject.org/about/overview')">About Tor</li>
  </menu>
 </li>
 </menu>
</div>
<popup #for-tor>Tor</popup>
]])
end

function Tor:ViewRequestEditor()
 if str_beginswith(tab.url,'http') then
  if browser.proxy_getserver() ~= self.proxyserver then
   app.showmessage(self.msgpartialanon)
  end
  Syhunt:ViewRequestEditor()
  extbar.setval('button[name="viator"]','True')
 else
  app.showmessage('No URL loaded.')
 end
end

function Tor:SetProxy(proxy)
 if proxy == 'Tor' then
  browser.plugins_enable(false)
  browser.proxy_enable(true,self.proxyserver)
  app.showmessage(self.msgenabled)
 else 
  browser.plugins_enable(true)
  browser.proxy_enable(false)
  app.showmessage(self.msgdisabled)
 end
end

function Tor:IsRunning()
 return task_isrunning(self.executable)
end

function Tor:Run()
  if task_isrunning(self.executable) == false then
   file_exechide(ProgDir..self.executable)
  end
end

function Tor:Startup()
 if browser.proxy_getserver() == self.proxyserver then
  Tor:Run()
  Tor:AddButton('enabled')
 else
  Tor:AddButton('disabled')
 end
end

function Tor:Shutdown()
  if task_isrunning(self.executable) == true then
   task_kill(self.executable)
  end
end