import socket, hashlib
class R:
    def __init__(s,h,p=8728): s.h=h; s.p=p; s.s=None
    def c(s,u,p):
        s.s=socket.socket(); s.s.settimeout(15)
        s.s.connect((s.h,s.p))
        s._ws('/login','=name='+u,'=password='+p)
        r=s._rr()
        if any(isinstance(x,tuple) for x in r):
            for x in r:
                if isinstance(x,dict) and 'ret' in x: ch=x['ret']
            if ch and len(ch)==32:
                hh=hashlib.md5(b'\x00'+p.encode()+bytes.fromhex(ch)).hexdigest()
                s._ws('/login','=name='+u,'=response=00'+hh); s._rr()
        return s
    def _el(s,l):
        if l<0x80: return bytes([l])
        elif l<0x4000: l|=0x8000; return bytes([(l>>8)&0xFF,l&0xFF])
        else: return bytes([0xC0|((l>>16)&0x0F),(l>>8)&0xFF,l&0xFF]) if l<0x200000 else bytes([0xE0|((l>>24)&0x07),(l>>16)&0xFF,(l>>8)&0xFF,l&0xFF])
    def _ww(s,w): s.s.sendall(s._el(len(w))+w.encode())
    def _ws(s,c,*ws): s._ww(c); [s._ww(w) for w in ws]; s.s.sendall(b'\x00')
    def _rw(s):
        f=s.s.recv(1)
        if f==b'\x00': return None
        lb=f[0]; l=lb
        if lb&0x80:
            if (lb&0xC0)==0x80: l=((lb&0x3F)<<8)+s.s.recv(1)[0]
            elif (lb&0xE0)==0xC0: l=((lb&0x1F)<<16)+(s.s.recv(1)[0]<<8)+s.s.recv(1)[0]
            else: l=((lb&0x0F)<<24)+(s.s.recv(1)[0]<<16)+(s.s.recv(1)[0]<<8)+s.s.recv(1)[0]
        d=b''
        while len(d)<l: d+=s.s.recv(l-len(d))
        return d.decode(errors='replace')
    def _rss(s):
        ws=[]
        while True:
            w=s._rw()
            if w is None: break
            ws.append(w)
        return ws
    def _rr(s):
        rs=[]
        while True:
            s2=s._rss()
            if not s2: break
            r=s2[0]; t=s2[1:]
            if r=='!done': break
            elif r=='!re':
                d={}
                for w in t:
                    if '=' in w:
                        p=w.split('=',2)
                        d[p[1]]=p[2] if len(p)>=3 else ''
                    else: d[w]=''
                rs.append(d)
            elif r=='!trap': rs.append(('!trap',t))
        return rs
    def cmd(s,c,**a):
        ws=[c]
        for k,v in a.items():
            ws.append(('?' if k.startswith('?') else '=')+k+'='+v)
        s._ws(*ws)
        return s._rr()
    def close(s):
        if s.s: s.s.close()

api=R('remote.vpnmurahjogja.my.id',33206).c('mi4','m14')

# Get Family on-event and find exact block
for si in api.cmd('/system/scheduler/print'):
    if si['name'] == 'Family':
        oe = si.get('on-event','')
        oid = si.get('.id','')
        
        # Find the notification block - from "set limit-uptime=1s" to end
        idx = oe.find('set limit-uptime=1s')
        print('Found block at index: %d' % idx)
        print()
        print('Block from set limit-uptime:')
        block = oe[idx:]
        print(block)
        print()
        print('Block length: %d' % len(block))
        
        # Build new block
        new_block = ':local curlim [ /ip hotspot user get $i limit-uptime ]; ' + block
        print()
        print('New block:')
        print(new_block[:200] + '...')
        
        # Replace
        new_oe = oe[:idx] + new_block
        print()
        print('New on-event length: %d (was %d)' % (len(new_oe), len(oe)))
        
        # Apply
        print()
        print('Applying...')
        api.cmd('/system/scheduler/set', **{'.id': oid, 'on-event': new_oe})
        
        # Verify
        for si2 in api.cmd('/system/scheduler/print'):
            if si2['name'] == 'Family':
                oe2 = si2.get('on-event','')
                print('Verified: curlim check=%s, length=%d' % ('curlim' in oe2, len(oe2)))
                break
        break

api.close()