Move bot files into mosfet/
This commit is contained in:
		
							
								
								
									
										0
									
								
								mosfet/protocol/__init__.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										0
									
								
								mosfet/protocol/__init__.py
									
									
									
									
									
										Normal file
									
								
							
							
								
								
									
										157
									
								
								test_pycraft.py
									
									
									
									
									
								
							
							
						
						
									
										157
									
								
								test_pycraft.py
									
									
									
									
									
								
							@@ -1,157 +0,0 @@
 | 
				
			|||||||
#!/usr/bin/env python
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
import getpass
 | 
					 | 
				
			||||||
import sys
 | 
					 | 
				
			||||||
import re
 | 
					 | 
				
			||||||
from optparse import OptionParser
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
from custom.managers import DataManager, ChunksManager
 | 
					 | 
				
			||||||
from custom.networking.packets.clientbound.play import chunk_data, block_change_packet
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
import minecraft.networking.packets
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def get_packets(old_get_packets):
 | 
					 | 
				
			||||||
    def wrapper(func, context):
 | 
					 | 
				
			||||||
        print('Monkey-patched.')
 | 
					 | 
				
			||||||
        packets = func(context)
 | 
					 | 
				
			||||||
        packets.add(chunk_data.ChunkDataPacket)
 | 
					 | 
				
			||||||
        packets.add(block_change_packet.BlockChangePacket)
 | 
					 | 
				
			||||||
        packets.add(block_change_packet.MultiBlockChangePacket)
 | 
					 | 
				
			||||||
        return packets
 | 
					 | 
				
			||||||
    return lambda x: wrapper(old_get_packets, x)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
minecraft.networking.packets.clientbound.play.get_packets = get_packets(minecraft.networking.packets.clientbound.play.get_packets)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
from minecraft import authentication
 | 
					 | 
				
			||||||
from minecraft.exceptions import YggdrasilError
 | 
					 | 
				
			||||||
from minecraft.networking.connection import Connection
 | 
					 | 
				
			||||||
