Pyrkki v1

pyrkki.py

@@ -0,0 +1,246 @@
+#!/usr/bin/env python3
+#
+# Pyrkki v1, by nortti
+#
+# CC0 1.0 Universal
+#
+#     CREATIVE COMMONS CORPORATION IS NOT A LAW FIRM AND DOES NOT PROVIDE
+#     LEGAL SERVICES. DISTRIBUTION OF THIS DOCUMENT DOES NOT CREATE AN
+#     ATTORNEY-CLIENT RELATIONSHIP. CREATIVE COMMONS PROVIDES THIS
+#     INFORMATION ON AN "AS-IS" BASIS. CREATIVE COMMONS MAKES NO WARRANTIES
+#     REGARDING THE USE OF THIS DOCUMENT OR THE INFORMATION OR WORKS
+#     PROVIDED HEREUNDER, AND DISCLAIMS LIABILITY FOR DAMAGES RESULTING FROM
+#     THE USE OF THIS DOCUMENT OR THE INFORMATION OR WORKS PROVIDED
+#     HEREUNDER.
+#
+# Statement of Purpose
+#
+# The laws of most jurisdictions throughout the world automatically confer
+# exclusive Copyright and Related Rights (defined below) upon the creator
+# and subsequent owner(s) (each and all, an "owner") of an original work of
+# authorship and/or a database (each, a "Work").
+#
+# Certain owners wish to permanently relinquish those rights to a Work for
+# the purpose of contributing to a commons of creative, cultural and
+# scientific works ("Commons") that the public can reliably and without fear
+# of later claims of infringement build upon, modify, incorporate in other
+# works, reuse and redistribute as freely as possible in any form whatsoever
+# and for any purposes, including without limitation commercial purposes.
+# These owners may contribute to the Commons to promote the ideal of a free
+# culture and the further production of creative, cultural and scientific
+# works, or to gain reputation or greater distribution for their Work in
+# part through the use and efforts of others.
+#
+# For these and/or other purposes and motivations, and without any
+# expectation of additional consideration or compensation, the person
+# associating CC0 with a Work (the "Affirmer"), to the extent that he or she
+# is an owner of Copyright and Related Rights in the Work, voluntarily
+# elects to apply CC0 to the Work and publicly distribute the Work under its
+# terms, with knowledge of his or her Copyright and Related Rights in the
+# Work and the meaning and intended legal effect of CC0 on those rights.
+#
+# 1. Copyright and Related Rights. A Work made available under CC0 may be
+# protected by copyright and related or neighboring rights ("Copyright and
+# Related Rights"). Copyright and Related Rights include, but are not
+# limited to, the following:
+#
+#   i. the right to reproduce, adapt, distribute, perform, display,
+#      communicate, and translate a Work;
+#  ii. moral rights retained by the original author(s) and/or performer(s);
+# iii. publicity and privacy rights pertaining to a person's image or
+#      likeness depicted in a Work;
+#  iv. rights protecting against unfair competition in regards to a Work,
+#      subject to the limitations in paragraph 4(a), below;
+#   v. rights protecting the extraction, dissemination, use and reuse of data
+#      in a Work;
+#  vi. database rights (such as those arising under Directive 96/9/EC of the
+#      European Parliament and of the Council of 11 March 1996 on the legal
+#      protection of databases, and under any national implementation
+#      thereof, including any amended or successor version of such
+#      directive); and
+# vii. other similar, equivalent or corresponding rights throughout the
+#      world based on applicable law or treaty, and any national
+#      implementations thereof.
+#
+# 2. Waiver. To the greatest extent permitted by, but not in contravention
+# of, applicable law, Affirmer hereby overtly, fully, permanently,
+# irrevocably and unconditionally waives, abandons, and surrenders all of
+# Affirmer's Copyright and Related Rights and associated claims and causes
+# of action, whether now known or unknown (including existing as well as
+# future claims and causes of action), in the Work (i) in all territories
+# worldwide, (ii) for the maximum duration provided by applicable law or
+# treaty (including future time extensions), (iii) in any current or future
+# medium and for any number of copies, and (iv) for any purpose whatsoever,
+# including without limitation commercial, advertising or promotional
+# purposes (the "Waiver"). Affirmer makes the Waiver for the benefit of each
+# member of the public at large and to the detriment of Affirmer's heirs and
+# successors, fully intending that such Waiver shall not be subject to
+# revocation, rescission, cancellation, termination, or any other legal or
+# equitable action to disrupt the quiet enjoyment of the Work by the public
+# as contemplated by Affirmer's express Statement of Purpose.
+#
+# 3. Public License Fallback. Should any part of the Waiver for any reason
+# be judged legally invalid or ineffective under applicable law, then the
+# Waiver shall be preserved to the maximum extent permitted taking into
+# account Affirmer's express Statement of Purpose. In addition, to the
+# extent the Waiver is so judged Affirmer hereby grants to each affected
+# person a royalty-free, non transferable, non sublicensable, non exclusive,
+# irrevocable and unconditional license to exercise Affirmer's Copyright and
+# Related Rights in the Work (i) in all territories worldwide, (ii) for the
+# maximum duration provided by applicable law or treaty (including future
+# time extensions), (iii) in any current or future medium and for any number
+# of copies, and (iv) for any purpose whatsoever, including without
+# limitation commercial, advertising or promotional purposes (the
+# "License"). The License shall be deemed effective as of the date CC0 was
+# applied by Affirmer to the Work. Should any part of the License for any
+# reason be judged legally invalid or ineffective under applicable law, such
+# partial invalidity or ineffectiveness shall not invalidate the remainder
+# of the License, and in such case Affirmer hereby affirms that he or she
+# will not (i) exercise any of his or her remaining Copyright and Related
+# Rights in the Work or (ii) assert any associated claims and causes of
+# action with respect to the Work, in either case contrary to Affirmer's
+# express Statement of Purpose.
+#
+# 4. Limitations and Disclaimers.
+#
+#  a. No trademark or patent rights held by Affirmer are waived, abandoned,
+#     surrendered, licensed or otherwise affected by this document.
+#  b. Affirmer offers the Work as-is and makes no representations or
+#     warranties of any kind concerning the Work, express, implied,
+#     statutory or otherwise, including without limitation warranties of
+#     title, merchantability, fitness for a particular purpose, non
+#     infringement, or the absence of latent or other defects, accuracy, or
+#     the present or absence of errors, whether or not discoverable, all to
+#     the greatest extent permissible under applicable law.
+#  c. Affirmer disclaims responsibility for clearing rights of other persons
+#     that may apply to the Work or any use thereof, including without
+#     limitation any person's Copyright and Related Rights in the Work.
+#     Further, Affirmer disclaims responsibility for obtaining any necessary
+#     consents, permissions or other rights required for any use of the
+#     Work.
+#  d. Affirmer understands and acknowledges that Creative Commons is not a
+#     party to this document and has no duty or obligation with respect to
+#     this CC0 or use of the Work.
+
+import os
+import socket
+import sys
+import threading
+
+def sendline(sock, line):
+	assert '\n' not in line
+	sock.sendall(line.encode() + b'\r\n')
+
+class Receiver(threading.Thread):
+	def __init__(self, s):
+		self.s = s
+		super().__init__()
+		
+	def run(self):
+		buffer = bytearray()
+		while True:
+			data = self.s.recv(1024)
+			if data == b'': break
+			buffer.extend(data)
+			
+			while b'\r\n' in buffer:
+				line, _, buffer = buffer.partition(b'\r\n')
+				fields = line.split(b' ')
+				if len(fields) > 0 and fields[0] == b'PING':
+					self.s.sendall(b'PONG ' + b' '.join(fields[1:]) + b'\r\n')
+					continue
+				elif len(fields) > 1 and fields[1] == b'PRIVMSG':
+					prefix, _, target, *message = fields
+					msg_nick = prefix.split(b'!')[0][1:].decode(errors='replace')
+					target = target.decode(errors='replace')
+					if target == nick:
+						target = msg_nick
+					message = b' '.join(message).decode(errors='replace')[1:]
+					display_line = '{} <{}> {}'.format(target, msg_nick, message)
+				else:
+					display_line = line.decode(errors='replace')
+
+				sys.stdout.buffer.raw.write(display_line.encode(sys.stdout.encoding, errors='replace') + b'\n')
+
+if len(sys.argv) > 4:
+	print('{} [server [port [nick]]]'.format(sys.argv[0]))
+	sys.exit(1)
+
+if len(sys.argv) > 1:
+	host = sys.argv[1]
+else:
+	host = input('host [irc.libera.chat]> ').strip()
+	if host == '': host = 'irc.libera.chat'
+	
+if len(sys.argv) > 2:
+	try:
+		port = int(sys.argv[2])
+	except ValueError:
+		print('Port must be an intenger')
+else:
+	while True:
+		port = input('port [6667]> ').strip()
+		if port == '': port = '6667'
+		try:
+			port = int(port)
+		except ValueError:
+			print('Port must be an intenger')
+		break
+
+if 'USER' in os.environ:
+	username = os.environ['USER']
+elif 'USERNAME' in os.environ:
+	username = os.environ['USERNAME']
+else:
+	username = 'pyrkki-user'
+
+if len(sys.argv) > 3:
+	nick = sys.argv[3]
+else:
+	nick = input('nick [{}]> '.format(username)).strip()
+	if nick == '': nick = username
+
+s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+s.connect((host, port))
+sendline(s, 'nick {}'.format(nick))
+sendline(s, 'USER {} a a :{}'.format(username, username))
+
+Receiver(s).start()
+
+channel = None
+while True:
+	line = input()
+	if line == '': continue
+	if line.startswith('/'):
+		command, _, args = line[1:].lstrip().partition(' ')
+		command = command.lower()
+		if command == 'join':
+			channel = args.strip()
+			sendline(s, 'JOIN {}'.format(channel))
+		elif command == 'msg':
+			target, _, message = args.lstrip().partition(' ')
+			sendline(s, 'PRIVMSG {} :{}'.format(target, message))
+			print('{} <{}> {}'.format(target, target, message))
+		elif command == 'nick':
+			nick = args.strip()
+			sendline(s, 'NICK {}'.format(nick))
+		elif command == 'part':
+			part_channel = channel if args.strip() == '' else args.strip()
+			if part_channel == channel: channel = None
+			sendline(s, 'PART {}'.format(part_channel))
+		elif command == 'quit':
+			reason = 'pyrkki quit' if args.strip() == '' else args.strip()
+			sendline(s, 'QUIT :{}'.format(reason))
+			break
+		elif command == 'raw':
+			sendline(s, args)
+		elif command == 'target':
+			channel = args.strip()
+		else:
+			print('! Unknown command /{}'.format(command))
+	else:
+		if channel is not None:
+			sendline(s, 'PRIVMSG {} :{}'.format(channel, line))
+			print('{} <{}> {}'.format(channel, nick, line))
+		else:
+			print('! Not on a channel')