#!/usr/bin/env python3
# vim: set expandtab tabstop=4 shiftwidth=4:

# Copyright (c) 2020-2022 CJ Kucera (cj@apocalyptech.com)
# 
# This software is provided 'as-is', without any express or implied warranty.
# In no event will the authors be held liable for any damages arising from
# the use of this software.
# 
# Permission is granted to anyone to use this software for any purpose,
# including commercial applications, and to alter it and redistribute it
# freely, subject to the following restrictions:
# 
# 1. The origin of this software must not be misrepresented; you must not
#    claim that you wrote the original software. If you use this software in a
#    product, an acknowledgment in the product documentation would be
#    appreciated but is not required.
# 
# 2. Altered source versions must be plainly marked as such, and must not be
#    misrepresented as being the original software.
# 
# 3. This notice may not be removed or altered from any source distribution.

# The encryption/decryption stanzas in TTWLSave.__init__ and TTWLSave.save_to
# were helpfully provided by Gibbed (rick 'at' gibbed 'dot' us), so many
# thanks for that!  https://twitter.com/gibbed/status/1246863435868049410?s=19
#
# The rest of the savegame format was gleaned from 13xforever/Ilya's
# "gvas-converter" project: https://github.com/13xforever/gvas-converter

import uuid
import struct
import random
import google.protobuf
import google.protobuf.json_format
from . import *
from . import datalib
from . import OakSave_pb2, OakShared_pb2
from .ttwlbase import TTWLBase
from .ttwlprofile import TTWLProfile

MissionState = OakSave_pb2.MissionStatusPlayerSaveGameData.MissionState

class WLEquipSlot(object):
    """
    Real simple wrapper for a WL equipment slot.

    We touch this in a couple of different ways, so it felt like maybe we should
    wrap it up a bit.

    All these getters/setters are rather un-Pythonic; should be using
    some decorations for that instead.  Alas!
    """

    def __init__(self, protobuf):
        self.protobuf = protobuf
        self.slot = InvSlot.has_value(self.protobuf.slot_data_path)
        self.label = InvSlot.get_label(self.protobuf.slot_data_path)

    @staticmethod
    def create(index, obj_name, enabled=True):
        # TODO: test this
        return WLEquipSlot(OakSave_pb2.EquippedInventorySaveGameData(
            inventory_list_index=index,
            enabled=enabled,
            slot_data_path=obj_name,
            ))

    def get_inventory_idx(self):
        """
        Gets the inventory index that we're pointing to
        """
        return self.protobuf.inventory_list_index

    def set_inventory_idx(self, new_idx):
        """
        Sets the inventory index that we're pointing to
        """
        self.protobuf.inventory_list_index = new_idx

    def enabled(self):
        """
        Returns whether we're enabled or not
        """
        return self.protobuf.enabled

    def set_enabled(self, enabled=True):
        """
        Sets our enabled state
        """
        self.protobuf.enabled = enabled

    def get_obj_name(self):
        """
        Returns our path object name
        """
        return self.protobuf.slot_data_path

