1 """Miscellaneous functions for the mudpy engine."""
3 # Copyright (c) 2004-2019 mudpy authors. Permission to use, copy,
4 # modify, and distribute this software is granted under terms
5 # provided in the LICENSE file distributed with this software.
25 """An element of the universe."""
27 def __init__(self, key, universe, origin=None):
28 """Set up a new element."""
30 # keep track of our key name
33 # keep track of what universe it's loading into
34 self.universe = universe
36 # set of facet keys from the universe
37 self.facethash = dict()
39 # not owned by a user by default (used for avatars)
42 # no contents in here by default
45 if self.key.find(".") > 0:
46 self.group, self.subkey = self.key.split(".")[-2:]
49 self.subkey = self.key
50 if self.group not in self.universe.groups:
51 self.universe.groups[self.group] = {}
53 # get an appropriate origin
55 self.universe.add_group(self.group)
56 origin = self.universe.files[
57 self.universe.origins[self.group]["fallback"]]
59 # record or reset a pointer to the origin file
60 self.origin = self.universe.files[origin.source]
62 # add or replace this element in the universe
63 self.universe.contents[self.key] = self
64 self.universe.groups[self.group][self.subkey] = self
67 """Create a new element and replace this one."""
68 args = (self.key, self.universe, self.origin)
73 """Remove an element from the universe and destroy it."""
74 for facet in dict(self.facethash):
75 self.remove_facet(facet)
76 del self.universe.groups[self.group][self.subkey]
77 del self.universe.contents[self.key]
81 """Return a list of non-inherited facets for this element."""
84 def has_facet(self, facet):
85 """Return whether the non-inherited facet exists."""
86 return facet in self.facets()
88 def remove_facet(self, facet):
89 """Remove a facet from the element."""
90 if ".".join((self.key, facet)) in self.origin.data:
91 del self.origin.data[".".join((self.key, facet))]
92 if facet in self.facethash:
93 del self.facethash[facet]
94 self.origin.modified = True
97 """Return a list of the element's inheritance lineage."""
98 if self.has_facet("inherit"):
99 ancestry = self.get("inherit")
102 for parent in ancestry[:]:
103 ancestors = self.universe.contents[parent].ancestry()
104 for ancestor in ancestors:
105 if ancestor not in ancestry:
106 ancestry.append(ancestor)
111 def get(self, facet, default=None):
112 """Retrieve values."""
116 return self.origin.data[".".join((self.key, facet))]
117 except (KeyError, TypeError):
119 if self.has_facet("inherit"):
120 for ancestor in self.ancestry():
121 if self.universe.contents[ancestor].has_facet(facet):
122 return self.universe.contents[ancestor].get(facet)
126 def set(self, facet, value):
128 if not self.origin.is_writeable() and not self.universe.loading:
129 # break if there is an attempt to update an element from a
130 # read-only file, unless the universe is in the midst of loading
131 # updated data from files
132 raise PermissionError("Altering elements in read-only files is "
134 # Coerce some values to appropriate data types
135 # TODO(fungi) Move these to a separate validation mechanism
136 if facet in ["loglevel"]:
138 elif facet in ["administrator"]:
141 # The canonical node for this facet within its origin
142 node = ".".join((self.key, facet))
144 if node not in self.origin.data or self.origin.data[node] != value:
145 # Be careful to only update the origin's contents when required,
146 # since that affects whether the backing file gets written
147 self.origin.data[node] = value
148 self.origin.modified = True
150 # Make sure this facet is included in the element's facets
151 self.facethash[facet] = self.origin.data[node]
153 def append(self, facet, value):
154 """Append value to a list."""
155 newlist = self.get(facet)
158 if type(newlist) is not list:
159 newlist = list(newlist)
160 newlist.append(value)
161 self.set(facet, newlist)
171 add_terminator=False,
174 """Convenience method to pass messages to an owner."""
187 def can_run(self, command):
188 """Check if the user can run this command object."""
190 # has to be in the commands group
191 if command not in self.universe.groups["command"].values():
194 # avatars of administrators can run any command
195 elif self.owner and self.owner.account.get("administrator"):
198 # everyone can run non-administrative commands
199 elif not command.get("administrative"):
202 # otherwise the command cannot be run by this actor
206 # pass back the result
209 def update_location(self):
210 """Make sure the location's contents contain this element."""
211 area = self.get("location")
212 if area in self.universe.contents:
213 self.universe.contents[area].contents[self.key] = self
215 def clean_contents(self):
216 """Make sure the element's contents aren't bogus."""
217 for element in self.contents.values():
218 if element.get("location") != self.key:
219 del self.contents[element.key]
221 def go_to(self, area):
222 """Relocate the element to a specific area."""
223 current = self.get("location")
224 if current and self.key in self.universe.contents[current].contents:
225 del universe.contents[current].contents[self.key]
226 if area in self.universe.contents:
227 self.set("location", area)
228 self.universe.contents[area].contents[self.key] = self
232 """Relocate the element to its default location."""
233 self.go_to(self.get("default_location"))
234 self.echo_to_location(
235 "You suddenly realize that " + self.get("name") + " is here."
238 def move_direction(self, direction):
239 """Relocate the element in a specified direction."""
240 motion = self.universe.contents["mudpy.movement.%s" % direction]
241 enter_term = motion.get("enter_term")
242 exit_term = motion.get("exit_term")
243 self.echo_to_location("%s exits %s." % (self.get("name"), exit_term))
244 self.send("You exit %s." % exit_term, add_prompt=False)
246 self.universe.contents[
247 self.get("location")].link_neighbor(direction)
249 self.echo_to_location("%s arrives from %s." % (
250 self.get("name"), enter_term))
252 def look_at(self, key):
253 """Show an element to another element."""
255 element = self.universe.contents[key]
257 name = element.get("name")
259 message += "$(cyn)" + name + "$(nrm)$(eol)"
260 description = element.get("description")
262 message += description + "$(eol)"
263 portal_list = list(element.portals().keys())
266 message += "$(cyn)[ Exits: " + ", ".join(
269 for element in self.universe.contents[
272 if element.get("is_actor") and element is not self:
273 message += "$(yel)" + element.get(
275 ) + " is here.$(nrm)$(eol)"
276 elif element is not self:
277 message += "$(grn)" + element.get(
283 """Map the portal directions for an area to neighbors."""
285 if re.match(r"""^area\.-?\d+,-?\d+,-?\d+$""", self.key):
286 coordinates = [(int(x))
287 for x in self.key.split(".")[-1].split(",")]
290 self.universe.contents["mudpy.movement.%s" % x].get("vector")
291 ) for x in self.universe.directions)
292 for portal in self.get("gridlinks"):
293 adjacent = map(lambda c, o: c + o,
294 coordinates, offsets[portal])
295 neighbor = "area." + ",".join(
296 [(str(x)) for x in adjacent]
298 if neighbor in self.universe.contents:
299 portals[portal] = neighbor
300 for facet in self.facets():
301 if facet.startswith("link_"):
302 neighbor = self.get(facet)
303 if neighbor in self.universe.contents:
304 portal = facet.split("_")[1]
305 portals[portal] = neighbor
308 def link_neighbor(self, direction):
309 """Return the element linked in a given direction."""
310 portals = self.portals()
311 if direction in portals:
312 return portals[direction]
314 def echo_to_location(self, message):
315 """Show a message to other elements in the current location."""
316 for element in self.universe.contents[
319 if element is not self:
320 element.send(message)
327 def __init__(self, filename="", load=False):
328 """Initialize the universe."""
331 self.directions = set()
335 self.reload_flag = False
336 self.setup_loglines = []
337 self.startdir = os.getcwd()
338 self.terminate_flag = False
342 possible_filenames = [
344 "/usr/local/mudpy/etc/mudpy.yaml",
345 "/usr/local/etc/mudpy.yaml",
346 "/etc/mudpy/mudpy.yaml",
349 for filename in possible_filenames:
350 if os.access(filename, os.R_OK):
352 if not os.path.isabs(filename):
353 filename = os.path.join(self.startdir, filename)
354 self.filename = filename
356 # make sure to preserve any accumulated log entries during load
357 self.setup_loglines += self.load()
360 """Load universe data from persistent storage."""
362 # while loading, it's safe to update elements from read-only files
365 # it's possible for this to enter before logging configuration is read
366 pending_loglines = []
368 # start populating the (re)files dict from the base config
370 mudpy.data.Data(self.filename, self)
372 # load default storage locations for groups
373 if hasattr(self, "contents") and "mudpy.filing" in self.contents:
374 self.origins.update(self.contents["mudpy.filing"].get(
377 # add some builtin groups we know we'll need
378 for group in ("account", "actor", "internal"):
379 self.add_group(group)
381 # make a list of inactive avatars
382 inactive_avatars = []
383 for account in self.groups.get("account", {}).values():
384 for avatar in account.get("avatars"):
386 inactive_avatars.append(self.contents[avatar])
388 pending_loglines.append((
389 'Missing avatar "%s", possible data corruption' %
391 for user in self.userlist:
392 if user.avatar in inactive_avatars:
393 inactive_avatars.remove(user.avatar)
395 # another pass to straighten out all the element contents
396 for element in self.contents.values():
397 element.update_location()
398 element.clean_contents()
400 # done loading, so disallow updating elements from read-only files
403 return pending_loglines
406 """Create a new, empty Universe (the Big Bang)."""
407 new_universe = Universe()
408 for attribute in vars(self).keys():
409 setattr(new_universe, attribute, getattr(self, attribute))
410 new_universe.reload_flag = False
415 """Save the universe to persistent storage."""
416 for key in self.files:
417 self.files[key].save()
419 def initialize_server_socket(self):
420 """Create and open the listening socket."""
422 # need to know the local address and port number for the listener
423 host = self.contents["mudpy.network"].get("host")
424 port = self.contents["mudpy.network"].get("port")
426 # if no host was specified, bind to all local addresses (preferring
434 # figure out if this is ipv4 or v6
435 family = socket.getaddrinfo(host, port)[0][0]
436 if family is socket.AF_INET6 and not socket.has_ipv6:
437 log("No support for IPv6 address %s (use IPv4 instead)." % host)
439 # create a new stream-type socket object
440 self.listening_socket = socket.socket(family, socket.SOCK_STREAM)
442 # set the socket options to allow existing open ones to be
443 # reused (fixes a bug where the server can't bind for a minute
444 # when restarting on linux systems)
445 self.listening_socket.setsockopt(
446 socket.SOL_SOCKET, socket.SO_REUSEADDR, 1
449 # bind the socket to to our desired server ipa and port
450 self.listening_socket.bind((host, port))
452 # disable blocking so we can proceed whether or not we can
454 self.listening_socket.setblocking(0)
456 # start listening on the socket
457 self.listening_socket.listen(1)
459 # note that we're now ready for user connections
460 log("Listening for Telnet connections on %s port %s" % (
464 """Convenience method to get the elapsed time counter."""
465 return self.groups["internal"]["counters"].get("elapsed")
467 def add_group(self, group, fallback=None):
468 """Set up group tracking/metadata."""
469 if group not in self.origins:
470 self.origins[group] = {}
472 fallback = mudpy.data.find_file(
473 ".".join((group, "yaml")), universe=self)
474 if "fallback" not in self.origins[group]:
475 self.origins[group]["fallback"] = fallback
476 flags = self.origins[group].get("flags", None)
477 if fallback not in self.files:
478 mudpy.data.Data(fallback, self, flags=flags)
483 """This is a connected user."""
486 """Default values for the in-memory user variables."""
489 self.authenticated = False
493 self.connection = None
495 self.input_queue = []
496 self.last_address = ""
497 self.last_input = universe.get_time()
498 self.menu_choices = {}
499 self.menu_seen = False
500 self.negotiation_pause = 0
501 self.output_queue = []
502 self.partial_input = b""
503 self.password_tries = 0
504 self.state = "telopt_negotiation"
506 self.universe = universe
509 """Log, close the connection and remove."""
511 name = self.account.get("name", self)
514 log("Logging out %s" % name, 2)
515 self.deactivate_avatar()
516 self.connection.close()
519 def check_idle(self):
520 """Warn or disconnect idle users as appropriate."""
521 idletime = universe.get_time() - self.last_input
522 linkdead_dict = universe.contents[
523 "mudpy.timing.idle.disconnect"].facets()
524 if self.state in linkdead_dict:
525 linkdead_state = self.state
527 linkdead_state = "default"
528 if idletime > linkdead_dict[linkdead_state]:
530 "$(eol)$(red)You've done nothing for far too long... goodbye!"
535 logline = "Disconnecting "
536 if self.account and self.account.get("name"):
537 logline += self.account.get("name")
539 logline += "an unknown user"
540 logline += (" after idling too long in the " + self.state
543 self.state = "disconnecting"
544 self.menu_seen = False
545 idle_dict = universe.contents["mudpy.timing.idle.warn"].facets()
546 if self.state in idle_dict:
547 idle_state = self.state
549 idle_state = "default"
550 if idletime == idle_dict[idle_state]:
552 "$(eol)$(red)If you continue to be unproductive, "
553 + "you'll be shown the door...$(nrm)$(eol)"
557 """Save, load a new user and relocate the connection."""
559 # copy old attributes
560 attributes = self.__dict__
562 # get out of the list
565 # get rid of the old user object
568 # create a new user object
571 # set everything equivalent
572 new_user.__dict__ = attributes
574 # the avatar needs a new owner
576 new_user.account = universe.contents[new_user.account.key]
577 new_user.avatar = universe.contents[new_user.avatar.key]
578 new_user.avatar.owner = new_user
581 universe.userlist.append(new_user)
583 def replace_old_connections(self):
584 """Disconnect active users with the same name."""
586 # the default return value
589 # iterate over each user in the list
590 for old_user in universe.userlist:
592 # the name is the same but it's not us
595 ) and old_user.account and old_user.account.get(
597 ) == self.account.get(
599 ) and old_user is not self:
603 "User " + self.account.get(
605 ) + " reconnected--closing old connection to "
606 + old_user.address + ".",
610 "$(eol)$(red)New connection from " + self.address
611 + ". Terminating old connection...$(nrm)$(eol)",
616 # close the old connection
617 old_user.connection.close()
619 # replace the old connection with this one
621 "$(eol)$(red)Taking over old connection from "
622 + old_user.address + ".$(nrm)"
624 old_user.connection = self.connection
625 old_user.last_address = old_user.address
626 old_user.address = self.address
628 # take this one out of the list and delete
634 # true if an old connection was replaced, false if not
637 def authenticate(self):
638 """Flag the user as authenticated and disconnect duplicates."""
639 if self.state != "authenticated":
640 self.authenticated = True
641 log("User %s authenticated for account %s." % (
642 self, self.account.subkey), 2)
643 if ("mudpy.limit" in universe.contents and self.account.subkey in
644 universe.contents["mudpy.limit"].get("admins")):
645 self.account.set("administrator", True)
646 log("Account %s is an administrator." % (
647 self.account.subkey), 2)
650 """Send the user their current menu."""
651 if not self.menu_seen:
652 self.menu_choices = get_menu_choices(self)
654 get_menu(self.state, self.error, self.menu_choices),
658 self.menu_seen = True
660 self.adjust_echoing()
663 """"Generate and return an input prompt."""
665 # Start with the user's preference, if one was provided
666 prompt = self.account.get("prompt")
668 # If the user has not set a prompt, then immediately return the default
669 # provided for the current state
671 return get_menu_prompt(self.state)
673 # Allow including the World clock state
674 if "$_(time)" in prompt:
675 prompt = prompt.replace(
677 str(universe.groups["internal"]["counters"].get("elapsed")))
679 # Append a single space for clear separation from user input
680 if prompt[-1] != " ":
681 prompt = "%s " % prompt
683 # Return the cooked prompt
686 def adjust_echoing(self):
687 """Adjust echoing to match state menu requirements."""
688 if mudpy.telnet.is_enabled(self, mudpy.telnet.TELOPT_ECHO,
690 if menu_echo_on(self.state):
691 mudpy.telnet.disable(self, mudpy.telnet.TELOPT_ECHO,
693 elif not menu_echo_on(self.state):
694 mudpy.telnet.enable(self, mudpy.telnet.TELOPT_ECHO,
698 """Remove a user from the list of connected users."""
699 log("Disconnecting account %s." % self, 0)
700 universe.userlist.remove(self)
710 add_terminator=False,
713 """Send arbitrary text to a connected user."""
715 # unless raw mode is on, clean it up all nice and pretty
718 # strip extra $(eol) off if present
719 while output.startswith("$(eol)"):
721 while output.endswith("$(eol)"):
723 extra_lines = output.find("$(eol)$(eol)$(eol)")
724 while extra_lines > -1:
725 output = output[:extra_lines] + output[extra_lines + 6:]
726 extra_lines = output.find("$(eol)$(eol)$(eol)")
728 # start with a newline, append the message, then end
729 # with the optional eol string passed to this function
730 # and the ansi escape to return to normal text
731 if not just_prompt and prepend_padding:
732 if (not self.output_queue or not
733 self.output_queue[-1].endswith(b"\r\n")):
734 output = "$(eol)" + output
735 elif not self.output_queue[-1].endswith(
737 ) and not self.output_queue[-1].endswith(
740 output = "$(eol)" + output
741 output += eol + chr(27) + "[0m"
743 # tack on a prompt if active
744 if self.state == "active":
748 output += self.prompt()
749 mode = self.avatar.get("mode")
751 output += "(" + mode + ") "
753 # find and replace macros in the output
754 output = replace_macros(self, output)
756 # wrap the text at the client's width (min 40, 0 disables)
758 if self.columns < 40:
762 output = wrap_ansi_text(output, wrap)
764 # if supported by the client, encode it utf-8
765 if mudpy.telnet.is_enabled(self, mudpy.telnet.TELOPT_BINARY,
767 encoded_output = output.encode("utf-8")
769 # otherwise just send ascii
771 encoded_output = output.encode("ascii", "replace")
773 # end with a terminator if requested
774 if add_prompt or add_terminator:
775 if mudpy.telnet.is_enabled(
776 self, mudpy.telnet.TELOPT_EOR, mudpy.telnet.US):
777 encoded_output += mudpy.telnet.telnet_proto(
778 mudpy.telnet.IAC, mudpy.telnet.EOR)
779 elif not mudpy.telnet.is_enabled(
780 self, mudpy.telnet.TELOPT_SGA, mudpy.telnet.US):
781 encoded_output += mudpy.telnet.telnet_proto(
782 mudpy.telnet.IAC, mudpy.telnet.GA)
784 # and tack it onto the queue
785 self.output_queue.append(encoded_output)
787 # if this is urgent, flush all pending output
791 # just dump raw bytes as requested
793 self.output_queue.append(output)
797 """All the things to do to the user per increment."""
799 # if the world is terminating, disconnect
800 if universe.terminate_flag:
801 self.state = "disconnecting"
802 self.menu_seen = False
804 # check for an idle connection and act appropriately
808 # if output is paused, decrement the counter
809 if self.state == "telopt_negotiation":
810 if self.negotiation_pause:
811 self.negotiation_pause -= 1
813 self.state = "entering_account_name"
815 # show the user a menu as needed
816 elif not self.state == "active":
819 # flush any pending output in the queue
822 # disconnect users with the appropriate state
823 if self.state == "disconnecting":
826 # check for input and add it to the queue
829 # there is input waiting in the queue
831 handle_user_input(self)
834 """Try to send the last item in the queue and remove it."""
835 if self.output_queue:
837 self.connection.send(self.output_queue[0])
838 except (BrokenPipeError, ConnectionResetError):
839 if self.account and self.account.get("name"):
840 account = self.account.get("name")
842 account = "an unknown user"
843 self.state = "disconnecting"
844 log("Disconnected while sending to %s." % account, 7)
845 del self.output_queue[0]
847 def enqueue_input(self):
848 """Process and enqueue any new input."""
850 # check for some input
852 raw_input = self.connection.recv(1024)
853 except (BlockingIOError, OSError):
859 # tack this on to any previous partial
860 self.partial_input += raw_input
862 # reply to and remove any IAC negotiation codes
863 mudpy.telnet.negotiate_telnet_options(self)
865 # separate multiple input lines
866 new_input_lines = self.partial_input.split(b"\r\0")
867 if len(new_input_lines) == 1:
868 new_input_lines = new_input_lines[0].split(b"\r\n")
870 # if input doesn't end in a newline, replace the
871 # held partial input with the last line of it
873 self.partial_input.endswith(b"\r\0") or
874 self.partial_input.endswith(b"\r\n")):
875 self.partial_input = new_input_lines.pop()
877 # otherwise, chop off the extra null input and reset
878 # the held partial input
880 new_input_lines.pop()
881 self.partial_input = b""
883 # iterate over the remaining lines
884 for line in new_input_lines:
886 # strip off extra whitespace
889 # log non-printable characters remaining
890 if not mudpy.telnet.is_enabled(
891 self, mudpy.telnet.TELOPT_BINARY, mudpy.telnet.HIM):
892 asciiline = bytes([x for x in line if 32 <= x <= 126])
893 if line != asciiline:
894 logline = "Non-ASCII characters from "
895 if self.account and self.account.get("name"):
896 logline += self.account.get("name") + ": "
898 logline += "unknown user: "
899 logline += repr(line)
904 line = line.decode("utf-8")
905 except UnicodeDecodeError:
906 logline = "Non-UTF-8 sequence from "
907 if self.account and self.account.get("name"):
908 logline += self.account.get("name") + ": "
910 logline += "unknown user: "
911 logline += repr(line)
915 line = unicodedata.normalize("NFKC", line)
917 # put on the end of the queue
918 self.input_queue.append(line)
920 def new_avatar(self):
921 """Instantiate a new, unconfigured avatar for this user."""
923 while ("avatar_%s_%s" % (self.account.get("name"), counter)
924 in universe.groups.get("actor", {}).keys()):
926 self.avatar = Element(
927 "actor.avatar_%s_%s" % (self.account.get("name"), counter),
929 self.avatar.append("inherit", "archetype.avatar")
930 self.account.append("avatars", self.avatar.key)
931 log("Created new avatar %s for user %s." % (
932 self.avatar.key, self.account.get("name")), 0)
934 def delete_avatar(self, avatar):
935 """Remove an avatar from the world and from the user's list."""
936 if self.avatar is universe.contents[avatar]:
938 log("Deleting avatar %s for user %s." % (
939 avatar, self.account.get("name")), 0)
940 universe.contents[avatar].destroy()
941 avatars = self.account.get("avatars")
942 avatars.remove(avatar)
943 self.account.set("avatars", avatars)
945 def activate_avatar_by_index(self, index):
946 """Enter the world with a particular indexed avatar."""
947 self.avatar = universe.contents[
948 self.account.get("avatars")[index]]
949 self.avatar.owner = self
950 self.state = "active"
951 log("Activated avatar %s (%s)." % (
952 self.avatar.get("name"), self.avatar.key), 0)
953 self.avatar.go_home()
955 def deactivate_avatar(self):
956 """Have the active avatar leave the world."""
958 log("Deactivating avatar %s (%s) for user %s." % (
959 self.avatar.get("name"), self.avatar.key,
960 self.account.get("name")), 0)
961 current = self.avatar.get("location")
963 self.avatar.set("default_location", current)
964 self.avatar.echo_to_location(
965 "You suddenly wonder where " + self.avatar.get(
969 del universe.contents[current].contents[self.avatar.key]
970 self.avatar.remove_facet("location")
971 self.avatar.owner = None
975 """Destroy the user and associated avatars."""
976 for avatar in self.account.get("avatars"):
977 self.delete_avatar(avatar)
978 log("Destroying account %s for user %s." % (
979 self.account.get("name"), self), 0)
980 self.account.destroy()
982 def list_avatar_names(self):
983 """List names of assigned avatars."""
985 for avatar in self.account.get("avatars"):
987 avatars.append(universe.contents[avatar].get("name"))
989 log('Missing avatar "%s", possible data corruption.' %
994 def broadcast(message, add_prompt=True):
995 """Send a message to all connected users."""
996 for each_user in universe.userlist:
997 each_user.send("$(eol)" + message, add_prompt=add_prompt)
1000 def log(message, level=0):
1001 """Log a message."""
1003 # a couple references we need
1004 if "mudpy.log" in universe.contents:
1005 file_name = universe.contents["mudpy.log"].get("file", "")
1006 max_log_lines = universe.contents["mudpy.log"].get("lines", 0)
1007 syslog_name = universe.contents["mudpy.log"].get("syslog", "")
1012 timestamp = datetime.datetime.now().isoformat(' ')
1014 # turn the message into a list of nonempty lines
1015 lines = [x for x in [(x.rstrip()) for x in message.split("\n")] if x != ""]
1017 # send the timestamp and line to a file
1019 if not os.path.isabs(file_name):
1020 file_name = os.path.join(universe.startdir, file_name)
1021 os.makedirs(os.path.dirname(file_name), exist_ok=True)
1022 file_descriptor = codecs.open(file_name, "a", "utf-8")
1024 file_descriptor.write(timestamp + " " + line + "\n")
1025 file_descriptor.flush()
1026 file_descriptor.close()
1028 # send the timestamp and line to standard output
1029 if ("mudpy.log" in universe.contents and
1030 universe.contents["mudpy.log"].get("stdout")):
1032 print(timestamp + " " + line)
1034 # send the line to the system log
1037 syslog_name.encode("utf-8"),
1039 syslog.LOG_INFO | syslog.LOG_DAEMON
1045 # display to connected administrators
1046 for user in universe.userlist:
1047 if user.state == "active" and user.account.get(
1049 ) and user.account.get("loglevel", 0) <= level:
1050 # iterate over every line in the message
1054 "$(bld)$(red)" + timestamp + " "
1055 + line.replace("$(", "$_(") + "$(nrm)$(eol)")
1056 user.send(full_message, flush=True)
1058 # add to the recent log list
1060 while 0 < len(universe.loglines) >= max_log_lines:
1061 del universe.loglines[0]
1062 universe.loglines.append((level, timestamp + " " + line))
1065 def get_loglines(level, start, stop):
1066 """Return a specific range of loglines filtered by level."""
1068 # filter the log lines
1069 loglines = [x for x in universe.loglines if x[0] >= level]
1071 # we need these in several places
1072 total_count = str(len(universe.loglines))
1073 filtered_count = len(loglines)
1075 # don't proceed if there are no lines
1078 # can't start before the begining or at the end
1079 if start > filtered_count:
1080 start = filtered_count
1084 # can't stop before we start
1091 message = "There are " + str(total_count)
1092 message += " log lines in memory and " + str(filtered_count)
1093 message += " at or above level " + str(level) + "."
1094 message += " The matching lines from " + str(stop) + " to "
1095 message += str(start) + " are:$(eol)$(eol)"
1097 # add the text from the selected lines
1099 range_lines = loglines[-start:-(stop - 1)]
1101 range_lines = loglines[-start:]
1102 for line in range_lines:
1103 message += " (" + str(line[0]) + ") " + line[1].replace(
1107 # there were no lines
1109 message = "None of the " + str(total_count)
1110 message += " lines in memory matches your request."
1116 def glyph_columns(character):
1117 """Convenience function to return the column width of a glyph."""
1118 if unicodedata.east_asian_width(character) in "FW":
1124 def wrap_ansi_text(text, width):
1125 """Wrap text with arbitrary width while ignoring ANSI colors."""
1127 # the current position in the entire text string, including all
1128 # characters, printable or otherwise
1131 # the current text position relative to the begining of the line,
1132 # ignoring color escape sequences
1135 # the absolute and relative positions of the most recent whitespace
1137 last_abs_whitespace = 0
1138 last_rel_whitespace = 0
1140 # whether the current character is part of a color escape sequence
1143 # normalize any potentially composited unicode before we count it
1144 text = unicodedata.normalize("NFKC", text)
1146 # iterate over each character from the begining of the text
1147 for each_character in text:
1149 # the current character is the escape character
1150 if each_character == "\x1b" and not escape:
1154 # the current character is within an escape sequence
1157 if each_character == "m":
1158 # the current character is m, which terminates the
1162 # the current character is a space
1163 elif each_character == " ":
1164 last_abs_whitespace = abs_pos
1165 last_rel_whitespace = rel_pos
1167 # the current character is a newline, so reset the relative
1168 # position too (start a new line)
1169 elif each_character == "\n":
1171 last_abs_whitespace = abs_pos
1172 last_rel_whitespace = rel_pos
1174 # the current character meets the requested maximum line width, so we
1175 # need to wrap unless the current word is wider than the terminal (in
1176 # which case we let it do the wrapping instead)
1177 if last_rel_whitespace != 0 and (rel_pos > width or (
1178 rel_pos > width - 1 and glyph_columns(each_character) == 2)):
1180 # insert an eol in place of the last space
1181 text = (text[:last_abs_whitespace] + "\r\n" +
1182 text[last_abs_whitespace + 1:])
1184 # increase the absolute position because an eol is two
1185 # characters but the space it replaced was only one
1188 # now we're at the begining of a new line, plus the
1189 # number of characters wrapped from the previous line
1190 rel_pos -= last_rel_whitespace
1191 last_rel_whitespace = 0
1193 # as long as the character is not a carriage return and the
1194 # other above conditions haven't been met, count it as a
1195 # printable character
1196 elif each_character != "\r":
1197 rel_pos += glyph_columns(each_character)
1198 if each_character in (" ", "\n"):
1199 last_abs_whitespace = abs_pos
1200 last_rel_whitespace = rel_pos
1202 # increase the absolute position for every character
1205 # return the newly-wrapped text
1209 def weighted_choice(data):
1210 """Takes a dict weighted by value and returns a random key."""
1212 # this will hold our expanded list of keys from the data
1215 # create the expanded list of keys
1216 for key in data.keys():
1217 for _count in range(data[key]):
1218 expanded.append(key)
1220 # return one at random
1221 # Whitelist the random.randrange() call in bandit since it's not used for
1222 # security/cryptographic purposes
1223 return random.choice(expanded) # nosec
1227 """Returns a random character name."""
1229 # the vowels and consonants needed to create romaji syllables
1258 # this dict will hold our weighted list of syllables
1261 # generate the list with an even weighting
1262 for consonant in consonants:
1263 for vowel in vowels:
1264 syllables[consonant + vowel] = 1
1266 # we'll build the name into this string
1269 # create a name of random length from the syllables
1270 # Whitelist the random.randrange() call in bandit since it's not used for
1271 # security/cryptographic purposes
1272 for _syllable in range(random.randrange(2, 6)): # nosec
1273 name += weighted_choice(syllables)
1275 # strip any leading quotemark, capitalize and return the name
1276 return name.strip("'").capitalize()
1279 def replace_macros(user, text, is_input=False):
1280 """Replaces macros in text output."""
1282 # third person pronouns
1284 "female": {"obj": "her", "pos": "hers", "sub": "she"},
1285 "male": {"obj": "him", "pos": "his", "sub": "he"},
1286 "neuter": {"obj": "it", "pos": "its", "sub": "it"}
1289 # a dict of replacement macros
1292 "bld": chr(27) + "[1m",
1293 "nrm": chr(27) + "[0m",
1294 "blk": chr(27) + "[30m",
1295 "blu": chr(27) + "[34m",
1296 "cyn": chr(27) + "[36m",
1297 "grn": chr(27) + "[32m",
1298 "mgt": chr(27) + "[35m",
1299 "red": chr(27) + "[31m",
1300 "yel": chr(27) + "[33m",
1303 # add dynamic macros where possible
1305 account_name = user.account.get("name")
1307 macros["account"] = account_name
1309 avatar_gender = user.avatar.get("gender")
1311 macros["tpop"] = pronouns[avatar_gender]["obj"]
1312 macros["tppp"] = pronouns[avatar_gender]["pos"]
1313 macros["tpsp"] = pronouns[avatar_gender]["sub"]
1318 # find and replace per the macros dict
1319 macro_start = text.find("$(")
1320 if macro_start == -1:
1322 macro_end = text.find(")", macro_start) + 1
1323 macro = text[macro_start + 2:macro_end - 1]
1324 if macro in macros.keys():
1325 replacement = macros[macro]
1327 # this is how we handle local file inclusion (dangerous!)
1328 elif macro.startswith("inc:"):
1329 incfile = mudpy.data.find_file(macro[4:], universe=universe)
1330 if os.path.exists(incfile):
1331 incfd = codecs.open(incfile, "r", "utf-8")
1334 if line.endswith("\n") and not line.endswith("\r\n"):
1335 line = line.replace("\n", "\r\n")
1337 # lose the trailing eol
1338 replacement = replacement[:-2]
1341 log("Couldn't read included " + incfile + " file.", 7)
1343 # if we get here, log and replace it with null
1347 log("Unexpected replacement macro " +
1348 macro + " encountered.", 6)
1350 # and now we act on the replacement
1351 text = text.replace("$(" + macro + ")", replacement)
1353 # replace the look-like-a-macro sequence
1354 text = text.replace("$_(", "$(")
1359 def escape_macros(value):
1360 """Escapes replacement macros in text."""
1361 if type(value) is str:
1362 return value.replace("$(", "$_(")
1367 def first_word(text, separator=" "):
1368 """Returns a tuple of the first word and the rest."""
1370 if text.find(separator) > 0:
1371 return text.split(separator, 1)
1379 """The things which should happen on each pulse, aside from reloads."""
1381 # open the listening socket if it hasn't been already
1382 if not hasattr(universe, "listening_socket"):
1383 universe.initialize_server_socket()
1385 # assign a user if a new connection is waiting
1386 user = check_for_connection(universe.listening_socket)
1388 universe.userlist.append(user)
1390 # iterate over the connected users
1391 for user in universe.userlist:
1394 # add an element for counters if it doesn't exist
1395 if "counters" not in universe.groups.get("internal", {}):
1396 Element("internal.counters", universe)
1398 # update the log every now and then
1399 if not universe.groups["internal"]["counters"].get("mark"):
1400 log(str(len(universe.userlist)) + " connection(s)")
1401 universe.groups["internal"]["counters"].set(
1402 "mark", universe.contents["mudpy.timing"].get("status")
1405 universe.groups["internal"]["counters"].set(
1406 "mark", universe.groups["internal"]["counters"].get(
1411 # periodically save everything
1412 if not universe.groups["internal"]["counters"].get("save"):
1414 universe.groups["internal"]["counters"].set(
1415 "save", universe.contents["mudpy.timing"].get("save")
1418 universe.groups["internal"]["counters"].set(
1419 "save", universe.groups["internal"]["counters"].get(
1424 # pause for a configurable amount of time (decimal seconds)
1425 time.sleep(universe.contents["mudpy.timing"].get("increment"))
1427 # increase the elapsed increment counter
1428 universe.groups["internal"]["counters"].set(
1429 "elapsed", universe.groups["internal"]["counters"].get(
1436 """Reload all relevant objects."""
1438 old_userlist = universe.userlist[:]
1439 old_loglines = universe.loglines[:]
1440 for element in list(universe.contents.values()):
1443 new_loglines = universe.loglines[:]
1444 universe.loglines = old_loglines + new_loglines
1445 for user in old_userlist:
1449 def check_for_connection(listening_socket):
1450 """Check for a waiting connection and return a new user object."""
1452 # try to accept a new connection
1454 connection, address = listening_socket.accept()
1455 except BlockingIOError:
1458 # note that we got one
1459 log("New connection from %s." % address[0], 2)
1461 # disable blocking so we can proceed whether or not we can send/receive
1462 connection.setblocking(0)
1464 # create a new user object
1466 log("Instantiated %s for %s." % (user, address[0]), 0)
1468 # associate this connection with it
1469 user.connection = connection
1471 # set the user's ipa from the connection's ipa
1472 user.address = address[0]
1474 # let the client know we WILL EOR (RFC 885)
1475 mudpy.telnet.enable(user, mudpy.telnet.TELOPT_EOR, mudpy.telnet.US)
1476 user.negotiation_pause = 2
1478 # return the new user object
1482 def find_command(command_name):
1483 """Try to find a command by name or abbreviation."""
1485 # lowercase the command
1486 command_name = command_name.lower()
1489 if command_name in universe.groups["command"]:
1490 # the command matches a command word for which we have data
1491 command = universe.groups["command"][command_name]
1493 for candidate in sorted(universe.groups["command"]):
1494 if candidate.startswith(command_name) and not universe.groups[
1495 "command"][candidate].get("administrative"):
1496 # the command matches the start of a command word and is not
1497 # restricted to administrators
1498 command = universe.groups["command"][candidate]
1503 def get_menu(state, error=None, choices=None):
1504 """Show the correct menu text to a user."""
1506 # make sure we don't reuse a mutable sequence by default
1510 # get the description or error text
1511 message = get_menu_description(state, error)
1513 # get menu choices for the current state
1514 message += get_formatted_menu_choices(state, choices)
1516 # try to get a prompt, if it was defined
1517 message += get_menu_prompt(state)
1519 # throw in the default choice, if it exists
1520 message += get_formatted_default_menu_choice(state)
1522 # display a message indicating if echo is off
1523 message += get_echo_message(state)
1525 # return the assembly of various strings defined above
1529 def menu_echo_on(state):
1530 """True if echo is on, false if it is off."""
1531 return universe.groups["menu"][state].get("echo", True)
1534 def get_echo_message(state):
1535 """Return a message indicating that echo is off."""
1536 if menu_echo_on(state):
1539 return "(won't echo) "
1542 def get_default_menu_choice(state):
1543 """Return the default choice for a menu."""
1544 return universe.groups["menu"][state].get("default")
1547 def get_formatted_default_menu_choice(state):
1548 """Default menu choice foratted for inclusion in a prompt string."""
1549 default_choice = get_default_menu_choice(state)
1551 return "[$(red)" + default_choice + "$(nrm)] "
1556 def get_menu_description(state, error):
1557 """Get the description or error text."""
1559 # an error condition was raised by the handler
1562 # try to get an error message matching the condition
1564 description = universe.groups[
1565 "menu"][state].get("error_" + error)
1567 description = "That is not a valid choice..."
1568 description = "$(red)" + description + "$(nrm)"
1570 # there was no error condition
1573 # try to get a menu description for the current state
1574 description = universe.groups["menu"][state].get("description")
1576 # return the description or error message
1578 description += "$(eol)$(eol)"
1582 def get_menu_prompt(state):
1583 """Try to get a prompt, if it was defined."""
1584 prompt = universe.groups["menu"][state].get("prompt")
1590 def get_menu_choices(user):
1591 """Return a dict of choice:meaning."""
1592 state = universe.groups["menu"][user.state]
1593 create_choices = state.get("create")
1595 choices = call_hook_function(create_choices, (user,))
1601 for facet in state.facets():
1602 if facet.startswith("demand_") and not call_hook_function(
1603 universe.groups["menu"][user.state].get(facet), (user,)):
1604 ignores.append(facet.split("_", 2)[1])
1605 elif facet.startswith("create_"):
1606 creates[facet] = facet.split("_", 2)[1]
1607 elif facet.startswith("choice_"):
1608 options[facet] = facet.split("_", 2)[1]
1609 for facet in creates.keys():
1610 if not creates[facet] in ignores:
1611 choices[creates[facet]] = call_hook_function(
1612 state.get(facet), (user,))
1613 for facet in options.keys():
1614 if not options[facet] in ignores:
1615 choices[options[facet]] = state.get(facet)
1619 def get_formatted_menu_choices(state, choices):
1620 """Returns a formatted string of menu choices."""
1622 choice_keys = list(choices.keys())
1624 for choice in choice_keys:
1625 choice_output += " [$(red)" + choice + "$(nrm)] " + choices[
1629 choice_output += "$(eol)"
1630 return choice_output
1633 def get_menu_branches(state):
1634 """Return a dict of choice:branch."""
1636 for facet in universe.groups["menu"][state].facets():
1637 if facet.startswith("branch_"):
1639 facet.split("_", 2)[1]
1640 ] = universe.groups["menu"][state].get(facet)
1644 def get_default_branch(state):
1645 """Return the default branch."""
1646 return universe.groups["menu"][state].get("branch")
1649 def get_choice_branch(user):
1650 """Returns the new state matching the given choice."""
1651 branches = get_menu_branches(user.state)
1652 if user.choice in branches.keys():
1653 return branches[user.choice]
1654 elif user.choice in user.menu_choices.keys():
1655 return get_default_branch(user.state)
1660 def get_menu_actions(state):
1661 """Return a dict of choice:branch."""
1663 for facet in universe.groups["menu"][state].facets():
1664 if facet.startswith("action_"):
1666 facet.split("_", 2)[1]
1667 ] = universe.groups["menu"][state].get(facet)
1671 def get_default_action(state):
1672 """Return the default action."""
1673 return universe.groups["menu"][state].get("action")
1676 def get_choice_action(user):
1677 """Run any indicated script for the given choice."""
1678 actions = get_menu_actions(user.state)
1679 if user.choice in actions.keys():
1680 return actions[user.choice]
1681 elif user.choice in user.menu_choices.keys():
1682 return get_default_action(user.state)
1687 def call_hook_function(fname, arglist):
1688 """Safely execute named function with supplied arguments, return result."""
1690 # all functions relative to mudpy package
1693 for component in fname.split("."):
1695 function = getattr(function, component)
1696 except AttributeError:
1697 log('Could not find mudpy.%s() for arguments "%s"'
1698 % (fname, arglist), 7)
1703 return function(*arglist)
1705 log('Calling mudpy.%s(%s) raised an exception...\n%s'
1706 % (fname, (*arglist,), traceback.format_exc()), 7)
1709 def handle_user_input(user):
1710 """The main handler, branches to a state-specific handler."""
1712 # if the user's client echo is off, send a blank line for aesthetics
1713 if mudpy.telnet.is_enabled(user, mudpy.telnet.TELOPT_ECHO,
1715 user.send("", add_prompt=False, prepend_padding=False)
1717 # check to make sure the state is expected, then call that handler
1719 globals()["handler_" + user.state](user)
1721 generic_menu_handler(user)
1723 # since we got input, flag that the menu/prompt needs to be redisplayed
1724 user.menu_seen = False
1726 # update the last_input timestamp while we're at it
1727 user.last_input = universe.get_time()
1730 def generic_menu_handler(user):
1731 """A generic menu choice handler."""
1733 # get a lower-case representation of the next line of input
1734 if user.input_queue:
1735 user.choice = user.input_queue.pop(0)
1737 user.choice = user.choice.lower()
1741 user.choice = get_default_menu_choice(user.state)
1742 if user.choice in user.menu_choices:
1743 action = get_choice_action(user)
1745 call_hook_function(action, (user,))
1746 new_state = get_choice_branch(user)
1748 user.state = new_state
1750 user.error = "default"
1753 def handler_entering_account_name(user):
1754 """Handle the login account name."""
1756 # get the next waiting line of input
1757 input_data = user.input_queue.pop(0)
1759 # did the user enter anything?
1762 # keep only the first word and convert to lower-case
1763 name = input_data.lower()
1765 # fail if there are non-alphanumeric characters
1766 if name != "".join(filter(
1767 lambda x: x >= "0" and x <= "9" or x >= "a" and x <= "z",
1769 user.error = "bad_name"
1771 # if that account exists, time to request a password
1772 elif name in universe.groups.get("account", {}):
1773 user.account = universe.groups["account"][name]
1774 user.state = "checking_password"
1776 # otherwise, this could be a brand new user
1778 user.account = Element("account.%s" % name, universe)
1779 user.account.set("name", name)
1780 log("New user: " + name, 2)
1781 user.state = "checking_new_account_name"
1783 # if the user entered nothing for a name, then buhbye
1785 user.state = "disconnecting"
1788 def handler_checking_password(user):
1789 """Handle the login account password."""
1791 # get the next waiting line of input
1792 input_data = user.input_queue.pop(0)
1794 if "mudpy.limit" in universe.contents:
1795 max_password_tries = universe.contents["mudpy.limit"].get(
1796 "password_tries", 3)
1798 max_password_tries = 3
1800 # does the hashed input equal the stored hash?
1801 if mudpy.password.verify(input_data, user.account.get("passhash")):
1803 # if so, set the username and load from cold storage
1804 if not user.replace_old_connections():
1806 user.state = "main_utility"
1808 # if at first your hashes don't match, try, try again
1809 elif user.password_tries < max_password_tries - 1:
1810 user.password_tries += 1
1811 user.error = "incorrect"
1813 # we've exceeded the maximum number of password failures, so disconnect
1816 "$(eol)$(red)Too many failed password attempts...$(nrm)$(eol)"
1818 user.state = "disconnecting"
1821 def handler_entering_new_password(user):
1822 """Handle a new password entry."""
1824 # get the next waiting line of input
1825 input_data = user.input_queue.pop(0)
1827 if "mudpy.limit" in universe.contents:
1828 max_password_tries = universe.contents["mudpy.limit"].get(
1829 "password_tries", 3)
1831 max_password_tries = 3
1833 # make sure the password is strong--at least one upper, one lower and
1834 # one digit, seven or more characters in length
1835 if len(input_data) > 6 and len(
1836 list(filter(lambda x: x >= "0" and x <= "9", input_data))
1838 list(filter(lambda x: x >= "A" and x <= "Z", input_data))
1840 list(filter(lambda x: x >= "a" and x <= "z", input_data))
1843 # hash and store it, then move on to verification
1844 user.account.set("passhash", mudpy.password.create(input_data))
1845 user.state = "verifying_new_password"
1847 # the password was weak, try again if you haven't tried too many times
1848 elif user.password_tries < max_password_tries - 1:
1849 user.password_tries += 1
1852 # too many tries, so adios
1855 "$(eol)$(red)Too many failed password attempts...$(nrm)$(eol)"
1857 user.account.destroy()
1858 user.state = "disconnecting"
1861 def handler_verifying_new_password(user):
1862 """Handle the re-entered new password for verification."""
1864 # get the next waiting line of input
1865 input_data = user.input_queue.pop(0)
1867 if "mudpy.limit" in universe.contents:
1868 max_password_tries = universe.contents["mudpy.limit"].get(
1869 "password_tries", 3)
1871 max_password_tries = 3
1873 # hash the input and match it to storage
1874 if mudpy.password.verify(input_data, user.account.get("passhash")):
1877 # the hashes matched, so go active
1878 if not user.replace_old_connections():
1879 user.state = "main_utility"
1881 # go back to entering the new password as long as you haven't tried
1883 elif user.password_tries < max_password_tries - 1:
1884 user.password_tries += 1
1885 user.error = "differs"
1886 user.state = "entering_new_password"
1888 # otherwise, sayonara
1891 "$(eol)$(red)Too many failed password attempts...$(nrm)$(eol)"
1893 user.account.destroy()
1894 user.state = "disconnecting"
1897 def handler_active(user):
1898 """Handle input for active users."""
1900 # get the next waiting line of input
1901 input_data = user.input_queue.pop(0)
1906 # split out the command and parameters
1908 mode = actor.get("mode")
1909 if mode and input_data.startswith("!"):
1910 command_name, parameters = first_word(input_data[1:])
1911 elif mode == "chat":
1912 command_name = "say"
1913 parameters = input_data
1915 command_name, parameters = first_word(input_data)
1917 # expand to an actual command
1918 command = find_command(command_name)
1920 # if it's allowed, do it
1922 if actor.can_run(command):
1923 action_fname = command.get("action", command.key)
1925 result = call_hook_function(action_fname, (actor, parameters))
1927 # if the command was not run, give an error
1929 mudpy.command.error(actor, input_data)
1931 # if no input, just idle back with a prompt
1933 user.send("", just_prompt=True)
1936 def daemonize(universe):
1937 """Fork and disassociate from everything."""
1939 # only if this is what we're configured to do
1940 if "mudpy.process" in universe.contents and universe.contents[
1941 "mudpy.process"].get("daemon"):
1943 # log before we start forking around, so the terminal gets the message
1944 log("Disassociating from the controlling terminal.")
1946 # fork off and die, so we free up the controlling terminal
1950 # switch to a new process group
1953 # fork some more, this time to free us from the old process group
1957 # reset the working directory so we don't needlessly tie up mounts
1960 # clear the file creation mask so we can bend it to our will later
1963 # redirect stdin/stdout/stderr and close off their former descriptors
1964 for stdpipe in range(3):
1966 sys.stdin = codecs.open("/dev/null", "r", "utf-8")
1967 sys.__stdin__ = codecs.open("/dev/null", "r", "utf-8")
1968 sys.stdout = codecs.open("/dev/null", "w", "utf-8")
1969 sys.stderr = codecs.open("/dev/null", "w", "utf-8")
1970 sys.__stdout__ = codecs.open("/dev/null", "w", "utf-8")
1971 sys.__stderr__ = codecs.open("/dev/null", "w", "utf-8")
1974 def create_pidfile(universe):
1975 """Write a file containing the current process ID."""
1976 pid = str(os.getpid())
1977 log("Process ID: " + pid)
1978 if "mudpy.process" in universe.contents:
1979 file_name = universe.contents["mudpy.process"].get("pidfile", "")
1983 if not os.path.isabs(file_name):
1984 file_name = os.path.join(universe.startdir, file_name)
1985 os.makedirs(os.path.dirname(file_name), exist_ok=True)
1986 file_descriptor = codecs.open(file_name, "w", "utf-8")
1987 file_descriptor.write(pid + "\n")
1988 file_descriptor.flush()
1989 file_descriptor.close()
1992 def remove_pidfile(universe):
1993 """Remove the file containing the current process ID."""
1994 if "mudpy.process" in universe.contents:
1995 file_name = universe.contents["mudpy.process"].get("pidfile", "")
1999 if not os.path.isabs(file_name):
2000 file_name = os.path.join(universe.startdir, file_name)
2001 if os.access(file_name, os.W_OK):
2002 os.remove(file_name)
2005 def excepthook(excepttype, value, tracebackdata):
2006 """Handle uncaught exceptions."""
2008 # assemble the list of errors into a single string
2010 traceback.format_exception(excepttype, value, tracebackdata)
2013 # try to log it, if possible
2016 except Exception as e:
2017 # try to write it to stderr, if possible
2018 sys.stderr.write(message + "\nException while logging...\n%s" % e)
2021 def sighook(what, where):
2022 """Handle external signals."""
2025 message = "Caught signal: "
2027 # for a hangup signal
2028 if what == signal.SIGHUP:
2029 message += "hangup (reloading)"
2030 universe.reload_flag = True
2032 # for a terminate signal
2033 elif what == signal.SIGTERM:
2034 message += "terminate (halting)"
2035 universe.terminate_flag = True
2037 # catchall for unexpected signals
2039 message += str(what) + " (unhandled)"
2045 def override_excepthook():
2046 """Redefine sys.excepthook with our own."""
2047 sys.excepthook = excepthook
2050 def assign_sighook():
2051 """Assign a customized handler for some signals."""
2052 signal.signal(signal.SIGHUP, sighook)
2053 signal.signal(signal.SIGTERM, sighook)
2057 """This contains functions to be performed when starting the engine."""
2059 # see if a configuration file was specified
2060 if len(sys.argv) > 1:
2061 conffile = sys.argv[1]
2067 universe = Universe(conffile, True)
2069 # report any loglines which accumulated during setup
2070 for logline in universe.setup_loglines:
2072 universe.setup_loglines = []
2074 # fork and disassociate
2077 # override the default exception handler so we get logging first thing
2078 override_excepthook()
2080 # set up custom signal handlers
2084 create_pidfile(universe)
2086 # load and store diagnostic info
2087 universe.versions = mudpy.version.Versions("mudpy")
2089 # log startup diagnostic messages
2090 log("On %s at %s" % (universe.versions.python_version, sys.executable), 1)
2091 log("Import path: %s" % ", ".join(sys.path), 1)
2092 log("Installed dependencies: %s" % universe.versions.dependencies_text, 1)
2093 log("Other python packages: %s" % universe.versions.environment_text, 1)
2094 log("Running version: %s" % universe.versions.version, 1)
2095 log("Initial directory: %s" % universe.startdir, 1)
2096 log("Command line: %s" % " ".join(sys.argv), 1)
2098 # pass the initialized universe back
2103 """These are functions performed when shutting down the engine."""
2105 # the loop has terminated, so save persistent data
2108 # log a final message
2109 log("Shutting down now.")
2111 # get rid of the pidfile
2112 remove_pidfile(universe)