main.py

# coded by lagcleaner
# email: lagcleaner@gmail.com

from io import BytesIO
import sys
from typing import Callable, Union
from datetime import datetime, date as Date

# pylint: disable=unused-import
from tfprotocol_client.misc.constants import (
    KEY_LEN_INTERVAL,
    RESPONSE_LOGGER,
    SECFS_ALL_PERMISSIONS,
)

# pylint: disable=unused-import
from tfprotocol_client.models.keepalive_options import (
    KeepAliveMechanismType,
    KeepAliveOptions,
)
from tfprotocol_client.tfprotocol import TfProtocol
from tfprotocol_client.models.status_info import StatusInfo
from tfprotocol_client.models.file_stat import FileStat, FileStatTypeEnum
from tfprotocol_client.connection.protocol_client import ProtocolClient
from tfprotocol_client.models.putget_commands import PutGetCommandEnum
from tfprotocol_client.models.transfer_state import TransferStatus
from tfprotocol_client.connection.codes_sender_recvr import CodesSenderRecvr


def get_publickey() -> str:
    """extract public key from the keyfile"""
    with open('public.key', mode='r', encoding='utf8') as f:
        public_key = f.read()
    return public_key


class MyHandler:
    """My handler

    Args:
        SuperProtoHandler ([type]): [description]
    """

    def mkdir_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def ftod_callback(self, timestamp: int, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            'timestamp:(',
            timestamp,
            ')',
        )

    def datef_callback(self, date: datetime, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            'date:(',
            date,
            ')',
        )

    def del_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def rmdir_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def copy_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def touch_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def fupd_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.payload,
        )

    def date_callback(self, timestamp: int, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            'timestamp:(',
            timestamp,
            ')',
            status.status.name.ljust(8, ' '),
        )

    def dtof_callback(self, _: Date, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def fstat_callback(self, _: FileStatTypeEnum, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.payload,
        )

    def cpdir_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def xcopy_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def xdel_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def xrmdir_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def xcpdir_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def lock_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def sendfile_callback(
        self,
        is_overriten: bool,
        path: str,
        status: StatusInfo,
        payload: Union[bytes, bytearray],
    ):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def rcvfile_callback(
        self, delete_after: bool, path: str, status: StatusInfo, sink: BytesIO = None
    ):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            len(status.payload),
            sep=' ',
        )
        # if len(status.payload) > 21588:
        #     print('-------')
        #     print(status.payload)
        #     print('-------')
        if sink is None:
            print(
                "CLIENT-HANDLER: ",
                sys._getframe()
                .f_code.co_name.replace('_callback', '', 1)
                .ljust(10, ' '),
                status.status.name.ljust(8, ' '),
                ' ',
                'NO-SINK-PASSED',
            )
            return
        if sink.writable() and status.payload:
            sink.write(status.payload)

    def ls_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def lsr_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.message,
        )

    def renam_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def keepalive_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def login_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def chmod_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.payload,
        )

    def chown_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def getcan_callback(
        self,
        status: StatusInfo,
        _: ProtocolClient,
        transfer_status: TransferStatus,
    ):
        if transfer_status is not None:
            if transfer_status.handling_canpt:
                # transfer_status.dummy = PutGetCommandEnum.HPFCANCEL.value
                pass
            elif transfer_status.client_command in (
                PutGetCommandEnum.HPFCANCEL.value,
                PutGetCommandEnum.HPFEND.value,
            ):
                print(
                    "CLIENT-HANDLER: ",
                    sys._getframe()
                    .f_code.co_name.replace('_callback', '', 1)
                    .ljust(10, ' '),
                    '-'.ljust(8, '-'),
                    'finished',
                )
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' ')
            if status is not None
            else '-'.ljust(8, '-'),
            transfer_status,
        )

    def putcan_callback(
        self,
        status: StatusInfo,
        _: ProtocolClient,
        transfer_status: TransferStatus,
    ):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' ')
            if status is not None
            else '-'.ljust(8, '-'),
            transfer_status,
        )

    def sha256_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.payload,
        )

    def prockey_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.payload,
        )

    def freesp_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def udate_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def ndate_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def getstatus_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.code,
        )

    def get_callback(self, codes: CodesSenderRecvr):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            codes.last_header,
        )
        # sleep(0.3)
        # codes.send_get(PutGetCommandEnum.HPFCANCEL.value)

    def put_callback(self, codes: CodesSenderRecvr):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            '-'.ljust(8, '-'),
            codes.last_header,
        )

    def putstatus_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def nigma_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            'NewSessionKey:(',
            status.payload,
            ')',
        )

    def rmsecuredirectory_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def injail_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def tlb_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def sdown_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def sup_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def fsize_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.code,
        )

    def fsizels_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.code,
        )

    def lsv2_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def lsrv2_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def ftype_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.payload,
        )

    def ftypels_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.code,
            status.payload,
        )

    def fstatls_callback(self, file_stat: FileStat):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            '-'.ljust(8, '-'),
            file_stat,
        )

    def addntfy_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            '-'.ljust(8, '-'),
        )

    def notification_callback(
        self,
        status: StatusInfo,
        send_ok: Callable,
        send_del: Callable,
    ):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            '-'.ljust(8, '-'),
        )
        # The handler always have to end with send_ok() or send_del() and should be
        # called once per notification.
        send_del()

    def intread_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.code,
        )

    def intwrite_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.code,
        )

    def netlock_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.payload,
        )

    def netunlock_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.payload,
        )

    def netlocktry_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.payload,
        )

    def netmutacqtry_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.payload,
        )

    def netmutrel_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.payload,
        )

    def setfsid_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.payload,
        )

    def setfsperm_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.payload,
        )

    def remfsperm_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.payload,
        )

    def getfsperm_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.payload,
        )

    def issecfs_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.payload,
        )

    def locksystem_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
            status.payload,
        )

    def echo_callback(self, value: str):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            '-'.ljust(8, ' '),
            'ServerSays:("',
            value,
            '")',
        )
        print(f'Server: {value}')

    def status_server(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def response_server_callback(self, status: StatusInfo):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1).ljust(10, ' '),
            status.status.name.ljust(8, ' '),
        )

    def instance_tfprotocol(self, instance: any):
        print(
            "CLIENT-HANDLER: ",
            sys._getframe().f_code.co_name.replace('_callback', '', 1),
        )