class TTWLSave(TTWLBase):
    """
    Real simple wrapper for a WL savegame file.
    
    Only tested on PC versions.  Thanks to Gibbed for the encryption method and
    the Protobuf definitions!

    https://twitter.com/gibbed/status/1246863435868049410?s=19

    All these getters/setters are rather un-Pythonic; should be using
    some decorations for that instead.  Alas!
    """

    _prefix_magic = bytearray([
        0x71, 0x34, 0x36, 0xB3, 0x56, 0x63, 0x25, 0x5F,
        0xEA, 0xE2, 0x83, 0x73, 0xF4, 0x98, 0xB8, 0x18,
        0x2E, 0xE5, 0x42, 0x2E, 0x50, 0xA2, 0x0F, 0x49,
        0x87, 0x24, 0xE6, 0x65, 0x9A, 0xF0, 0x7C, 0xD7,
        ])

    _xor_magic = bytearray([
        0x7C, 0x07, 0x69, 0x83, 0x31, 0x7E, 0x0C, 0x82,
        0x5F, 0x2E, 0x36, 0x7F, 0x76, 0xB4, 0xA2, 0x71,
        0x38, 0x2B, 0x6E, 0x87, 0x39, 0x05, 0x02, 0xC6,
        0xCD, 0xD8, 0xB1, 0xCC, 0xA1, 0x33, 0xF9, 0xB6,
        ])

    def __init__(self, filename, debug=False):
        super().__init__()
        self.filename = filename
        self.datawrapper = datalib.DataWrapper()
        with open(filename, 'rb') as df:

            header = df.read(4)
            assert(header == b'GVAS')

            self.sg_version = self._read_int(df)
            if debug:
                print('Savegame version: {}'.format(self.sg_version))
            self.pkg_version = self._read_int(df)
            if debug:
                print('Package version: {}'.format(self.pkg_version))
            self.engine_major = self._read_short(df)
            self.engine_minor = self._read_short(df)
            self.engine_patch = self._read_short(df)
            self.engine_build = self._read_int(df)
            if debug:
                print('Engine version: {}.{}.{}.{}'.format(
                    self.engine_major,
                    self.engine_minor,
                    self.engine_patch,
                    self.engine_build,
                    ))
            self.build_id = self._read_str(df)
            if debug:
                print('Build ID: {}'.format(self.build_id))
            self.fmt_version = self._read_int(df)
            if debug:
                print('Custom Format Version: {}'.format(self.fmt_version))
            fmt_count = self._read_int(df)
            if debug:
                print('Custom Format Data Count: {}'.format(fmt_count))
            self.custom_format_data = []
            for _ in range(fmt_count):
                guid = self._read_guid(df)
                entry = self._read_int(df)
                if debug:
                    print(' - GUID {}: {}'.format(guid, entry))
                self.custom_format_data.append((guid, entry))
            self.sg_type = self._read_str(df)
            if debug:
                print('Savegame type: {}'.format(self.sg_type))

            # Read in the actual data
            remaining_data_len = self._read_int(df)
            data = bytearray(df.read(remaining_data_len))

            # Decrypt
            for i in range(len(data)-1, -1, -1):
                if i < 32:
                    b = TTWLSave._prefix_magic[i]
                else:
                    b = data[i - 32]
                b ^= TTWLSave._xor_magic[i % 32]
                data[i] ^= b

            # Make sure that was all there was
            last = df.read()
            assert(len(last) == 0)

            # Parse protobufs
            self.import_protobuf(data)

    def import_protobuf(self, data):
        """
        Given raw protobuf data, load it into ourselves so
        that we can work with it.  This also sets up a few
        convenience vars for our later use
        """

        # Now parse the protobufs
        self.save = OakSave_pb2.Character()
        self.base_obj = self.save
        try:
            self.save.ParseFromString(data)
        except google.protobuf.message.DecodeError as e:
            raise Exception('Unable to parse savegame (did you pass a profile, instead?): {}'.format(e)) from None

        # Some sanity checks, since this is a potentially problematic
        # operation.
        assert(self.save.IsInitialized())
        assert(len(self.save.UnknownFields()) == 0)

        # Not checking on byte size, in case anyone had v2 protobufs
        # exported and were trying to import them now that we've
        # otherwise switched to v3.
        #assert(len(data) == self.save.ByteSize())

        # Do some data processing so that we can wrap things APIwise
        # First: Items
        self.items = [datalib.WLItem(i, self.datawrapper) for i in self.save.inventory_items]

        # Next: Equip slots
        self.equipslots = {}
        for e in self.save.equipped_inventory_list:
            equip = WLEquipSlot(e)
            self.equipslots[equip.slot] = equip

    def import_json(self, json_str):
        """
        Given JSON data, convert to protobuf and load it into ourselves so
        that we can work with it.  This also sets up a few convenience vars
        for our later use
        """
        message = google.protobuf.json_format.Parse(json_str, OakSave_pb2.Character())
        self.import_protobuf(message.SerializeToString())

    def save_to(self, filename):
        """
        Saves ourselves to a new filename
        """
        with open(filename, 'wb') as df:

            # Header info
            df.write(b'GVAS')
            self._write_int(df, self.sg_version)
            self._write_int(df, self.pkg_version)
            self._write_short(df, self.engine_major)
            self._write_short(df, self.engine_minor)
            self._write_short(df, self.engine_patch)
            self._write_int(df, self.engine_build)
            self._write_str(df, self.build_id)
            self._write_int(df, self.fmt_version)
            self._write_int(df, len(self.custom_format_data))
            for guid, entry in self.custom_format_data:
                self._write_guid(df, guid)
                self._write_int(df, entry)
            self._write_str(df, self.sg_type)

            # Turn our parsed protobuf back into data
            data = bytearray(self.save.SerializeToString())

            # Encrypt
            for i in range(len(data)):
                if i < 32:
                    b = self._prefix_magic[i]
                else:
                    b = data[i - 32]
                b ^= self._xor_magic[i % 32]
                data[i] ^= b

            # Write out to the file
            self._write_int(df, len(data))
            df.write(data)

    def save_protobuf_to(self, filename):
        """
        Saves the raw protobufs to the specified filename
        """
        with open(filename, 'wb') as df:
            df.write(self.save.SerializeToString())

    def save_json_to(self, filename):
        """
        Saves a JSON version of our protobuf to the specfied filename
        """
        with open(filename, 'w') as df:
            df.write(google.protobuf.json_format.MessageToJson(self.save,
                including_default_value_fields=True,
                preserving_proto_field_name=True,
                ))

    def _read_int(self, df):
        return struct.unpack('<I', df.read(4))[0]

    def _write_int(self, df, value):
        df.write(struct.pack('<I', value))

    def _read_short(self, df):
        return struct.unpack('<H', df.read(2))[0]

    def _write_short(self, df, value):
        df.write(struct.pack('<H', value))

    def _read_str(self, df):
        datalen = self._read_int(df)
        if datalen == 0:
            return None
        elif datalen == 1:
            return ''
        else:
            value = df.read(datalen)
            return value[:-1].decode('utf-8')

    def _write_str(self, df, value):
        if value is None:
            self._write_int(df, 0)
        elif value == '':
            self._write_int(df, 1)
        else:
            data = value.encode('utf-8') + b'\0'
            self._write_int(df, len(data))
            df.write(data)

    def _read_guid(self, df):
        data = df.read(16)
        return data
        # A bit silly to bother formatting it, since we don't care.
        #arr = ''.join(['{:02x}'.format(d) for d in data])
        #return '{}-{}-{}-{}-{}'.format(
        #        arr[0:8],
        #        arr[8:12],
        #        arr[12:16],
        #        arr[16:20],
        #        arr[20:32],
        #        )

    def _write_guid(self, df, value):
        df.write(value)

    def get_char_name(self):
        """
        Returns the character name
        """
        return self.save.preferred_character_name

    def set_char_name(self, new_name):
        """
        Sets the character name
        """
        self.save.preferred_character_name = new_name

    def get_savegame_id(self):
        """
        Returns the savegame ID (not sure if this is important at all)
        """
        return self.save.save_game_id

    def set_savegame_id(self, new_id):
        """
        Sets the savegame ID (not sure if this is important at all)
        """
        self.save.save_game_id = new_id

    def get_companion_names(self, eng=False):
        """
        Returns a dict mapping companion types to companion names, if any
        are defined.  The companion type key is a constant by default, or
        an English label if `eng` is `True`
        """
        ret = {}
        for name in self.save.nickname_mappings:
            companion = Companion.has_value(name.key.lower())
            if companion:
                key = companion
            else:
                key = name.key.lower()
            ret[key] = name.value
        return ret

    def get_companion_name(self, companion_type):
        """
        Returns the companion name matching the given type constant
        """
        companion_names = self.get_companion_names()
        if companion_type in companion_names:
            return companion_names[companion_type]
        return None

    def _get_class(self, class_path, eng=False):
        """
        Gets a character class given the specified `class_path`.  By
        default the return value will be a CharClass instance, but if
        `eng` is `True`, the value will be the English name of the class
        instead.  If the class path isn't known to us, it will be
        returned back.
        """
        classval = CharClass.has_value(class_path)
        if classval:
            if eng:
                return classval.label
            else:
                return classval
        else:
            return class_path

    def get_primary_class(self, eng=False):
        """
        Returns the primary class of this character.  By default the return
        value will be a CharClass instance, but if `eng` is `True`, the value
        will be the English name of the class instead.  If the class path isn't
        known to us, it will be returned back.
        """
        return self._get_class(self.save.ability_data.dual_class_save_data.primary_branch_path, eng=eng)

    def get_secondary_class(self, eng=False):
        """
        Returns the secondary class of this character.  By default the return
        value will be a CharClass instance, but if `eng` is `True`, the value
        will be the English name of the class instead.  If the class path isn't
        known to us, it will be returned back.
        """
        return self._get_class(self.save.ability_data.dual_class_save_data.slotted_secondary_branch_path, eng=eng)

    def get_xp(self):
        """
        Returns the character's XP
        """
        return self.save.experience_points

    def get_level(self):
        """
        Returns the character's level
        """
        xp = self.get_xp()
        cur_lvl = 0
        for req_xp_lvl in required_xp_list:
            if xp >= req_xp_lvl:
                cur_lvl += 1
            else:
                return cur_lvl
        return cur_lvl

    def set_level(self, level, top_val=False):
        """
        Sets the character's level to the given `level`.  By default will
        assign the least amount of XP possible to gain the specified level,
        but if `top_val` is `True`, will assign the maximum possible XP in
        that level, instead (so 1XP more will cause the character to level
        up).  Will return the number of skillpoints added to the char as
        as result of this (will not *remove* any skill points when going
        backwards).
        """
        if level > len(required_xp_list):
            raise Exception('Unknown level {}'.format(level))

        # Rather than checking against `max_level`, we're going to set the
        # max to 80 (which is the highest XP value I'm 100% sure of).  That
        # way folks would be able to pre-level a char, even if this editor's
        # not been updated after a level cap increase.
        if level > max_supported_level:
            raise Exception('Maximum level is {}'.format(max_supported_level))
        if level < 1:
            raise Exception('Level must be at least 1')

        # WL doesn't auto-calculate skill points like all previous games
        # did, so if we're levelling *up*, we should add in skill points.
        # Do so now, before we've updated the char level, so we know what
        # the difference is.
        points_to_add = 0
        cur_level = self.get_level()
        if level > cur_level:
            for step in range(cur_level+1, level+1):
                points_to_add += skill_point_exceptions.get(step, 1)
            self.add_skill_points(points_to_add)
        
        # If we've been told to assign the max level, we can't do top_val
        if level == max_level:
            top_val = False

        # Now assign
        if top_val:
            self.save.experience_points = required_xp_list[level]-1
        else:
            self.save.experience_points = required_xp_list[level-1]

        # Make sure our stats level is set properly
        self.set_stats_obj(level_stat, level)

        # Also unlock the appropriate challenges for the char's level, though
        # it probably doesn't actually matter at all.  Whatever.
        for challenge_level, challenge_obj in level_challenges:
            if level >= challenge_level:
                self.unlock_challenge_obj(challenge_obj)

        # Return skill points added
        return points_to_add

    def add_skill_points(self, points_to_add):
        """
        Adds the specified `points_to_add` number of skill points to the character.
        """
        self.save.ability_data.ability_points += points_to_add

    def get_playthroughs_completed(self):
        """
        Returns the number of playthroughs completed
        """
        return self.save.playthroughs_completed

    def set_playthroughs_completed(self, playthrough_count):
        """
        Sets the number of playthroughs completed
        """
        self.save.playthroughs_completed = playthrough_count

    def get_max_playthrough_with_data(self):
        """
        Returns the maximum playthrough (zero-indexed) for which we have actual
        data in the savegame.  Even if TVHM is unlocked, for instance, we may
        only have NVHM data in the savefile.
        """
        # Really I don't think that any of these numbers will ever be
        # different, but what the hell, we'll check 'em all anyway.
        return min(
                len(self.save.mission_playthroughs_data),
                len(self.save.active_travel_stations_for_playthrough),
                len(self.save.last_active_travel_station_for_playthrough),
                len(self.save.game_state_save_data_for_playthrough),
                )-1

    def get_chaos_level(self):
        """
        Returns the current Chaos Level
        """
        if len(self.save.game_state_save_data_for_playthrough) > 0:
            return self.save.game_state_save_data_for_playthrough[0].mayhem_level
        else:
            return 0

    def get_chaos_level_with_max(self):
        """
        Returns a tuple with the current Chaos Level, plus the max unlocked level.
        """
        if len(self.save.game_state_save_data_for_playthrough) > 0:
            return (self.save.game_state_save_data_for_playthrough[0].mayhem_level,
                    self.save.game_state_save_data_for_playthrough[0].mayhem_unlocked_level)
        else:
            return (0, 0)

    def set_chaos_level(self, level, unlock_only=False):
        """
        Sets the current Chaos Level (will also set at least that level to be unlocked).
        If `unlock_only` is `True`, this will *only* ensure that the specified level
        is unlocked, but not update the currently-active Chaos Level.
        """
        if len(self.save.game_state_save_data_for_playthrough) == 0:
            self.save.game_state_save_data_for_playthrough.append(OakSave_pb2.GameStateSaveData(
                last_traveled_map_id=OakSave_pb2.MapIDData(
                    zone_name_id=0,
                    map_name_id=0,
                    ),
                mayhem_level=0,
                mayhem_random_seed=0,
                mayhem_unlocked_level=0,
                ))
        if not unlock_only:
            self.save.game_state_save_data_for_playthrough[0].mayhem_level = level
        if level > self.save.game_state_save_data_for_playthrough[0].mayhem_unlocked_level:
            self.save.game_state_save_data_for_playthrough[0].mayhem_unlocked_level = level
        # If we set this value, we can use Chaos Mode right from the beginning of the game
        self.set_playthroughs_completed(1)

    def clear_game_state_pt(self, playthrough=0):
        """
        Clears out all game state data from the given `playthrough` (zero-indexed).  The
        `playthrough` given must be the *last* one in the list.
        """
        if playthrough < 0:
            raise Exception('playthrough cannot be negative')
        if (len(self.save.game_state_save_data_for_playthrough)-1) != playthrough:
            raise Exception('playthrough must be {} for this save'.format(
                len(self.save.game_state_save_data_for_playthrough)-1,
                ))
        self.save.game_state_save_data_for_playthrough.pop()

    def get_pt_last_stations(self):
        """
        Returns a list of the object names of the last station (fast travel,
        resurrection, level) the player has been near, for each Playthrough
        """
        return self.save.last_active_travel_station_for_playthrough

    def get_pt_last_station(self, pt=0):
        """
        Returns the last station (fast travel, resurrection, level) the player
        has been near, for the given Playthrough (zero-indexed)
        """
        if len(self.save.last_active_travel_station_for_playthrough) > pt:
            return self.save.last_active_travel_station_for_playthrough[pt]
        return None

    def clear_last_station_pt(self, playthrough=0):
        """
        Clears out all Last Station Visited data from the given `playthrough` (zero-indexed).  The
        `playthrough` given must be the *last* one in the list.
        """
        if playthrough < 0:
            raise Exception('playthrough cannot be negative')
        if (len(self.save.last_active_travel_station_for_playthrough)-1) != playthrough:
            raise Exception('playthrough must be {} for this save'.format(
                len(self.save.last_active_travel_station_for_playthrough)-1,
                ))
        self.save.last_active_travel_station_for_playthrough.pop()

    def get_pt_last_maps(self, eng=False):
        """
        Returns a list maps which the player is currently in, for each Playthrough.
        Maps will be their in-game IDs by default, or English names if `eng`
        is `True`
        """
        # TODO: should maybe handle these edge cases better?
        maps = []
        for station in self.get_pt_last_stations():
            if station is None:
                maps.append('(NO MAP)')
            elif station == '':
                maps.append('(BLANK MAP)')
            else:
                lower = station.lower()
                if lower in fts_to_map:
                    mapname = fts_to_map[lower]
                    if eng:
                        if mapname in map_to_eng:
                            mapname = map_to_eng[mapname]
                        else:
                            mapname = '(Unknown map: {})'.format(mapname)
                    maps.append(mapname)
                else:
                    maps.append('(Unknown station: {})'.format(station))
        return maps

    def get_last_maps(self, eng=False):
        """
        Returns the map which the player has last been in.  Maps will be
        their in-game IDs by default, or English names if `eng` is `True`.
        At the moment WL only has a single playthrough, but we're wrapping it
        like this so it's easier to revert, if GBX ever does add
        TVHM-or-whatever.
        """
        maplist = self.get_pt_last_maps(eng=eng)
        if len(maplist) > 0:
            return maplist[0]
        else:
            return '(NO MAP)'

    def get_pt_last_map(self, pt=0, eng=False):
        """
        Returns the last map the player has been in, for the specified
        Playthrough (zero-indexed).  The map will be its in-game ID by
        default, or the English name if `eng` is `True`
        """
        map_ids = self.get_pt_last_maps(eng=eng)
        if len(map_ids) > pt:
            return map_ids[pt]
        return None

    def get_pt_active_ft_station_lists(self):
        """
        Returns a list of Fast travel station names active for each playthrough
        """
        to_ret = []
        for data in self.save.active_travel_stations_for_playthrough:
            to_ret.append([d.active_travel_station_name for d in data.active_travel_stations])
        return to_ret

    def get_active_ft_station_lists(self):
        """
        Returns a list of Fast travel station names active.  At the moment WL only has
        a single playthrough, but we're wrapping it like this so it's easier to revert,
        if GBX ever does add TVHM-or-whatever.
        """
        active_stations = self.get_pt_active_ft_station_lists()
        if active_stations:
            return active_stations[0]
        return []

    def get_pt_active_ft_station_list(self, pt=0):
        """
        Returns a list of Fast Travel station names active for the given
        Playthrough (zero-indexed)
        """
        ptlist = self.get_pt_active_ft_station_lists()
        if len(ptlist) > pt:
            return ptlist[pt]
        return None

    def clear_active_ft_stations_pt(self, playthrough=0):
        """
        Clears out all active Fast Travel data from the given `playthrough` (zero-indexed).  The
        `playthrough` given must be the *last* one in the list.
        """
        if playthrough < 0:
            raise Exception('playthrough cannot be negative')
        if (len(self.save.active_travel_stations_for_playthrough)-1) != playthrough:
            raise Exception('playthrough must be {} for this save'.format(
                len(self.save.active_travel_stations_for_playthrough)-1,
                ))
        self.save.active_travel_stations_for_playthrough.pop()

    def get_pt_mission_lists(self, mission_status, eng=False):
        """
        Returns a list of missions in the given `mission_status`, for each
        Playthrough.  Missions will be in their object name by default, or
        their English names if `eng` is `True`.
        """
        to_ret = []
        for pt in self.save.mission_playthroughs_data:
            active_missions = []
            for mission in pt.mission_list:
                if mission.status == mission_status:
                    mission_name = mission.mission_class_path
                    if eng:
                        if mission_name.lower() in mission_to_name:
                            mission_name = mission_to_name[mission_name.lower()]
                        else:
                            mission_name = '(Unknown mission: {})'.format(mission_name)
                    active_missions.append(mission_name)
            to_ret.append(active_missions)
        return to_ret

    def get_mission_lists(self, mission_status, eng=False):
        """
        Returns a list of missions in the given `mission_status`.  Missions will
        be in their object name by default, or their English names if `eng` is `True`.
        At the moment WL only has a single playthrough, but we're wrapping it
        like this so it's easier to revert, if GBX ever does add
        TVHM-or-whatever.

        """
        return self.get_pt_mission_lists(mission_status, eng=eng)[0]

    def get_pt_active_mission_lists(self, eng=False):
        """
        Returns a list of active missions for each Playthrough.  Missions will
        be in their object name by default, or their English names if `eng` is
        `True`.
        """
        return self.get_pt_mission_lists(MissionState.MS_Active, eng)

    def get_active_mission_lists(self, eng=False):
        """
        Returns a list of active missions.  Missions will be in their object name
        by default, or their English names if `eng` is `True`.
        At the moment WL only has a single playthrough, but we're wrapping it
        like this so it's easier to revert, if GBX ever does add
        TVHM-or-whatever.
        """
        mission_lists = self.get_pt_active_mission_lists(eng=eng)
        if len(mission_lists) > 0:
            return mission_lists[0]
        else:
            return []

    def get_pt_completed_mission_lists(self, eng=False):
        """
        Returns a list of completed missions for each Playthrough.  Missions will
        be in their object name by default, or their English names if `eng` is
        `True`.
        """
        return self.get_pt_mission_lists(MissionState.MS_Complete, eng)

    def get_completed_mission_lists(self, eng=False):
        """
        Returns a list of completed missions.  Missions will be in their object name
        by default, or their English names if `eng` is `True`.
        At the moment WL only has a single playthrough, but we're wrapping it
        like this so it's easier to revert, if GBX ever does add
        TVHM-or-whatever.
        """
        mission_lists = self.get_pt_completed_mission_lists(eng=eng)
        if len(mission_lists) > 0:
            return mission_lists[0]
        else:
            return []

    def get_pt_mission_list(self, mission_status, pt=0, eng=False):
        """
        Returns a list of missions in the given `mission_status`, for the given
        Playthrough (zero-indexed).  Missions will be in their object name
        by default, or their English names if `eng` is `True`
        """
        missions = self.get_pt_mission_lists(mission_status, eng=eng)
        if len(missions) > pt:
            return missions[pt]
        return None

    def get_pt_active_mission_list(self, pt=0, eng=False):
        """
        Returns a list of active missions for the given Playthrough (zero-indexed).
        Missions will be in their object name by default, or their English names
        if `eng` is `True`
        """
        return self.get_pt_mission_list(MissionState.MS_Active, pt, eng)

    def get_pt_completed_mission_list(self, pt=0, eng=False):
        """
        Returns a list of completed missions for the given Playthrough (zero-indexed).
        Missions will be in their object name by default, or their English names
        if `eng` is `True`
        """
        return self.get_pt_mission_list(MissionState.MS_Complete, pt, eng)

    def get_pt_completed_mission_counts(self):
        """
        Returns a count of completed missions for each Playthrough.
        """
        to_ret = []
        for missions in self.get_pt_completed_mission_lists():
            to_ret.append(len(missions))
        return to_ret

    def get_completed_mission_counts(self):
        """
        Returns a count of completed missions.  At the moment WL only has a
        single playthrough, but we're wrapping it like this so it's easier to
        revert, if GBX ever does add TVHM-or-whatever.
        """
        return self.get_pt_completed_mission_counts()[0]

    def get_pt_completed_mission_count(self, pt=0):
        """
        Returns a count of completed mission object names for the given
        Playthrough (zero-indexed).
        """
        counts = self.get_pt_completed_mission_counts()
        if len(counts) > pt:
            return counts[pt]
        return None

    def clear_mission_pt(self, playthrough=0):
        """
        Clears out all mission data from the given `playthrough` (zero-indexed).  The
        `playthrough` given must be the *last* one in the list.
        """
        if playthrough < 0:
            raise Exception('playthrough cannot be negative')
        if (len(self.save.mission_playthroughs_data)-1) != playthrough:
            raise Exception('playthrough must be {} for this save'.format(
                len(self.save.mission_playthroughs_data)-1,
                ))
        self.save.mission_playthroughs_data.pop()

    def clear_playthrough_data(self, playthrough=0):
        """
        Clears out all data from the given `playthrough` (zero-indexed), including any higher
        playthroughs.  This is completely untested for clearing NVHM data, but is used by
        the cli_edit to clear out THVM.  This supports clearing higher levels just because my
        original scripts to generate my WL Save Archive accidentally unlocked an extra
        playthrough higher than it intended to, so this will allow me to "clear TVHM" while
        also getting rid of that higher playthrough data.

        It's permissible to pass a `playthrough` higher than the max that we have data for,
        since THVM can be technically unlocked without actually having THVM data in the save
        file.
        """
        max_pt = self.get_max_playthrough_with_data()
        if playthrough <= max_pt:
            for pt in range(max_pt, playthrough-1, -1):
                self.clear_mission_pt(pt)
                self.clear_active_ft_stations_pt(pt)
                self.clear_last_station_pt(pt)
                self.clear_game_state_pt(pt)

    def get_items(self):
        """
        Returns a list of the character's inventory items, as WLItem objects.
        """
        return self.items

    def get_equipped_items(self, eng=False):
        """
        Returns a dict containing the slot and the equipped item.  The slot will
        be a constant by default, or an English label if `eng` is `True`
        """
        to_ret = {}
        for (key, equipslot) in self.equipslots.items():
            if eng:
                key = equipslot.label
            else:
                key = key.value
            if equipslot.get_inventory_idx() >= 0:
                to_ret[key] = self.items[equipslot.get_inventory_idx()]
            else:
                to_ret[key] = None
        return to_ret

    def get_equipped_item_slot(self, slot):
        """
        Given a slot, return the item equipped in that slot
        """
        if slot in self.equipslots:
            inv_idx = self.equipslots[slot].get_inventory_idx()
            if inv_idx >= 0 and len(self.items) > inv_idx:
                return self.items[inv_idx]
        return None

    def get_equip_slots(self, eng=False):
        """
        Returns a dict of slot ID and WLEquipSlot objects, for all inventory
        slots.  The slot will be a constant by default, or an English label if
        `eng` is `True`
        """
        return self.equipslots

    def get_equip_slot(self, slot):
        """
        Returns the WLEquipSlot object in the specified `slot`.
        """
        if slot in self.equipslots:
            return self.equipslots[slot]
        return None

    def unlock_slots(self, slots=None):
        """
        Unlocks the specified inventory `slots`, which should be a list of InvSlot
        members.  If `slots` is not passed in, will unlock all inventory slots.
        """
        if slots is None:
            slots = InvSlot
        for slot in slots:
            self.equipslots[slot].set_enabled()

    def wipe_inventory(self):
        """
        Removes all items from inventory
        """

        del self.save.inventory_items[:]
        self.items = []
        for slot in self.equipslots.values():
            slot.protobuf.inventory_list_index = -1

    def add_item(self, new_item):
        """
        Adds a new `new_item` (WLItem object) to our item list.  Returns the item's
        new index in our item list.
        """

        # Add the item to the protobuf
        self.save.inventory_items.append(new_item.protobuf)

        # The protobuf reference that we append to the protobuf list
        # ends up *not* being the one that's actually used when we
        # save, so if we want to be able to alter it later (say, below
        # when levelling up items), we have to grab a fresh reference
        # to it.
        new_item.protobuf = self.save.inventory_items[-1]

        # Now update our internal items list and return
        self.items.append(new_item)
        return len(self.items)-1

    def create_new_item(self, item_serial):
        """
        Creates a new item from the given binary `item_serial`, which can later
        be added to our item list.
        """

        # Okay, I have no idea what this pickup_order_index attribute is about, but let's
        # make sure it's unique anyway.  It might be related to ordering when picking
        # up multiple items at once, which would probably make it more useful for auto-pick-up
        # items like money and ammo...
        max_pickup_order = 0
        for item in self.items:
            if item.get_pickup_order_idx() > max_pickup_order:
                max_pickup_order = item.get_pickup_order_idx()

        # Create the item and return it
        new_item = datalib.WLItem.create(self.datawrapper,
                serial_number=item_serial,
                pickup_order_idx=max_pickup_order+1,
                is_favorite=True,
                )
        return new_item

    def create_new_item_encoded(self, item_serial_b64):
        """
        Creates a new item from the base64-encoded (and "WL()"-wrapped)
        `item_serial_b64`, which can later be added to our item list.
        """
        return self.create_new_item(datalib.WLSerial.decode_serial_base64(item_serial_b64))

    def add_new_item(self, item_serial):
        """
        Adds a new item to our item list using the binary `item_serial`.
        Returns a tuple containing the new WLItem object itself, and its
        new index in our item list.
        """
        new_item = self.create_new_item(item_serial)
        return (new_item, self.add_item(new_item))

    def add_new_item_encoded(self, item_serial_b64):
        """
        Adds a new item to our item list using the base64-encoded (and
        "WL()"-wrapped) `item_serial_b64`.  Returns a tuple containing the
        new WLItem object itself, and its new index in our item list.
        """
        return self.add_new_item(datalib.WLSerial.decode_serial_base64(item_serial_b64))

    def overwrite_item_in_slot(self, slot, itemdata):
        """
        Given a binary `itemdata`, overwrite whatever item is in the given `slot`.  Will create
        a new item object if the slot is empty.
        """
        item = self.get_equipped_item_slot(slot)
        if item:
            item.set_serial(itemdata)
        else:
            # Now create a new item
            (new_item, new_index) = self.add_new_item(itemdata)

            # Now assign it to the slot
            found_slot = False
            if slot in self.equipslots:
                found_slot = True
                self.equipslots[slot].set_inventory_idx(new_index)

            # If we didn't find a slot, create it (I don't *think* this should ever happen?)
            # TODO: test this
            if not found_slot:
                equipslot = WLEquipSlot.create(new_index, slot_obj)
                self.save.equipped_inventory_list.append(equipslot.protobuf)
                self.equipslots[slot] = equipslot

    def overwrite_item_in_slot_encoded(self, slot, item_serial_b64):
        """
        Given a base64 (and "WL()"-wrapped) `item_serial_b64`, overwrite
        whatever item is in the given `slot`.  Will create a new item object
        if the slot is empty.
        """
        self.overwrite_item_in_slot(slot, datalib.WLSerial.decode_serial_base64(item_serial_b64))

    def get_currency(self, currency_type):
        """
        Returns the amount of currency of the given type
        """
        for cat_save_data in self.save.inventory_category_list:
            if cat_save_data.base_category_definition_hash == currency_type.value:
                return cat_save_data.quantity
        return 0

    def set_currency(self, currency_type, new_value, quiet=False):
        """
        Sets a new currency value
        """

        # Warn about invalid values (but let the user do it anyway)
        if not quiet:
            if currency_type.num is not None and new_value > currency_type.num:
                print(f'WARNING: Maximum value for {currency_type.label} is {currency_type.num:,} - setting to {new_value:,} anyway')

        # Update an existing value, if we have it
        for cat_save_data in self.save.inventory_category_list:
            if cat_save_data.base_category_definition_hash == currency_type.value:
                cat_save_data.quantity = new_value
                return

        # Add a new one, if we don't
        self.save.inventory_category_list.append(OakShared_pb2.InventoryCategorySaveData(
            base_category_definition_hash=currency_type.value,
            quantity=new_value,
            ))

    def get_money(self):
        """
        Returns the amount of money we have
        """
        return self.get_currency(Currency.MONEY)

    def set_money(self, new_value):
        """
        Sets the amount of money we have
        """
        return self.set_currency(Currency.MONEY, new_value)

    def get_moon_orbs(self):
        """
        Returns the amount of moon orbs we have
        """
        return self.get_currency(Currency.MOON_ORBS)

    def set_moon_orbs(self, new_value):
        """
        Sets the amount of moon orbs we have
        """
        return self.set_currency(Currency.MOON_ORBS, new_value)

    def get_souls(self):
        """
        Returns the amount of souls we have
        """
        return self.get_currency(Currency.SOULS)

    def set_souls(self, new_value):
        """
        Sets the amount of souls we have
        """
        return self.set_currency(Currency.SOULS, new_value)

    def get_sdus(self, eng=False):
        """
        Returns a dict containing the SDU type and the number purchased.  The SDU
        type key will be a constant by default, or an English label if `eng` is `True`
        """
        to_ret = {}
        for sdu in self.save.sdu_list:
            if eng:
                key = SDU.get_label(sdu.sdu_data_path)
            else:
                key = SDU(sdu.sdu_data_path)
            to_ret[key] = sdu.sdu_level
        return to_ret

    def get_sdus_with_max(self, eng=False):
        """
        Returns a dict whose keys are the SDU type, and the values are tuples with
        two values:
            1. The number of that SDU type purchased
            2. The maximum number of SDUs available for that type
        The SDU type key will be a constant by default, or an English label if
        `eng` is `True`.  This is just a convenience function suitable for
        giving more information to users.
        """
        to_ret = {}
        for sdu_proto in self.save.sdu_list:
            sdu = SDU(sdu_proto.sdu_data_path)
            if eng:
                key = SDU.get_label(sdu_proto.sdu_data_path)
            else:
                key = sdu.value
            to_ret[key] = (sdu_proto.sdu_level, sdu.num)
        return to_ret

    def get_sdu(self, sdu):
        """
        Returns the number of SDUs purchased for the specified type
        """
        sdus = self.get_sdus()
        if sdu in sdus:
            return sdus[sdu]
        return 0

    def set_max_sdus(self, sdulist=None):
        """
        Sets the specified SDUs (or all SDUs that we know about) to be at the max level
        """
        if sdulist is None:
            all_sdus = set(SDU)
        else:
            all_sdus = set(sdulist)

        # Set all existing SDUs to max
        for sdu_proto in self.save.sdu_list:
            sdu = SDU(sdu_proto.sdu_data_path)
            if sdu in all_sdus:
                all_sdus.remove(sdu)
                sdu_proto.sdu_level = sdu.num

        # If we're missing any, add them.
        for sdu in all_sdus:
            self.save.sdu_list.append(OakShared_pb2.OakSDUSaveGameData(
                sdu_data_path=sdu.value,
                sdu_level=sdu.num,
                ))

    def get_ammo_counts(self, eng=False):
        """
        Returns a dict containing the Ammo type and count.  The ammo type key will
        be a constant by default, or an English label if `eng` is `True`.
        """
        to_ret = {}
        for pool in self.save.resource_pools:
            # `Resource_Ammo_Spell` shows up in here but isn't actually used; ignore it
            if 'Resource_Ammo_Spell' in pool.resource_path:
                continue
            ammo = Ammo.has_value(pool.resource_path)
            if ammo:
                if eng:
                    key = Ammo.get_label(pool.resource_path)
                else:
                    key = ammo.value
            else:
                key = pool.resource_path
            to_ret[key] = int(pool.amount)
        return to_ret

    def get_ammo_count(self, ammo):
        """
        Returns the ammo count for the specified ammo type
        """
        ammo_vals = self.get_ammo_counts()
        if ammo in ammo_vals:
            return ammo_vals[ammo]
        return 0

    def set_max_ammo(self):
        """
        Sets our ammo counts to be at the max level
        """

        # Set all existing ammo pools to max (shouldn't have to worry about
        # pools not being in here)
        for pool in self.save.resource_pools:
            ammo = Ammo.has_value(pool.resource_path)
            if ammo:
                pool.amount = ammo.num

    def set_stats_obj(self, stat_obj, stat_value):
        """
        Sets the given `stat_obj`, which lives in `game_stats_data`.
        `stat_value` will be the value of the statistic.
        """
        for stat in self.save.game_stats_data:
            if stat.stat_path == stat_obj:
                stat.stat_value = stat_value
                return

        # If we get here, the stat wasn't found, and we'll have to add
        # it ourselves.
        self.save.game_stats_data.append(OakShared_pb2.GameStatSaveGameData(
            stat_value=stat_value,
            stat_path=stat_obj,
            ))

    def get_savegame_guid(self):
        """
        Returns the savegame GUID
        """
        return self.save.save_game_guid

    def set_savegame_guid(self,guid):
        """
        Sets the savegame GUID
        """
        self.save.save_game_guid = guid

    def randomize_guid(self):
        """
        Randomizes our savegame GUID, in case anything takes that into account.
        """
        self.save.save_game_guid = uuid.uuid4().hex.upper()

    def finish_game(self):
        """
        Marks the Final Missions as completed, for all currently-available
        playthroughs
        """
        """
            >>> [x for x in ml if x.mission_class_path == final]
            [status: MS_Complete
            objectives_progress: 1
            objectives_progress: 1
            objectives_progress: 1
            objectives_progress: 1
            objectives_progress: 1
            objectives_progress: 1
            objectives_progress: 1
            objectives_progress: 1
            objectives_progress: 0
            objectives_progress: 1
            objectives_progress: 0
            objectives_progress: 1
            objectives_progress: 1
            objectives_progress: 1
            objectives_progress: 1
            mission_class_path: "/Game/Missions/Plot/Mission_Plot11.Mission_Plot11_C"
            active_objective_set_path: "/Game/Missions/Plot/Mission_Plot11.Set_ReturnToLobby_ObjectiveSet"
            kickoff_played: true
            ]
        """
        final_missions = {
            '/Game/Missions/Plot/Mission_Plot11.Mission_Plot11_C':(
                "/Game/Missions/Plot/Mission_Plot11.Set_ReturnToLobby_ObjectiveSet",
                [ 1, 1, 1,1,1,1,1,1,0,1,0,1,1,1,1,]
            ),
        }
        for pt in self.save.mission_playthroughs_data:
            # First, complete the missions if they're already present
            mission_seen = set()
            for mission in pt.mission_list:
                if mission.mission_class_path in final_missions:
                    # print(f"Found mission: {mission.mission_class_path}")
                    (_, objectives) = final_missions[mission.mission_class_path]
                    mission_seen.add(mission.mission_class_path)
                    mission.status = MissionState.MS_Complete
                    del mission.objectives_progress[:]
                    mission.objectives_progress.extend(objectives)
                    mission.kickoff_played = True
                    mission.has_been_viewed_in_log = True

            # Now, if we didn't find one of 'em, inject it
            for mission_path, (objectiveset, objectives) in final_missions.items():
                if mission_path not in mission_seen:
                    # print(f"Making mission: {mission.mission_class_path}")
                    pt.mission_list.append(OakSave_pb2.MissionStatusPlayerSaveGameData(
                        status=MissionState.MS_Complete,
                        objectives_progress=objectives,
                        mission_class_path=mission_path,
                        active_objective_set_path=objectiveset,
                        kickoff_played=True,
                        has_been_viewed_in_log=True,
                        dlc_package_id=0,
                        # league_instance=0,
                        ))

                    
    def finish_mission(self, mission_name, pt=0):
        # TODO: docstring, test, better handling of `pt`
        self.save.mission_playthroughs_data[pt].mission_list.append(OakSave_pb2.MissionStatusPlayerSaveGameData(
            status=MissionState.MS_Complete,
            objectives_progress=objectives,
            mission_class_path=mission_path,
            active_objective_set_path=objectiveset,
            kickoff_played=True,
            has_been_viewed_in_log=True,
            dlc_package_id=0,
            league_instance=0,
        ))

    def delete_mission(self, mission_obj, pt=0, allow_plot=False):
        """
        Deletes the specified mission (with object path `mission_obj`), in the
        playthrough `pt` (0 = Normal/NVHM, 1 = TVHM) from the savegame entirely.
        By default, this method will refuse to delete plot missions, since doing
        so will lock out the game progress, if no other edits are made.  Setting
        `allow_plot` to `True` will permit that action.  Returns `True` if the
        mission was deleted, or `False` if the specified mission wasn't found,
        or wasn't permitted to be deleted on account of `allow_plot`.
        """
        lower = mission_obj.lower()
        if not allow_plot and lower in plot_missions:
            return False
        if len(self.save.mission_playthroughs_data) <= pt:
            return False
        # Originally tried a fancier `itertools.filterfalse` assignment here, but
        # repeated protobufs can't be assigned directly, and doing a del/extend
        # dance seemed sillier than just doing it this "dumb" way.
        to_del_idx = None
        for idx, mission in enumerate(self.save.mission_playthroughs_data[pt].mission_list):
            if mission.mission_class_path.lower() == lower:
                to_del_idx = idx
                break
        if to_del_idx is None:
            return False
        else:
            del self.save.mission_playthroughs_data[pt].mission_list[idx]
            return True

    def unlock_feat(self):
        """
        Unlocks the character's feat/companion
        """
        self.save.ability_data.dual_class_save_data.unlocked_class_features = True

    def unlock_multiclass(self):
        """
        Unlocks all Multiclass capability -- both the ability to choose one
        and the ability to respec it right away.  If multiclass was not
        already unlocked on the save, this will also add +2 Skill Points.
        Returns `True` if the points were added, or `False` otherwise.
        """
        # Add skill points if we need to
        added_points = False
        if not self.save.ability_data.dual_class_save_data.unlocked_initial_secondary_class:
            self.add_skill_points(2)
            added_points = True

        # Do the unlock
        self.save.ability_data.dual_class_save_data.unlocked_initial_secondary_class = True
        self.save.ability_data.dual_class_save_data.unlocked_secondary_class_swapping = True

        # Return
        return added_points

    def get_hero_stats(self):
        """
        Returns a dict describing Hero Stats for the character.  The keys will be
        a HeroStats enum member, and the value will be the current value in the
        save.  Note that these are *not* adjusted for Backstory or Myth Rank
        """
        to_ret = {}
        for stat in HeroStats:
            to_ret[stat] = getattr(self.save.hero_points_save_data, stat.value)
        return to_ret

    def set_hero_stats(self, stats, value):
        """
        Sets the given `stats` to the specified `value`, in the Hero Stats section
        of the save.  `stats` can be a single HeroStats enum, or an iterable
        object containing one or more HeroStats enums.
        """
        if type(stats) == HeroStats:
            stats = [stats]
        for stat in stats:
            setattr(self.save.hero_points_save_data, stat.value, value)

    def get_backstory(self, eng=False):
        """
        Returns the backstory for the character.  If `eng` is False, this will be
        a Backstory enum instance if possible, or the raw object path if not found.
        If `eng` is True, an English label will be returned instead (if possible)
        """
        bs = Backstory.has_value(self.save.hero_points_save_data.player_aspect_data_path)
        if bs:
            if eng:
                return bs.label
            else:
                return bs
        else:
            return self.save.hero_points_save_data.player_aspect_data_path

    def set_backstory(self, backstory):
        """
        Sets the character backstory, given a Backstory enum instance
        """
        self.save.hero_points_save_data.player_aspect_data_path = backstory.value

    def get_selected_customizations(self, eng=False):
        """
        Returns a tuple with two elements:
            1) A dict of our currently-selected customizations, minus emotes.  If
               `eng` is `False` (the default), The keys will be `Customization`
               enum entries.  If `eng` is `True`, the keys will be the English
               labels for the customization type.  The values will be the object paths.
            2) A list of currently-chosen Emote paths.  The list will always have
               four entries, and can have `None` as an element if the user hasn't
               filled an Emote slot.

        If a customization in the selected list isn't known to us, we'll be
        extremely improper and return totally different data in the tuple.  The
        first element will be `None`, and the second will be the object path
        of the customization we don't know about.
        """
        # Construct a dict of our current customizations (minus emotes)
        if eng:
            cur_customizations = {cust_type.value: None for cust_type in Customization}
            del cur_customizations[Customization.EMOTE.value]
        else:
            cur_customizations = {cust_type: None for cust_type in Customization}
            del cur_customizations[Customization.EMOTE]
        for obj_path in self.save.selected_customizations:
            if obj_path not in profile_customizations_to_type:
                return (None, obj_path)
            cust_type = profile_customizations_to_type[obj_path]
            if cust_type != Customization.EMOTE:
                if eng:
                    key = cust_type.value
                else:
                    key = cust_type
                cur_customizations[key] = obj_path

        # Now do emotes
        cur_emotes = []
        for index in self.save.equipped_emote_customizations:
            if index == -1:
                cur_emotes.append(None)
            else:
                cur_emotes.append(self.save.selected_customizations[index])

        # Return!
        return (cur_customizations, cur_emotes)

    def get_special_selected_customizations(self):
        """
        Returns a dict of "special" customizations which don't show up in the main
        customization-selection structure in the save.  This include things like
        body type, voice pack, etc.  We will also report on slider info.
        """
        to_ret = {}
        to_ret['Overdrive Sliders'] = self.save.disable_customization_suppression
        to_ret['Voice Pack'] = self.save.player_voice.data
        to_ret['Voice Pitch'] = self.save.player_voice.pitch
        to_ret['Pronouns'] = self.save.player_pronoun_selection
        link_state = {}
        for data in self.save.customization_link_data:
            link = CustomizationLink.has_value(data.customization_name)
            if link:
                link_state[link] = data.active
                if data.active:
                    value = 'On'
                else:
                    value = 'Off'
                to_ret[f'{link.label} Symmetry'] = value
        slider_vars = set()
        for slider in customization_main_sliders:
            slider_vars.add(slider.var)
            if slider.link is not None and not link_state[slider.link]:
                slider_vars.add(slider.link_var)
        for item in self.save.custom_float_customizations:
            if item.name in slider_vars:
                to_ret[f'{item.name} Slider'] = item.value
        return to_ret

    def randomize_customizations(self, profile, categories=None, quiet=False):
        """
        Randomizes character customizations, using a TTWLProfile object `profile` to
        know which customizations are unlocked.  If `categories` is specified, it should
        be an iterable object with `Customization` enum entries describing which categories
        will be randomized.  By default, this will randomize all categories.  Returns
        `True` if we were able to do the randomization, or `False` if we encountered some
        problem preventing us from understanding the status of the customizations.  This
        will also print an error to the user in such circumstances, unless `quiet` is
        set to `True`
        """

        # Get our currently-selected customizations
        (cur_customizations, cur_emotes) = self.get_selected_customizations()
        if cur_customizations is None:
            if not quiet:
                print(f'Aborted customization randomization - Unknown customization: {cur_emotes}')
            return False

        # Now grab a dict describing our unlocked customizations from the profile
        try:
            unlocked_customizations = profile.get_cur_customizations_by_type()
        except TTWLProfile.UnknownCustomizationException as e:
            if not quiet:
                print('Aborted customization randomization - Unknown unlocked customization: {}'.format(str(e)))
            return False

        # Now loop through our desired customization randomizations and update
        # cur_customizations where appropriate
        do_emotes = False
        if categories is None:
            categories = Customization
        for cust_type in categories:
            if cust_type == Customization.EMOTE:
                cur_emotes = random.sample(list(unlocked_customizations[Customization.EMOTE]), 4)
            else:
                cur_customizations[cust_type] = random.choice(list(unlocked_customizations[cust_type]))

        # Now reconstruct our save data
        del self.save.selected_customizations[:]
        for cust_obj in cur_customizations.values():
            #print('Chose: {}'.format(cust_obj))
            self.save.selected_customizations.append(cust_obj)

        # Emotes at the end, I suppose
        del self.save.equipped_emote_customizations[:]
        for emote in cur_emotes:
            if emote is None:
                self.save.equipped_emote_customizations.append(-1)
            else:
                self.save.equipped_emote_customizations.append(len(self.save.selected_customizations))
                #print('Chose: {}'.format(emote))
                self.save.selected_customizations.append(emote)

        # Return!
        return True

    def randomize_appearance_sliders(self, overdrive=False, asymmetry_chance=0.1):
        """
        Randomizes our character's appearance sliders (and also the chosen
        pronouns and voice pack).  Basically everything that's *not* in
        the customization-unlock area handled by `randomize_customizations`.
        """

        # First, log whether we're using overdrive or not.
        self.save.disable_customization_suppression = overdrive

        # Next: decide if we're allowing asymmetry.  Defaulting to a 10%
        # chance per category?  (Or, in terms of this object, an 90% chance
        # of being linked.)
        link_state = {}
        for link in CustomizationLink:
            link_state[link] = random.random() > asymmetry_chance
        for data in self.save.customization_link_data:
            link = CustomizationLink.has_value(data.customization_name)
            if link:
                data.active = link_state[link]

        # Main Sliders
        to_update = {}
        for slider in customization_main_sliders:
            to_update.update(slider.values(link_state, overdrive))
        for item in self.save.custom_float_customizations:
            if item.name in to_update:
                item.value = to_update[item.name]

        # Voice Sliders
        self.save.player_voice.pitch = customization_pitch_slider.values(link_state, overdrive, just_first=True)

        # Pronouns
        self.save.player_pronoun_selection = random.choice(customization_pronouns)

        # Voice Pack
        self.save.player_voice.data = random.choice(customization_voices)


    def clear_dice_interacts(self):
        """
        Clears out the savegame's `tracked_interactions` structure, used as an
        additional way to know which dice have been found.  If we ever support
        *finishing* savegame dice, we may want to re-populate this structure,
        though the game itself will rebuild it if not present (based on the
        dice challenges), so that may not be worth it.  For now we're only
        supporting dice clearing on saves, though, so it's academic.
        """
        for inter in self.save.tracked_interactions:
            if inter.tracked_interaction_data == '/Game/GameData/Challenges/GoldenDice/TrackedInteraction_GoldenDice.TrackedInteraction_GoldenDice':
                del inter.completed_instances[:]
                break


    def clear_overworld_challenges(self):
        """
        Clears out Overworld challenges, or specifically the Shrine completions and
        Bottlecap shortcuts, intended for use with the --fake-tvhm option on the save
        editor.  This is done via a custom routine because there's a bunch of them,
        and there's no real point to hardcoding them.  The above methods would be
        pretty inefficient to use, too, since we loop through challenges until we
        find the correct name.
        """
        for chal in self.base_obj.challenge_data:
            if chal.challenge_class_path.startswith('/Game/GameData/Challenges/Shrines/Shrine') \
                    or chal.challenge_class_path.startswith('/Game/GameData/Challenges/OverworldShortcuts/Challenge_Crew_OWShortcut'):
                chal.completed_count = 0
                chal.currently_completed = False

