internet-conection.py

import logging, os, pwnagotchi, requests, socket, traceback, shutil
import pwnagotchi.ui.components as components
import pwnagotchi.ui.view as view
import pwnagotchi.ui.fonts as fonts
import pwnagotchi.plugins as plugins
from pwnagotchi.ui.components import Text
from pwnagotchi import plugins
from PIL import ImageOps, Image

class InetIcon(pwnagotchi.ui.components.Widget):
    def __init__(self, value, xy=(225, 101), color=0, invert=False):
        super().__init__(xy, color)
        self.image_path = value
        self.invert = invert
        self.image = Image.open(self.image_path)
        if self.invert:
            self.image = ImageOps.invert(Image.open(self.image_path).convert('L'))
        else:
            self.image = Image.open(self.image_path)

    def draw(self, canvas, drawer):
        if self.image:
            try:
                canvas.paste(self.image, self.xy)
            except Exception as e:
                logging.error(f"Error drawing image: {e}")
                logging.error(traceback.format_exc())

class InternetConectionPlugin(plugins.Plugin):
    __author__ = 'neonlightning'
    __version__ = '1.2.4'
    __license__ = 'GPL3'
    __description__ = 'A plugin that displays the Internet connection status on the pwnagotchi display.'
    __name__ = 'InternetConectionPlugin'
    __help__ = """
    A plugin that displays the Internet connection status on the pwnagotchi display.
    """
    
    def __init__(self):
        super().__init__()
        self.current_state = False
        self.icon_on_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "internet-conection-on.png")
        self.icon_off_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "internet-conection-off.png")
        self.icon_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "internet-conection.png")

    def on_ready(self, agent):
        if not os.path.exists(self.icon_on_path):
            logging.info("[Internet Conection] on icon path not found")
            self.download_icon("https://raw.githubusercontent.com/NeonLightning/pwny/main/internet-conection-on.png", self.icon_on_path)
        if not os.path.exists(self.icon_off_path):
            logging.info("[Internet Conection] off icon path not found")
            self.download_icon("https://raw.githubusercontent.com/NeonLightning/pwny/main/internet-conection-off.png", self.icon_off_path)
        try:
            shutil.copy(self.icon_off_path, self.icon_path)
            logging.info("[Internet Conection] setup icon.")
        except Exception as e:
            logging.error(f"[Internet Conection] Error copying file: {e}")

    def download_icon(self, url, save_path):
        response = requests.get(url)
        with open(save_path, 'wb') as file:
            file.write(response.content)

    def _is_internet_available(self):
        try:
            socket.create_connection(("www.google.com", 80), timeout=1)
            return True
        except OSError:
            return False
        
    def invert(self):
        try:
            with open("/etc/pwnagotchi/config.toml", "r") as f:
                config = f.readlines()
        except FileNotFoundError:
            logging.warning("[Internet Conection] Config File not found")
            return False
        except EOFError:
            pass
        for line in config:
            line = line.strip()
            line = line.strip('\n')
            if "ui.invert = true" in line or "ui.invert = false" in line:
                if line.find("ui.invert = true") != -1:
                    logging.debug("[Internet Conection] Screen Invert True")
                    return True
                elif line.find("ui.invert = false") != -1:
                    logging.debug("[Internet Conection] Screen Invert False")
                    return False
        return False
    
    def on_loaded(self):
        is_connected = self._is_internet_available()
        if is_connected != self.current_state:
            self.current_state = is_connected
            try:
                source_path = self.icon_on_path if is_connected else self.icon_off_path
                with open(source_path, 'rb') as source_file:
                    icon_data = source_file.read()
                with open(self.icon_path, 'wb') as target_file:
                    target_file.write(icon_data)
            except Exception as e:
                logging.error(f"[Internet Conection] Error updating icon file: {e}")
        logging.info("[Internet Conection] Plugin loaded.")

    def on_ui_setup(self, ui):
        self.invert_status = self.invert()
        try:
            ui.add_element('connection_status', InetIcon(xy=(0,218), value=self.icon_path, invert=self.invert_status))
        except Exception as e:
            logging.info(f"[Internet Conection] Error loading {e}")

    def on_ui_update(self, ui):
        is_connected = self._is_internet_available()
        if is_connected != self.current_state:
            self.current_state = is_connected
            try:
                source_path = self.icon_on_path if is_connected else self.icon_off_path
                with open(source_path, 'rb') as source_file:
                    icon_data = source_file.read()
                with open(self.icon_path, 'wb') as target_file:
                    target_file.write(icon_data)
            except Exception as e:
                logging.error(f"[Internet Conection] Error updating icon file: {e}")


    def on_unload(self, ui):
        with ui._lock:
            try:
                ui.remove_element('connection_status')
            except KeyError:
                pass
        logging.info("[Internet Conection] Plugin unloaded.")