def test_updown_tinyfiles(proto: TfProtocol):
    # DONE
    with open('test2.java', 'rb') as f:
        proto.sndfile_command(True, 'test2.java', f)

    with open('test2-rec.java', 'wb') as f:
        proto.rcvfile_command(False, 'test2.java', f)

    proto.fstat_command('test2.java')


def test_supsdown_files(proto: TfProtocol):
    # DONE
    with open('test2.java', 'rb') as f:
        proto.sup_command('test2.2.java', f, 5)

    with open('test2.2-rec.java', 'wb') as f:
        proto.sdown_command('test2.2.java', f, 10)

    proto.fstat_command('test2.2.java')


def test_putcangetcan_files(proto: TfProtocol):
    # DONE
    with open('test2.java', 'rb') as f:
        proto.putcan_command(f, 'test2.3.java', 0, 1024, 1)

    with open('test2.3-rec.java', 'wb') as f:
        proto.getcan_command(f, 'test2.3.java', 0, 1024, 2)

    proto.fstat_command('test2.3.java')


def test_putget_files(proto: TfProtocol):
    # DONE
    with open('test2.java', 'rb') as f:
        proto.put_command(f, 'test2.4.java', 0, 48)

    with open('test2.4-rec.java', 'wb') as f:
        proto.get_command(f, 'test2.4.java', 0, 48)

    proto.fstat_command('test2.4.java')


def test_folders_commands(proto: TfProtocol):
    # DONE
    proto.rmdir_command('py_test')

    proto.mkdir_command('py_test')
    # proto.mkdir_command('py_test/pattern')
    # proto.mkdir_command('py_test/214pattern')
    proto.mkdir_command('py_test/test1')
    proto.mkdir_command('py_test/test2')
    proto.mkdir_command('py_test/test2/test21')
    # proto.mkdir_command('py_test/test2/test22')
    proto.mkdir_command('py_test/test3')

    proto.touch_command('py_test/test2/test.java')

    proto.lsr_command('/py_test')
    proto.rmdir_command('/py_test/test1')

    proto.renam_command('/py_test/test3', '/py_test/test3new')
    proto.lsr_command('/py_test')
    proto.rmdir_command('/py_test/test3new')

    proto.cpdir_command('/py_test/test2', '/py_test/test2cpdir')
    proto.lsr_command('/py_test')

    proto.mkdir_command('py_test/pattern')
    proto.xcpdir_command('testxcpdir', '/py_test/test2', 'pattern')
    proto.lsr_command('/py_test')

    proto.mkdir_command('/py_test/test2copy')
    proto.copy_command('py_test/test2/test.java', 'py_test/test2copy/test.java')
    proto.lsr_command('/py_test')
    proto.rmdir_command('/py_test/test2copy')

    proto.xcopy_command(
        'test_xcopy.java',
        '/py_test/test2/test.java',
        '/test21',
    )
    proto.lsr_command('/py_test')
    proto.rmdir_command('/py_test/pattern')

    proto.xdel_command('/py_test', 'test.java')
    proto.lsr_command('/py_test')

    proto.xrmdir_command('/py_test', 'test2')
    proto.lsr_command('/py_test')

    proto.touch_command('/py_test/file_touch.txt')
    proto.lsr_command('/py_test')

    proto.lsrv2_command('/py_test', '/py_test/testrls.txt')

    proto.lsv2_command('/py_test', '/py_test/testls.txt')

    proto.fsize_command('/py_test/testls.txt')

    proto.fsizels_command('/py_test/testls.txt')

    proto.fstatls_command(
        '/py_test/testls.txt', response_handler=MyHandler().fstatls_callback
    )

    proto.ftypels_command(
        '/py_test/testls.txt', response_handler=MyHandler().ftypels_callback
    )

    proto.rmdir_command('py_test')