from minecraft.networking.packets import Packet, clientbound, serverbound
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def get_options():
 | 
					 | 
				
			||||||
    parser = OptionParser()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    parser.add_option("-u", "--username", dest="username", default=None,
 | 
					 | 
				
			||||||
                      help="username to log in with")
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    parser.add_option("-p", "--password", dest="password", default=None,
 | 
					 | 
				
			||||||
                      help="password to log in with")
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    parser.add_option("-s", "--server", dest="server", default=None,
 | 
					 | 
				
			||||||
                      help="server host or host:port "
 | 
					 | 
				
			||||||
                           "(enclose IPv6 addresses in square brackets)")
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    parser.add_option("-o", "--offline", dest="offline", action="store_true",
 | 
					 | 
				
			||||||
                      help="connect to a server in offline mode "
 | 
					 | 
				
			||||||
                           "(no password required)")
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    parser.add_option("-d", "--dump-packets", dest="dump_packets",
 | 
					 | 
				
			||||||
                      action="store_true",
 | 
					 | 
				
			||||||
                      help="print sent and received packets to standard error")
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    parser.add_option("-v", "--dump-unknown-packets", dest="dump_unknown",
 | 
					 | 
				
			||||||
                      action="store_true",
 | 
					 | 
				
			||||||
                      help="include unknown packets in --dump-packets output")
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    (options, args) = parser.parse_args()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    if not options.username:
 | 
					 | 
				
			||||||
        options.username = input("Enter your username: ")
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    if not options.password and not options.offline:
 | 
					 | 
				
			||||||
        options.password = getpass.getpass("Enter your password (leave "
 | 
					 | 
				
			||||||
                                           "blank for offline mode): ")
 | 
					 | 
				
			||||||
        options.offline = options.offline or (options.password == "")
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    if not options.server:
 | 
					 | 
				
			||||||
        options.server = input("Enter server host or host:port "
 | 
					 | 
				
			||||||
                               "(enclose IPv6 addresses in square brackets): ")
 | 
					 | 
				
			||||||
    # Try to split out port and address
 | 
					 | 
				
			||||||
    match = re.match(r"((?P<host>[^\[\]:]+)|\[(?P<addr>[^\[\]]+)\])"
 | 
					 | 
				
			||||||
                     r"(:(?P<port>\d+))?$", options.server)
 | 
					 | 
				
			||||||
    if match is None:
 | 
					 | 
				
			||||||
        raise ValueError("Invalid server address: '%s'." % options.server)
 | 
					 | 
				
			||||||
    options.address = match.group("host") or match.group("addr")
 | 
					 | 
				
			||||||
    options.port = int(match.group("port") or 25565)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    return options
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def main():
 | 
					 | 
				
			||||||
    options = get_options()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    mcdata = DataManager('./mcdata')
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    if options.offline:
 | 
					 | 
				
			||||||
        print("Connecting in offline mode...")
 | 
					 | 
				
			||||||
        connection = Connection(
 | 
					 | 
				
			||||||
            options.address, options.port, username=options.username)
 | 
					 | 
				
			||||||
    else:
 | 
					 | 
				
			||||||
        auth_token = authentication.AuthenticationToken()
 | 
					 | 
				
			||||||
        try:
 | 
					 | 
				
			||||||
            auth_token.authenticate(options.username, options.password)
 | 
					 | 
				
			||||||
        except YggdrasilError as e:
 | 
					 | 
				
			||||||
            print(e)
 | 
					 | 
				
			||||||
            sys.exit()
 | 
					 | 
				
			||||||
        print("Logged in as %s..." % auth_token.username)
 | 
					 | 
				
			||||||
        connection = Connection(
 | 
					 | 
				
			||||||
            options.address, options.port, auth_token=auth_token)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    if options.dump_packets:
 | 
					 | 
				
			||||||
        def print_incoming(packet):
 | 
					 | 
				
			||||||
            if type(packet) is Packet:
 | 
					 | 
				
			||||||
                # This is a direct instance of the base Packet type, meaning
 | 
					 | 
				
			||||||
                # that it is a packet of unknown type, so we do not print it
 | 
					 | 
				
			||||||
                # unless explicitly requested by the user.
 | 
					 | 
				
			||||||
                if options.dump_unknown:
 | 
					 | 
				
			||||||
                    print('--> [unknown packet] %s' % packet, file=sys.stderr)
 | 
					 | 
				
			||||||
            else:
 | 
					 | 
				
			||||||
                print('--> %s' % packet, file=sys.stderr)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        def print_outgoing(packet):
 | 
					 | 
				
			||||||
            print('<-- %s' % packet, file=sys.stderr)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        connection.register_packet_listener(
 | 
					 | 
				
			||||||
            print_incoming, Packet, early=True)
 | 
					 | 
				
			||||||
        connection.register_packet_listener(
 | 
					 | 
				
			||||||
            print_outgoing, Packet, outgoing=True)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    def handle_join_game(join_game_packet):
 | 
					 | 
				
			||||||
        print('Connected.')
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    connection.register_packet_listener(
 | 
					 | 
				
			||||||
        handle_join_game, clientbound.play.JoinGamePacket)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    def print_chat(chat_packet):
 | 
					 | 
				
			||||||
        print("Message (%s): %s" % (
 | 
					 | 
				
			||||||
            chat_packet.field_string('position'), chat_packet.json_data))
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    chunks = ChunksManager(mcdata)
 | 
					 | 
				
			||||||
    chunks.register(connection)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    connection.register_packet_listener(
 | 
					 | 
				
			||||||
        print_chat, clientbound.play.ChatMessagePacket)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    connection.connect()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    while True:
 | 
					 | 
				
			||||||
        try:
 | 
					 | 
				
			||||||
            text = input()
 | 
					 | 
				
			||||||
            if text == "/respawn":
 | 
					 | 
				
			||||||
                print("respawning...")
 | 
					 | 
				
			||||||
                packet = serverbound.play.ClientStatusPacket()
 | 
					 | 
				
			||||||
                packet.action_id = serverbound.play.ClientStatusPacket.RESPAWN
 | 
					 | 
				
			||||||
                connection.write_packet(packet)
 | 
					 | 
				
			||||||
            elif text == '!test':
 | 
					 | 
				
			||||||
                print(chunks.get_block_at(91, 65, 57))
 | 
					 | 
				
			||||||
            else:
 | 
					 | 
				
			||||||
                packet = serverbound.play.ChatPacket()
 | 
					 | 
				
			||||||
                packet.message = text
 | 
					 | 
				
			||||||
                connection.write_packet(packet)
 | 
					 | 
				
			||||||
        except KeyboardInterrupt:
 | 
					 | 
				
			||||||
            print("Bye!")
 | 
					 | 
				
			||||||
            sys.exit()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
if __name__ == "__main__":
 | 
					 | 
				
			||||||
    main()
 | 
					 | 
				
			||||||
		Reference in New Issue
	
	Block a user