IPDisplay.py

############################
# setup is simple
# main.plugins.IPDisplay.skip_devices = [
#     'eth0',
#     'usb0',
#     'bnep0',
#     'wlan0',
#     'ect...'
# ]
# main.plugins.IPDisplay.position = "0, 82"

from pwnagotchi.ui.components import LabeledValue
from pwnagotchi.ui.view import BLACK
import pwnagotchi.ui.fonts as fonts
import pwnagotchi.plugins as plugins
import logging
import time
import subprocess
import ipaddress

class IPDisplay(plugins.Plugin):
    __author__ = 'NeonLightning(thank to NurseJackass and jayofelony)'
    __version__ = '1.0.0'
    __license__ = 'GPL3'
    __description__ = 'Display IP addresses on the Pwnagotchi UI'

    def __init__(self):
        self.options = dict()
        self.device_skip_list = ['lo']
        self.device_index = 0
        self.skip_time = 0
        self.ready = False
        self.last_update_time = 0

    def on_loaded(self):
        if 'skip_devices' in self.options:
            self.device_skip_list = self.options['skip_devices']
        self.options['skip_devices'] = self.device_skip_list
        logging.debug("IP Display Plugin loaded.")
        
    def on_ready(self, agent):
        self._agent = agent
        logging.info("IP Display Plugin ready.")
        self.ready = True

    def on_ui_setup(self, ui):
        pos1 = (0, 82)
        if 'position' in self.options:
            pos1 = self.options['position']
        ui.add_element('ip1', LabeledValue(color=BLACK, label="", value='Initializing...',
                                           position=pos1, label_font=fonts.Small, text_font=fonts.Small))

    def get_iface_addrs(self):
        command = f"ip -4 -o addr | awk '/inet / {{print $2 \":\" $4}}' | cut -d '/' -f 1"
        ifaces = []
        for line in subprocess.getoutput(command).split('\n'):
            pts = line.strip().split(":")
            if pts[0].lower() not in self.device_skip_list:
                ifaces.append(line.strip())
        return ifaces
    
    def on_ui_update(self, ui):
        try:
            if time.time() - self.last_update_time < 2:
                return
            self.last_update_time = time.time()
            if self.skip_time + 5 > time.time():
                return
            self.device_index += 1
            self.skip_time = time.time()
            ifaces = self.get_iface_addrs()
            if self.device_index >= len(ifaces):
                self.device_index = 0
            current_device = ifaces[self.device_index]
            ui.set('ip1', f'{current_device}')
        except Exception as e:
            logging.exception(repr(e))   
                
    def on_unload(self, ui):
        self.ready = False
        ui.remove_element('ip1')
        logging.info("IP Display Plugin unloaded.")