def test_files_modification_commands(proto: TfProtocol):
    proto.rmdir_command('/py_test')

    proto.mkdir_command('/py_test')
    proto.touch_command('/py_test/file_touch.txt')
    proto.fstat_command('/py_test/file_touch.txt')

    proto.chmod_command('/py_test/file_touch.txt', '600')
    proto.fstat_command('/py_test/file_touch.txt')

    # proto.chown_command('/py_test/file_touch.txt', '', '')
    # proto.fstat_command('/py_test/file_touch.txt')

    proto.fupd_command('/py_test/file_touch.txt')
    proto.fstat_command('/py_test/file_touch.txt')


def test_notify_system_commands(proto: TfProtocol):
    proto.addntfy_command('test', '')
    proto.startnfy_command(3)


def test_integrity_rw_commands(proto: TfProtocol):
    checksum: str = None

    def _handler_callback(status: StatusInfo):
        nonlocal checksum
        checksum = status.message
        print("CLIENT-HANDLER: ", status)

    with open('test.java', 'rb') as f:
        proto.sup_command('test2.java', f, 5)

    with open('test2-rec.java', 'wb') as f:
        proto.intread_command('test2.java', f, response_handler=_handler_callback)

    if checksum is not None:
        with open('test2.java', 'rb') as f:
            proto.intwrite_command(
                'test2.java',
                f,
                checksum,
                response_handler=MyHandler().intwrite_callback,
            )


def test_netsecurity_commands(proto: TfProtocol):
    myhandler = MyHandler()
    lock_id: str = None

    def _handler_callback(status: StatusInfo):
        nonlocal lock_id
        lock_id = status.message
        print("CLIENT-HANDLER: ", status)

    proto.touch_command('testlaglock', response_handler=myhandler.touch_callback)
    proto.netlock_command(5, 'testlaglock', response_handler=_handler_callback)
    proto.netlocktry_command(
        5,
        'testlaglock',
        response_handler=myhandler.netlocktry_callback,
    )

    proto.netunlock_command(lock_id)

    proto.touch_command('test.mut')
    proto.netmutacqtry_command(
        'test.mut',
        'testtoken',
        response_handler=myhandler.netmutacqtry_callback,
    )
    proto.netmutrel_command(
        'test.mut',
        'testtoken',
        response_handler=myhandler.netmutrel_callback,
    )

    proto.setfsid_command('secid', response_handler=myhandler.setfsid_callback)

    proto.mkdir_command('testlag')
    proto.setfsperm_command(
        'secid',
        SECFS_ALL_PERMISSIONS,
        'testlag',
        response_handler=myhandler.setfsperm_callback,
    )
    proto.issecfs_command(
        'testlag',
        response_handler=myhandler.issecfs_callback,
    )
    proto.getfsperm_command(
        'secid',
        'testlag',
        response_handler=myhandler.getfsperm_callback,
    )
    proto.remfsperm_command(
        'secid',
        'testlag',
        response_handler=myhandler.remfsperm_callback,
    )

    proto.mkdir_command(
        '/py_test_sys',
        response_handler=myhandler.mkdir_callback,
    )
    proto.locksys_command(
        '/py_test_sys',
        response_handler=myhandler.locksystem_callback,
    )
    proto.touch_command(
        '/py_test_sys/file_touch.txt',
        response_handler=myhandler.touch_callback,
    )


def test_regular_commands(proto: TfProtocol):
    proto.echo_command('Hola mundo')

    proto.echo_command(
        'Hola despues de cambiar el cerrojo', response_handler=RESPONSE_LOGGER
    )
    proto.date_command(response_handler=MyHandler().date_callback)
    proto.nigma_command(16, response_handler=MyHandler().nigma_callback)
    proto.datef_command()
    proto.dtof_command(datetime.timestamp(datetime.now()))
    proto.ftod_command('1999-12-10 12:12:0')
    proto.udate_command()
    proto.freesp_command()
    proto.prockey_command()
    proto.sha256_command('test2.java')

    proto.end_command()


def main():
    # ADDRESS = '192.168.0.128'
    ADDRESS = 'tfproto.expresscuba.com'
    PORT = 10345
    proto = TfProtocol(
        '0.0',
        get_publickey(),
        'testhash',
        ADDRESS,
        PORT,
        verbosity_mode=True,
    )

    proto.connect(
        # keepalive_options=KeepAliveOptions(KeepAliveMechanismType.TCP_NATIVE, 1, 5, 3)
    )
    # proto.prockey_command()
    # sleep(20)
    # proto.echo_command('sacamos cuela')

    # test_updown_tinyfiles(proto)
    # test_supsdown_files(proto)
    # test_putget_files(proto)
    # test_putcangetcan_files(proto)
    # test_folders_commands(proto)
    # test_files_modification_commands(proto)
    test_regular_commands(proto)
    # test_notify_system_commands(proto)
    # test_integrity_rw_commands(proto)
    # test_netsecurity_commands(proto)

    # proto.rmdir_command('prueba.sd')
    # proto.tlb_command()


if __name__ == '__main__':
    main()