mirror of
https://github.com/EDCD/EDMarketConnector.git
synced 2025-04-12 15:27:14 +03:00
1029 lines
39 KiB
Python
1029 lines
39 KiB
Python
"""
|
||
edsm.py - Handling EDSM Data and Display.
|
||
|
||
Copyright (c) EDCD, All Rights Reserved
|
||
Licensed under the GNU General Public License.
|
||
See LICENSE file.
|
||
|
||
This is an EDMC 'core' plugin.
|
||
All EDMC plugins are *dynamically* loaded at run-time.
|
||
|
||
We build for Windows using `py2exe`.
|
||
`py2exe` can't possibly know about anything in the dynamically loaded core plugins.
|
||
|
||
Thus, you **MUST** check if any imports you add in this file are only
|
||
referenced in this file (or only in any other core plugin), and if so...
|
||
|
||
YOU MUST ENSURE THAT PERTINENT ADJUSTMENTS ARE MADE IN
|
||
`build.py` TO ENSURE THE FILES ARE ACTUALLY PRESENT
|
||
IN AN END-USER INSTALLATION ON WINDOWS.
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
import threading
|
||
import tkinter as tk
|
||
from datetime import datetime, timedelta, timezone
|
||
from queue import Queue
|
||
from threading import Thread
|
||
from time import sleep
|
||
from tkinter import ttk
|
||
from typing import Any, Literal, Mapping, MutableMapping, cast, Sequence
|
||
import requests
|
||
import killswitch
|
||
import monitor
|
||
import myNotebook as nb # noqa: N813
|
||
import plug
|
||
from companion import CAPIData
|
||
from config import applongname, appname, appversion, config, debug_senders, user_agent
|
||
from edmc_data import DEBUG_WEBSERVER_HOST, DEBUG_WEBSERVER_PORT
|
||
from EDMCLogging import get_main_logger
|
||
from ttkHyperlinkLabel import HyperlinkLabel
|
||
from l10n import translations as tr
|
||
|
||
|
||
# TODO:
|
||
# 1) Re-factor EDSM API calls out of journal_entry() into own function.
|
||
# 2) Fix how StartJump already changes things, but only partially.
|
||
# 3) Possibly this and other two 'provider' plugins could do with being
|
||
# based on a single class that they extend. There's a lot of duplicated
|
||
# logic.
|
||
# 4) Ensure the EDSM API call(back) for setting the image at end of system
|
||
# text is always fired. i.e. CAPI cmdr_data() processing.
|
||
|
||
logger = get_main_logger()
|
||
|
||
EDSM_POLL = 0.1
|
||
_TIMEOUT = 20
|
||
DISCARDED_EVENTS_SLEEP = 10
|
||
|
||
# trace-if events
|
||
CMDR_EVENTS = 'plugin.edsm.cmdr-events'
|
||
CMDR_CREDS = 'plugin.edsm.cmdr-credentials'
|
||
|
||
|
||
class This:
|
||
"""Holds module globals."""
|
||
|
||
def __init__(self):
|
||
self.shutting_down = False # Plugin is shutting down.
|
||
|
||
self.game_version = ""
|
||
self.game_build = ""
|
||
|
||
# Handle only sending Live galaxy data
|
||
self.legacy_galaxy_last_notified: datetime | None = None
|
||
|
||
self.session: requests.Session = requests.Session()
|
||
self.session.headers['User-Agent'] = user_agent
|
||
self.queue: Queue = Queue() # Items to be sent to EDSM by worker thread
|
||
self.discarded_events: set[str] = set() # List discarded events from EDSM
|
||
self.lastlookup: dict[str, Any] # Result of last system lookup
|
||
|
||
# Game state
|
||
self.multicrew: bool = False # don't send captain's ship info to EDSM while on a crew
|
||
self.coordinates: tuple[int, int, int] | None = None
|
||
self.newgame: bool = False # starting up - batch initial burst of events
|
||
self.newgame_docked: bool = False # starting up while docked
|
||
self.navbeaconscan: int = 0 # batch up burst of Scan events after NavBeaconScan
|
||
self.system_link: tk.Widget | None = None
|
||
self.system_name: tk.Tk | None = None
|
||
self.system_address: int | None = None # Frontier SystemAddress
|
||
self.system_population: int | None = None
|
||
self.station_link: tk.Widget | None = None
|
||
self.station_name: str | None = None
|
||
self.station_marketid: int | None = None # Frontier MarketID
|
||
self.on_foot = False
|
||
|
||
self._IMG_KNOWN = None
|
||
self._IMG_UNKNOWN = None
|
||
self._IMG_NEW = None
|
||
self._IMG_ERROR = None
|
||
|
||
self.thread: threading.Thread | None = None
|
||
|
||
self.log: tk.IntVar | None = None
|
||
self.log_button: ttk.Checkbutton | None = None
|
||
|
||
self.label: tk.Widget | None = None
|
||
|
||
self.cmdr_label: nb.Label | None = None
|
||
self.cmdr_text: nb.Label | None = None
|
||
|
||
self.user_label: nb.Label | None = None
|
||
self.user: nb.EntryMenu | None = None
|
||
|
||
self.apikey_label: nb.Label | None = None
|
||
self.apikey: nb.EntryMenu | None = None
|
||
|
||
|
||
this = This()
|
||
show_password_var = tk.BooleanVar()
|
||
|
||
STATION_UNDOCKED: str = '×' # "Station" name to display when not docked = U+00D7
|
||
__cleanup = str.maketrans({' ': None, '\n': None})
|
||
IMG_KNOWN_B64 = """
|
||
R0lGODlhEAAQAMIEAFWjVVWkVWS/ZGfFZ////////////////yH5BAEKAAQALAAAAAAQABAAAAMvSLrc/lAFIUIkYOgNXt5g14Dk0AQlaC1CuglM6w7wgs7r
|
||
MpvNV4q932VSuRiPjQQAOw==
|
||
""".translate(__cleanup)
|
||
|
||
IMG_UNKNOWN_B64 = """
|
||
R0lGODlhEAAQAKEDAGVLJ+ddWO5fW////yH5BAEKAAMALAAAAAAQABAAAAItnI+pywYRQBtA2CtVvTwjDgrJFlreEJRXgKSqwB5keQ6vOKq1E+7IE5kIh4kC
|
||
ADs=
|
||
""".translate(__cleanup)
|
||
|
||
IMG_NEW_B64 = """
|
||
R0lGODlhEAAQAMZwANKVHtWcIteiHuiqLPCuHOS1MN22ZeW7ROG6Zuu9MOy+K/i8Kf/DAuvCVf/FAP3BNf/JCf/KAPHHSv7ESObHdv/MBv/GRv/LGP/QBPXO
|
||
PvjPQfjQSvbRSP/UGPLSae7Sfv/YNvLXgPbZhP7dU//iI//mAP/jH//kFv7fU//fV//ebv/iTf/iUv/kTf/iZ/vgiP/hc/vgjv/jbfriiPriiv7ka//if//j
|
||
d//sJP/oT//tHv/mZv/sLf/rRP/oYv/rUv/paP/mhv/sS//oc//lkf/mif/sUf/uPv/qcv/uTv/uUv/vUP/qhP/xP//pm//ua//sf//ubf/wXv/thv/tif/s
|
||
lv/tjf/smf/yYP/ulf/2R//2Sv/xkP/2av/0gP/ylf/2df/0i//0j//0lP/5cP/7a//1p//5gf/7ev/3o//2sf/5mP/6kv/2vP/3y//+jP//////////////
|
||
/////////////////////////////////////////////////yH5BAEKAH8ALAAAAAAQABAAAAePgH+Cg4SFhoJKPIeHYT+LhVppUTiPg2hrUkKPXWdlb2xH
|
||
Jk9jXoNJQDk9TVtkYCUkOy4wNjdGfy1UXGJYOksnPiwgFwwYg0NubWpmX1ArHREOFYUyWVNIVkxXQSoQhyMoNVUpRU5EixkcMzQaGy8xhwsKHiEfBQkSIg+G
|
||
BAcUCIIBBDSYYGiAAUMALFR6FAgAOw==
|
||
""".translate(__cleanup)
|
||
|
||
IMG_ERR_B64 = """
|
||
R0lGODlhEAAQAKEBAAAAAP///////////yH5BAEKAAIALAAAAAAQABAAAAIwlBWpeR0AIwwNPRmZuVNJinyWuClhBlZjpm5fqnIAHJPtOd3Hou9mL6NVgj2L
|
||
plEAADs=
|
||
""".translate(__cleanup)
|
||
|
||
|
||
# Main window clicks
|
||
def system_url(system_name: str) -> str:
|
||
"""
|
||
Construct an appropriate EDSM URL for the provided system.
|
||
|
||
:param system_name: Will be overridden with `this.system_address` if that
|
||
is set.
|
||
:return: The URL, empty if no data was available to construct it.
|
||
"""
|
||
if this.system_address:
|
||
return requests.utils.requote_uri(f'https://www.edsm.net/en/system?systemID64={this.system_address}')
|
||
|
||
if system_name:
|
||
return requests.utils.requote_uri(f'https://www.edsm.net/en/system?systemName={system_name}')
|
||
|
||
return ''
|
||
|
||
|
||
def station_url(system_name: str, station_name: str) -> str:
|
||
"""
|
||
Construct an appropriate EDSM URL for a station.
|
||
|
||
:param system_name: Name of the system the station is in.
|
||
:param station_name: Name of the station.
|
||
:return: The URL, empty if no data was available to construct it.
|
||
"""
|
||
if system_name and station_name:
|
||
return requests.utils.requote_uri(
|
||
f'https://www.edsm.net/en/system?systemName={system_name}&stationName={station_name}'
|
||
)
|
||
|
||
# monitor state might think these are gone, but we don't yet
|
||
if this.system_name and this.station_name:
|
||
return requests.utils.requote_uri(
|
||
f'https://www.edsm.net/en/system?systemName={this.system_name}&stationName={this.station_name}'
|
||
)
|
||
|
||
if system_name:
|
||
return requests.utils.requote_uri(
|
||
f'https://www.edsm.net/en/system?systemName={system_name}&stationName=ALL'
|
||
)
|
||
|
||
return ''
|
||
|
||
|
||
def plugin_start3(plugin_dir: str) -> str:
|
||
"""
|
||
Start the plugin.
|
||
|
||
:param plugin_dir: Name of directory this was loaded from.
|
||
:return: Identifier string for this plugin.
|
||
"""
|
||
# Can't be earlier since can only call PhotoImage after window is created
|
||
this._IMG_KNOWN = tk.PhotoImage(data=IMG_KNOWN_B64) # green circle
|
||
this._IMG_UNKNOWN = tk.PhotoImage(data=IMG_UNKNOWN_B64) # red circle
|
||
this._IMG_NEW = tk.PhotoImage(data=IMG_NEW_B64) # yellow star
|
||
this._IMG_ERROR = tk.PhotoImage(data=IMG_ERR_B64) # BBC Mode 5 '?'
|
||
|
||
# Migrate old settings
|
||
if not config.get_list('edsm_cmdrs'):
|
||
if isinstance(config.get_list('cmdrs'), list) and \
|
||
config.get_list('edsm_usernames') and config.get_list('edsm_apikeys'):
|
||
# Migrate <= 2.34 settings
|
||
config.set('edsm_cmdrs', config.get_list('cmdrs'))
|
||
|
||
elif config.get_list('edsm_cmdrname'):
|
||
# Migrate <= 2.25 settings. edsm_cmdrs is unknown at this time
|
||
config.set('edsm_usernames', [config.get_str('edsm_cmdrname', default='')])
|
||
config.set('edsm_apikeys', [config.get_str('edsm_apikey', default='')])
|
||
|
||
config.delete('edsm_cmdrname', suppress=True)
|
||
config.delete('edsm_apikey', suppress=True)
|
||
|
||
if config.get_int('output') & 256:
|
||
# Migrate <= 2.34 setting
|
||
config.set('edsm_out', 1)
|
||
|
||
config.delete('edsm_autoopen', suppress=True)
|
||
config.delete('edsm_historical', suppress=True)
|
||
|
||
logger.debug('Starting worker thread...')
|
||
this.thread = Thread(target=worker, name='EDSM worker')
|
||
this.thread.daemon = True
|
||
this.thread.start()
|
||
logger.debug('Done.')
|
||
|
||
return 'EDSM'
|
||
|
||
|
||
def plugin_app(parent: tk.Tk) -> None:
|
||
"""
|
||
Construct this plugin's main UI, if any.
|
||
|
||
:param parent: The tk parent to place our widgets into.
|
||
:return: See PLUGINS.md#display
|
||
"""
|
||
# system label in main window
|
||
this.system_link = parent.nametowidget(f".{appname.lower()}.system")
|
||
if this.system_link is None:
|
||
logger.error("Couldn't look up system widget!!!")
|
||
return
|
||
|
||
this.system_link.bind_all('<<EDSMStatus>>', update_status)
|
||
# station label in main window
|
||
this.station_link = parent.nametowidget(f".{appname.lower()}.station")
|
||
|
||
|
||
def plugin_stop() -> None:
|
||
"""Stop this plugin."""
|
||
logger.debug('Signalling queue to close...')
|
||
# Signal thread to close and wait for it
|
||
this.shutting_down = True
|
||
this.queue.put(None) # Still necessary to get `this.queue.get()` to unblock
|
||
this.thread.join() # type: ignore
|
||
this.thread = None
|
||
this.session.close()
|
||
# Suppress 'Exception ignored in: <function Image.__del__ at ...>' errors # TODO: this is bad.
|
||
this._IMG_KNOWN = this._IMG_UNKNOWN = this._IMG_NEW = this._IMG_ERROR = None
|
||
logger.debug('Done.')
|
||
|
||
|
||
def toggle_password_visibility():
|
||
"""Toggle if the API Key is visible or not."""
|
||
if show_password_var.get():
|
||
this.apikey.config(show="") # type: ignore
|
||
else:
|
||
this.apikey.config(show="*") # type: ignore
|
||
|
||
|
||
def plugin_prefs(parent: ttk.Notebook, cmdr: str | None, is_beta: bool) -> nb.Frame:
|
||
"""
|
||
Plugin preferences setup hook.
|
||
|
||
Any tkinter UI set up *must* be within an instance of `myNotebook.Frame`,
|
||
which is the return value of this function.
|
||
|
||
:param parent: tkinter Widget to place items in.
|
||
:param cmdr: Name of Commander.
|
||
:param is_beta: Whether game beta was detected.
|
||
:return: An instance of `myNotebook.Frame`.
|
||
"""
|
||
PADX = 10 # noqa: N806
|
||
BUTTONX = 12 # noqa: N806
|
||
PADY = 1 # noqa: N806
|
||
BOXY = 2 # noqa: N806
|
||
SEPY = 10 # noqa: N806
|
||
|
||
frame = nb.Frame(parent)
|
||
frame.columnconfigure(1, weight=1)
|
||
|
||
cur_row = 0
|
||
HyperlinkLabel(
|
||
frame,
|
||
text='Elite Dangerous Star Map',
|
||
background=nb.Label().cget('background'),
|
||
url='https://www.edsm.net/',
|
||
underline=True
|
||
).grid(row=cur_row, columnspan=2, padx=PADX, pady=PADY, sticky=tk.W)
|
||
cur_row += 1
|
||
|
||
this.log = tk.IntVar(value=config.get_int('edsm_out') and 1)
|
||
this.log_button = nb.Checkbutton(
|
||
frame,
|
||
# LANG: Settings>EDSM - Label on checkbox for 'send data'
|
||
text=tr.tl('Send flight log and CMDR status to EDSM'),
|
||
variable=this.log,
|
||
command=prefsvarchanged
|
||
)
|
||
if this.log_button:
|
||
this.log_button.grid(row=cur_row, columnspan=2, padx=BUTTONX, pady=PADY, sticky=tk.W)
|
||
cur_row += 1
|
||
|
||
ttk.Separator(frame, orient=tk.HORIZONTAL).grid(
|
||
columnspan=2, padx=PADX, pady=SEPY, sticky=tk.EW, row=cur_row
|
||
)
|
||
cur_row += 1
|
||
|
||
this.label = HyperlinkLabel(
|
||
frame,
|
||
text=tr.tl('Elite Dangerous Star Map credentials'), # LANG: Elite Dangerous Star Map credentials
|
||
background=nb.Label().cget('background'),
|
||
url='https://www.edsm.net/settings/api',
|
||
underline=True
|
||
)
|
||
if this.label:
|
||
this.label.grid(row=cur_row, columnspan=2, padx=PADX, pady=PADY, sticky=tk.W)
|
||
cur_row += 1
|
||
this.cmdr_label = nb.Label(frame, text=tr.tl('Cmdr')) # LANG: Game Commander name label in EDSM settings
|
||
this.cmdr_label.grid(row=cur_row, padx=PADX, pady=PADY, sticky=tk.W)
|
||
this.cmdr_text = nb.Label(frame)
|
||
this.cmdr_text.grid(row=cur_row, column=1, padx=PADX, pady=BOXY, sticky=tk.W)
|
||
|
||
cur_row += 1
|
||
# LANG: EDSM Commander name label in EDSM settings
|
||
this.user_label = nb.Label(frame, text=tr.tl('Commander Name'))
|
||
this.user_label.grid(row=cur_row, padx=PADX, pady=PADY, sticky=tk.W)
|
||
this.user = nb.EntryMenu(frame)
|
||
this.user.grid(row=cur_row, column=1, padx=PADX, pady=BOXY, sticky=tk.EW)
|
||
|
||
cur_row += 1
|
||
# LANG: EDSM API key label
|
||
this.apikey_label = nb.Label(frame, text=tr.tl('API Key'))
|
||
this.apikey_label.grid(row=cur_row, padx=PADX, pady=PADY, sticky=tk.W)
|
||
this.apikey = nb.EntryMenu(frame, show="*", width=50)
|
||
this.apikey.grid(row=cur_row, column=1, padx=PADX, pady=BOXY, sticky=tk.EW)
|
||
cur_row += 1
|
||
|
||
prefs_cmdr_changed(cmdr, is_beta)
|
||
|
||
show_password_var.set(False) # Password is initially masked
|
||
|
||
show_password_checkbox = nb.Checkbutton(
|
||
frame,
|
||
text=tr.tl('Show API Key'), # LANG: Text EDSM Show API Key
|
||
variable=show_password_var,
|
||
command=toggle_password_visibility
|
||
)
|
||
show_password_checkbox.grid(row=cur_row, columnspan=2, padx=BUTTONX, pady=PADY, sticky=tk.W)
|
||
|
||
return frame
|
||
|
||
|
||
def prefs_cmdr_changed(cmdr: str | None, is_beta: bool) -> None: # noqa: CCR001
|
||
"""
|
||
Handle the Commander name changing whilst Settings was open.
|
||
|
||
:param cmdr: The new current Commander name.
|
||
:param is_beta: Whether game beta was detected.
|
||
"""
|
||
if this.log_button:
|
||
this.log_button['state'] = tk.NORMAL if cmdr and not is_beta else tk.DISABLED
|
||
if this.user:
|
||
this.user['state'] = tk.NORMAL
|
||
this.user.delete(0, tk.END)
|
||
if this.apikey:
|
||
this.apikey['state'] = tk.NORMAL
|
||
this.apikey.delete(0, tk.END)
|
||
if cmdr:
|
||
if this.cmdr_text:
|
||
this.cmdr_text['text'] = f'{cmdr}{" [Beta]" if is_beta else ""}'
|
||
cred = credentials(cmdr)
|
||
if cred:
|
||
if this.user:
|
||
this.user.insert(0, cred[0])
|
||
if this.apikey:
|
||
this.apikey.insert(0, cred[1])
|
||
else:
|
||
if this.cmdr_text:
|
||
# LANG: We have no data on the current commander
|
||
this.cmdr_text['text'] = tr.tl('None')
|
||
|
||
to_set: Literal['normal'] | Literal['disabled'] = tk.DISABLED
|
||
if cmdr and not is_beta and this.log and this.log.get():
|
||
to_set = tk.NORMAL
|
||
|
||
set_prefs_ui_states(to_set)
|
||
|
||
|
||
def prefsvarchanged() -> None:
|
||
"""Handle the 'Send data to EDSM' tickbox changing state."""
|
||
to_set = tk.DISABLED
|
||
if this.log and this.log.get() and this.log_button:
|
||
to_set = this.log_button['state']
|
||
|
||
set_prefs_ui_states(to_set)
|
||
|
||
|
||
def set_prefs_ui_states(state: str) -> None:
|
||
"""
|
||
Set the state of various config UI entries.
|
||
|
||
:param state: the state to set each entry to
|
||
|
||
# NOTE: This may break things, watch out in testing. (5.10)
|
||
"""
|
||
elements = [
|
||
this.label,
|
||
this.cmdr_label,
|
||
this.cmdr_text,
|
||
this.user_label,
|
||
this.user,
|
||
this.apikey_label,
|
||
this.apikey
|
||
]
|
||
|
||
for element in elements:
|
||
if element:
|
||
element['state'] = state
|
||
|
||
|
||
def prefs_changed(cmdr: str, is_beta: bool) -> None:
|
||
"""
|
||
Handle any changes to Settings once the dialog is closed.
|
||
|
||
:param cmdr: Name of Commander.
|
||
:param is_beta: Whether game beta was detected.
|
||
"""
|
||
if this.log:
|
||
config.set('edsm_out', this.log.get())
|
||
|
||
if cmdr and not is_beta:
|
||
cmdrs: list[str] = config.get_list('edsm_cmdrs', default=[])
|
||
usernames: list[str] = config.get_list('edsm_usernames', default=[])
|
||
apikeys: list[str] = config.get_list('edsm_apikeys', default=[])
|
||
|
||
if this.user and this.apikey:
|
||
if cmdr in cmdrs:
|
||
idx = cmdrs.index(cmdr)
|
||
usernames.extend([''] * (1 + idx - len(usernames)))
|
||
usernames[idx] = this.user.get().strip()
|
||
apikeys.extend([''] * (1 + idx - len(apikeys)))
|
||
apikeys[idx] = this.apikey.get().strip()
|
||
else:
|
||
config.set('edsm_cmdrs', cmdrs + [cmdr])
|
||
usernames.append(this.user.get().strip())
|
||
apikeys.append(this.apikey.get().strip())
|
||
|
||
config.set('edsm_usernames', usernames)
|
||
config.set('edsm_apikeys', apikeys)
|
||
|
||
|
||
def credentials(cmdr: str) -> tuple[str, str] | None:
|
||
"""
|
||
Get credentials for the given commander, if they exist.
|
||
|
||
:param cmdr: The commander to get credentials for
|
||
:return: The credentials, or None
|
||
"""
|
||
logger.trace_if(CMDR_CREDS, f'{cmdr=}')
|
||
|
||
# Credentials for cmdr
|
||
if not cmdr:
|
||
return None
|
||
|
||
cmdrs = config.get_list('edsm_cmdrs')
|
||
if not cmdrs:
|
||
# Migrate from <= 2.25
|
||
cmdrs = [cmdr]
|
||
config.set('edsm_cmdrs', cmdrs)
|
||
|
||
edsm_usernames = config.get_list('edsm_usernames')
|
||
edsm_apikeys = config.get_list('edsm_apikeys')
|
||
|
||
if cmdr in cmdrs and len(cmdrs) == len(edsm_usernames) == len(edsm_apikeys):
|
||
idx = cmdrs.index(cmdr)
|
||
if idx < len(edsm_usernames) and idx < len(edsm_apikeys):
|
||
logger.trace_if(CMDR_CREDS, f'{cmdr=}: returning ({edsm_usernames[idx]=}, {edsm_apikeys[idx]=})')
|
||
return edsm_usernames[idx], edsm_apikeys[idx]
|
||
|
||
logger.trace_if(CMDR_CREDS, f'{cmdr=}: returning None')
|
||
return None
|
||
|
||
|
||
def journal_entry( # noqa: C901, CCR001
|
||
cmdr: str, is_beta: bool, system: str, station: str, entry: MutableMapping[str, Any], state: Mapping[str, Any]
|
||
) -> str:
|
||
"""
|
||
Handle a new Journal event.
|
||
|
||
:param cmdr: Name of Commander.
|
||
:param is_beta: Whether game beta was detected.
|
||
:param system: Name of current tracked system.
|
||
:param station: Name of current tracked station location.
|
||
:param entry: The journal event.
|
||
:param state: `monitor.state`
|
||
:return: None if no error, else an error string.
|
||
"""
|
||
should_return, new_entry = killswitch.check_killswitch('plugins.edsm.journal', entry, logger)
|
||
if should_return:
|
||
# LANG: EDSM plugin - Journal handling disabled by killswitch
|
||
plug.show_error(tr.tl('EDSM Handler disabled. See Log.'))
|
||
return ''
|
||
|
||
should_return, new_entry = killswitch.check_killswitch(
|
||
f'plugins.edsm.journal.event.{entry["event"]}', data=new_entry, log=logger
|
||
)
|
||
|
||
if should_return:
|
||
return ''
|
||
|
||
this.game_version = state['GameVersion']
|
||
this.game_build = state['GameBuild']
|
||
this.system_address = state['SystemAddress']
|
||
this.system_name = state['SystemName']
|
||
this.system_population = state['SystemPopulation']
|
||
this.station_name = state['StationName']
|
||
this.station_marketid = state['MarketID']
|
||
|
||
entry = new_entry
|
||
|
||
this.on_foot = state['OnFoot']
|
||
if entry['event'] in ('CarrierJump', 'FSDJump', 'Location', 'Docked'):
|
||
logger.trace_if(
|
||
'journal.locations', f'''{entry["event"]}
|
||
Commander: {cmdr}
|
||
System: {system}
|
||
Station: {station}
|
||
state: {state!r}
|
||
entry: {entry!r}'''
|
||
)
|
||
|
||
if config.get_str('station_provider') == 'EDSM':
|
||
to_set = this.station_name
|
||
if not this.station_name:
|
||
if this.system_population and this.system_population > 0:
|
||
to_set = STATION_UNDOCKED
|
||
else:
|
||
to_set = ''
|
||
|
||
if this.station_link:
|
||
this.station_link['text'] = to_set
|
||
this.station_link['url'] = station_url(str(this.system_name), str(this.station_name))
|
||
this.station_link.update_idletasks()
|
||
|
||
# Update display of 'EDSM Status' image
|
||
if this.system_link and this.system_link['text'] != system:
|
||
this.system_link['text'] = system if system else ''
|
||
this.system_link['image'] = ''
|
||
this.system_link.update_idletasks()
|
||
|
||
this.multicrew = bool(state['Role'])
|
||
if 'StarPos' in entry:
|
||
this.coordinates = entry['StarPos']
|
||
elif entry['event'] == 'LoadGame':
|
||
this.coordinates = None
|
||
|
||
if entry['event'] in ('LoadGame', 'Commander', 'NewCommander'):
|
||
this.newgame = True
|
||
this.newgame_docked = False
|
||
this.navbeaconscan = 0
|
||
elif entry['event'] == 'StartUp':
|
||
this.newgame = False
|
||
this.newgame_docked = False
|
||
this.navbeaconscan = 0
|
||
elif entry['event'] == 'Location':
|
||
this.newgame = True
|
||
this.newgame_docked = entry.get('Docked', False)
|
||
this.navbeaconscan = 0
|
||
elif entry['event'] == 'NavBeaconScan':
|
||
this.navbeaconscan = entry['NumBodies']
|
||
elif entry['event'] == 'BackPack':
|
||
# Use the stored file contents, not the empty journal event
|
||
if state['BackpackJSON']:
|
||
entry = state['BackpackJSON']
|
||
|
||
# Queue all events to send to EDSM. worker() will take care of dropping EDSM discarded events
|
||
if config.get_int('edsm_out') and not is_beta and not this.multicrew and credentials(cmdr):
|
||
if not monitor.monitor.is_live_galaxy():
|
||
logger.info("EDSM only accepts Live galaxy data")
|
||
# Since Update 14 on 2022-11-29 Inara only accepts Live data.
|
||
if (
|
||
this.legacy_galaxy_last_notified is None
|
||
or (datetime.now(timezone.utc) - this.legacy_galaxy_last_notified) > timedelta(seconds=300)
|
||
):
|
||
# LANG: The Inara API only accepts Live galaxy data, not Legacy galaxy data
|
||
logger.info("EDSM only accepts Live galaxy data")
|
||
this.legacy_galaxy_last_notified = datetime.now(timezone.utc)
|
||
return tr.tl("EDSM only accepts Live galaxy data") # LANG: EDSM - Only Live data
|
||
|
||
return ''
|
||
|
||
# Introduce transient states into the event
|
||
transient = {
|
||
'_systemName': system,
|
||
'_systemCoordinates': this.coordinates,
|
||
'_stationName': station,
|
||
'_shipId': state['ShipID'],
|
||
}
|
||
|
||
entry.update(transient)
|
||
|
||
if entry['event'] == 'LoadGame':
|
||
# Synthesise Materials events on LoadGame since we will have missed it
|
||
materials = {
|
||
'timestamp': entry['timestamp'],
|
||
'event': 'Materials',
|
||
'Raw': [{'Name': k, 'Count': v} for k, v in state['Raw'].items()],
|
||
'Manufactured': [{'Name': k, 'Count': v} for k, v in state['Manufactured'].items()],
|
||
'Encoded': [{'Name': k, 'Count': v} for k, v in state['Encoded'].items()],
|
||
}
|
||
materials.update(transient)
|
||
logger.trace_if(CMDR_EVENTS, f'"LoadGame" event, queueing Materials: {cmdr=}')
|
||
this.queue.put((cmdr, this.game_version, this.game_build, materials))
|
||
|
||
if entry['event'] in ('CarrierJump', 'FSDJump', 'Location', 'Docked'):
|
||
logger.trace_if(
|
||
'journal.locations', f'''{entry["event"]}
|
||
Queueing: {entry!r}'''
|
||
)
|
||
logger.trace_if(CMDR_EVENTS, f'"{entry["event"]=}" event, queueing: {cmdr=}')
|
||
this.queue.put((cmdr, this.game_version, this.game_build, entry))
|
||
|
||
return ''
|
||
|
||
|
||
# Update system data
|
||
def cmdr_data(data: CAPIData, is_beta: bool) -> str | None: # noqa: CCR001
|
||
"""
|
||
Process new CAPI data.
|
||
|
||
:param data: The latest merged CAPI data.
|
||
:param is_beta: Whether game beta was detected.
|
||
:return: Optional error string.
|
||
"""
|
||
system = data['lastSystem']['name']
|
||
|
||
# Always store initially, even if we're not the *current* system provider.
|
||
if not this.station_marketid and data['commander']['docked']:
|
||
this.station_marketid = data['lastStarport']['id']
|
||
# Only trust CAPI if these aren't yet set
|
||
if not this.system_name:
|
||
this.system_name = data['lastSystem']['name']
|
||
if not this.station_name and data['commander']['docked']:
|
||
this.station_name = data['lastStarport']['name']
|
||
|
||
# TODO: Fire off the EDSM API call to trigger the callback for the icons
|
||
|
||
if config.get_str('system_provider') == 'EDSM':
|
||
if this.system_link:
|
||
this.system_link['text'] = this.system_name
|
||
# Do *NOT* set 'url' here, as it's set to a function that will call
|
||
# through correctly. We don't want a static string.
|
||
this.system_link.update_idletasks()
|
||
if config.get_str('station_provider') == 'EDSM':
|
||
if this.station_link:
|
||
if data['commander']['docked'] or this.on_foot and this.station_name:
|
||
this.station_link['text'] = this.station_name
|
||
elif data['lastStarport']['name'] and data['lastStarport']['name'] != "":
|
||
this.station_link['text'] = STATION_UNDOCKED
|
||
else:
|
||
this.station_link['text'] = ''
|
||
|
||
# Do *NOT* set 'url' here, as it's set to a function that will call
|
||
# through correctly. We don't want a static string.
|
||
this.station_link.update_idletasks()
|
||
|
||
if this.system_link and not this.system_link['text']:
|
||
this.system_link['text'] = system
|
||
this.system_link['image'] = ''
|
||
this.system_link.update_idletasks()
|
||
|
||
return ''
|
||
|
||
|
||
TARGET_URL = 'https://www.edsm.net/api-journal-v1'
|
||
if 'edsm' in debug_senders:
|
||
TARGET_URL = f'http://{DEBUG_WEBSERVER_HOST}:{DEBUG_WEBSERVER_PORT}/edsm'
|
||
|
||
|
||
def get_discarded_events_list() -> None:
|
||
"""
|
||
Retrieve the list of events to discard from EDSM.
|
||
|
||
This function queries the EDSM API to obtain the list of events that should be discarded,
|
||
and stores them in the `discarded_events` attribute.
|
||
|
||
:return: None
|
||
"""
|
||
try:
|
||
r = this.session.get('https://www.edsm.net/api-journal-v1/discard', timeout=_TIMEOUT)
|
||
r.raise_for_status()
|
||
this.discarded_events = set(r.json())
|
||
# We discard 'Docked' events because should_send() assumes that we send them
|
||
this.discarded_events.discard('Docked')
|
||
if not this.discarded_events:
|
||
logger.warning(
|
||
'Unexpected empty discarded events list from EDSM: '
|
||
f'{type(this.discarded_events)} -- {this.discarded_events}'
|
||
)
|
||
except Exception as e:
|
||
logger.warning('Exception while trying to set this.discarded_events:', exc_info=e)
|
||
|
||
|
||
def process_discarded_events() -> None:
|
||
"""Process discarded events until the discarded events list is retrieved or the shutdown signal is received."""
|
||
while not this.discarded_events:
|
||
if this.shutting_down:
|
||
logger.debug(f'returning from discarded_events loop due to {this.shutting_down=}')
|
||
return
|
||
get_discarded_events_list()
|
||
if this.discarded_events:
|
||
break
|
||
sleep(DISCARDED_EVENTS_SLEEP)
|
||
|
||
logger.debug('Got "events to discard" list, commencing queue consumption...')
|
||
|
||
|
||
def send_to_edsm( # noqa: CCR001
|
||
data: dict[str, Sequence[object]], pending: list[Mapping[str, Any]], closing: bool
|
||
) -> list[Mapping[str, Any]]:
|
||
"""Send data to the EDSM API endpoint and handle the API response."""
|
||
response = this.session.post(TARGET_URL, data=data, timeout=_TIMEOUT)
|
||
logger.trace_if('plugin.edsm.api', f'API response content: {response.content!r}')
|
||
|
||
# Check for rate limit headers
|
||
rate_limit_remaining = response.headers.get('X-Rate-Limit-Remaining')
|
||
rate_limit_reset = response.headers.get('X-Rate-Limit-Reset')
|
||
|
||
# Convert headers to integers if they exist
|
||
try:
|
||
remaining = int(rate_limit_remaining) if rate_limit_remaining else None
|
||
reset = int(rate_limit_reset) if rate_limit_reset else None
|
||
except ValueError:
|
||
remaining = reset = None
|
||
|
||
if remaining is not None and reset is not None:
|
||
# Respect rate limits if they exist
|
||
if remaining == 0:
|
||
# Calculate sleep time until the rate limit reset time
|
||
reset_time = datetime.utcfromtimestamp(reset)
|
||
current_time = datetime.utcnow()
|
||
|
||
sleep_time = (reset_time - current_time).total_seconds()
|
||
|
||
if sleep_time > 0:
|
||
sleep(sleep_time)
|
||
|
||
response.raise_for_status()
|
||
reply = response.json()
|
||
msg_num = reply['msgnum']
|
||
msg = reply['msg']
|
||
# 1xx = OK
|
||
# 2xx = fatal error
|
||
# 3&4xx not generated at top-level
|
||
# 5xx = error but events saved for later processing
|
||
|
||
if msg_num // 100 == 2:
|
||
logger.warning(f'EDSM\t{msg_num} {msg}\t{json.dumps(pending, separators=(",", ": "))}')
|
||
# LANG: EDSM Plugin - Error message from EDSM API
|
||
plug.show_error(tr.tl('Error: EDSM {MSG}').format(MSG=msg))
|
||
else:
|
||
if msg_num // 100 == 1:
|
||
logger.trace_if('plugin.edsm.api', 'Overall OK')
|
||
pass
|
||
elif msg_num // 100 == 5:
|
||
logger.trace_if('plugin.edsm.api', 'Event(s) not currently processed, but saved for later')
|
||
pass
|
||
else:
|
||
logger.warning(f'EDSM API call status not 1XX, 2XX or 5XX: {msg.num}')
|
||
|
||
for e, r in zip(pending, reply['events']):
|
||
if not closing and e['event'] in ('StartUp', 'Location', 'FSDJump', 'CarrierJump'):
|
||
# Update main window's system status
|
||
this.lastlookup = r
|
||
# calls update_status in main thread
|
||
if not config.shutting_down and this.system_link is not None:
|
||
this.system_link.event_generate('<<EDSMStatus>>', when="tail")
|
||
if r['msgnum'] // 100 != 1:
|
||
logger.warning(f'EDSM event with not-1xx status:\n{r["msgnum"]}\n'
|
||
f'{r["msg"]}\n{json.dumps(e, separators=(",", ": "))}')
|
||
pending = []
|
||
return pending
|
||
|
||
|
||
def worker() -> None: # noqa: CCR001 C901
|
||
"""
|
||
Handle uploading events to EDSM API.
|
||
|
||
This function is the target function of a thread. It processes events from the queue until the
|
||
queued item is None, uploading the events to the EDSM API.
|
||
|
||
:return: None
|
||
"""
|
||
logger.debug('Starting...')
|
||
pending: list[Mapping[str, Any]] = [] # Unsent events
|
||
closing = False
|
||
cmdr: str = ""
|
||
last_game_version = ""
|
||
last_game_build = ""
|
||
|
||
# Process the Discard Queue
|
||
process_discarded_events()
|
||
|
||
while True:
|
||
if this.shutting_down:
|
||
logger.debug(f'{this.shutting_down=}, so setting closing = True')
|
||
closing = True
|
||
|
||
item: tuple[str, str, str, Mapping[str, Any]] | None = this.queue.get()
|
||
if item:
|
||
(cmdr, game_version, game_build, entry) = item
|
||
logger.trace_if(CMDR_EVENTS, f'De-queued ({cmdr=}, {game_version=}, {game_build=}, {entry["event"]=})')
|
||
else:
|
||
logger.debug('Empty queue message, setting closing = True')
|
||
closing = True # Try to send any unsent events before we close
|
||
entry = {'event': 'ShutDown'} # Dummy to allow for `entry['event']` below
|
||
|
||
retrying = 0
|
||
while retrying < 3:
|
||
if item is None:
|
||
item = cast(tuple[str, str, str, Mapping[str, Any]], ("", {}))
|
||
should_skip, new_item = killswitch.check_killswitch(
|
||
'plugins.edsm.worker',
|
||
item,
|
||
logger
|
||
)
|
||
|
||
if should_skip:
|
||
break
|
||
if item is not None:
|
||
item = new_item
|
||
|
||
try:
|
||
if item and entry['event'] not in this.discarded_events:
|
||
logger.trace_if(
|
||
CMDR_EVENTS, f'({cmdr=}, {entry["event"]=}): not in discarded_events, appending to pending')
|
||
|
||
# Discard the pending list if it's a new Journal file OR
|
||
# if the gameversion has changed. We claim a single
|
||
# gameversion for an entire batch of events so can't mix
|
||
# them.
|
||
# The specific gameversion check caters for scenarios where
|
||
# we took some time in the last POST, had new events queued
|
||
# in the meantime *and* the game client crashed *and* was
|
||
# changed to a different gameversion.
|
||
if (
|
||
entry['event'].lower() == 'fileheader'
|
||
or last_game_version != game_version or last_game_build != game_build
|
||
):
|
||
pending = []
|
||
pending.append(entry)
|
||
# drop events if required by killswitch
|
||
new_pending = []
|
||
for e in pending:
|
||
skip, new = killswitch.check_killswitch(f'plugin.edsm.worker.{e["event"]}', e, logger)
|
||
if skip:
|
||
continue
|
||
new_pending.append(new)
|
||
pending = new_pending
|
||
|
||
if pending and should_send(pending, entry['event']):
|
||
logger.trace_if(CMDR_EVENTS, f'({cmdr=}, {entry["event"]=}): should_send() said True')
|
||
logger.trace_if(CMDR_EVENTS, f'pending contains:\n{chr(0x0A).join(str(p) for p in pending)}')
|
||
|
||
if any(p for p in pending if p['event'] in ('CarrierJump', 'FSDJump', 'Location', 'Docked')):
|
||
logger.trace_if('journal.locations', "pending has at least one of "
|
||
"('CarrierJump', 'FSDJump', 'Location', 'Docked')"
|
||
" and it passed should_send()")
|
||
for p in pending:
|
||
if p['event'] in 'Location':
|
||
logger.trace_if(
|
||
'journal.locations',
|
||
f'"Location" event in pending passed should_send(), timestamp: {p["timestamp"]}'
|
||
)
|
||
|
||
creds = credentials(cmdr)
|
||
if creds is None:
|
||
raise ValueError("Unexpected lack of credentials")
|
||
|
||
username, apikey = creds
|
||
logger.trace_if(CMDR_EVENTS, f'({cmdr=}, {entry["event"]=}): Using {username=} from credentials()')
|
||
|
||
data = {
|
||
'commanderName': username.encode('utf-8'),
|
||
'apiKey': apikey,
|
||
'fromSoftware': applongname,
|
||
'fromSoftwareVersion': str(appversion()),
|
||
'fromGameVersion': game_version,
|
||
'fromGameBuild': game_build,
|
||
'message': json.dumps(pending, ensure_ascii=False).encode('utf-8'),
|
||
}
|
||
|
||
if any(p for p in pending if p['event'] in ('CarrierJump', 'FSDJump', 'Location', 'Docked')):
|
||
data_elided = data.copy()
|
||
data_elided['apiKey'] = '<elided>'
|
||
if isinstance(data_elided['message'], bytes):
|
||
data_elided['message'] = data_elided['message'].decode('utf-8')
|
||
if isinstance(data_elided['commanderName'], bytes):
|
||
data_elided['commanderName'] = data_elided['commanderName'].decode('utf-8')
|
||
logger.trace_if(
|
||
'journal.locations',
|
||
"pending has at least one of ('CarrierJump', 'FSDJump', 'Location', 'Docked')"
|
||
" Attempting API call with the following events:"
|
||
)
|
||
for p in pending:
|
||
logger.trace_if('journal.locations', f"Event: {p!r}")
|
||
if p['event'] in 'Location':
|
||
logger.trace_if(
|
||
'journal.locations',
|
||
f'Attempting API call for "Location" event with timestamp: {p["timestamp"]}'
|
||
)
|
||
logger.trace_if(
|
||
'journal.locations', f'Overall POST data (elided) is:\n{json.dumps(data_elided, indent=2)}'
|
||
)
|
||
|
||
pending = send_to_edsm(data, pending, closing)
|
||
|
||
break # No exception, so assume success
|
||
|
||
except Exception as e:
|
||
logger.debug(f'Attempt to send API events: retrying == {retrying}', exc_info=e)
|
||
retrying += 1
|
||
|
||
else:
|
||
# LANG: EDSM Plugin - Error connecting to EDSM API
|
||
plug.show_error(tr.tl("Error: Can't connect to EDSM"))
|
||
if entry['event'].lower() in ('shutdown', 'commander', 'fileheader'):
|
||
# Game shutdown or new login, so we MUST not hang on to pending
|
||
pending = []
|
||
logger.trace_if(CMDR_EVENTS, f'Blanked pending because of event: {entry["event"]}')
|
||
if closing:
|
||
logger.debug('closing, so returning.')
|
||
return
|
||
|
||
last_game_version = game_version
|
||
last_game_build = game_build
|
||
|
||
|
||
def should_send(entries: list[Mapping[str, Any]], event: str) -> bool: # noqa: CCR001
|
||
"""
|
||
Whether or not any of the given entries should be sent to EDSM.
|
||
|
||
:param entries: The entries to check
|
||
:param event: The latest event being processed
|
||
:return: bool indicating whether or not to send said entries
|
||
"""
|
||
def should_send_entry(entry: Mapping[str, Any]) -> bool:
|
||
if entry['event'] == 'Cargo':
|
||
return not this.newgame_docked
|
||
if entry['event'] == 'Docked':
|
||
return True
|
||
if this.newgame:
|
||
return True
|
||
if entry['event'] not in (
|
||
'CommunityGoal',
|
||
'ModuleBuy',
|
||
'ModuleSell',
|
||
'ModuleSwap',
|
||
'ShipyardBuy',
|
||
'ShipyardNew',
|
||
'ShipyardSwap'
|
||
):
|
||
return True
|
||
return False
|
||
|
||
if event.lower() in ('shutdown', 'fileheader'):
|
||
logger.trace_if(CMDR_EVENTS, f'True because {event=}')
|
||
return True
|
||
|
||
if this.navbeaconscan:
|
||
if entries and entries[-1]['event'] == 'Scan':
|
||
this.navbeaconscan -= 1
|
||
should_send_result = this.navbeaconscan == 0
|
||
logger.trace_if(CMDR_EVENTS, f'False because {this.navbeaconscan=}' if not should_send_result else '')
|
||
return should_send_result
|
||
logger.error('Invalid state NavBeaconScan exists, but passed entries either '
|
||
"doesn't exist or doesn't have the expected content")
|
||
this.navbeaconscan = 0
|
||
|
||
should_send_result = any(should_send_entry(entry) for entry in entries)
|
||
logger.trace_if(CMDR_EVENTS, f'False as default: {this.newgame_docked=}' if not should_send_result else '')
|
||
return should_send_result
|
||
|
||
|
||
def update_status(event=None) -> None:
|
||
"""Update listening plugins with our response to StartUp, Location, FSDJump, or CarrierJump."""
|
||
for plugin in plug.provides('edsm_notify_system'):
|
||
plug.invoke(plugin, None, 'edsm_notify_system', this.lastlookup)
|
||
|
||
|
||
# Called with EDSM's response to a 'StartUp', 'Location', 'FSDJump' or 'CarrierJump' event.
|
||
# https://www.edsm.net/en/api-journal-v1
|
||
# msgnum: 1xx = OK, 2xx = fatal error, 3xx = error, 4xx = ignorable errors.
|
||
def edsm_notify_system(reply: Mapping[str, Any]) -> None:
|
||
"""Update the image next to the system link."""
|
||
if this.system_link is not None:
|
||
if not reply:
|
||
this.system_link['image'] = this._IMG_ERROR
|
||
# LANG: EDSM Plugin - Error connecting to EDSM API
|
||
plug.show_error(tr.tl("Error: Can't connect to EDSM"))
|
||
elif reply['msgnum'] // 100 not in (1, 4):
|
||
this.system_link['image'] = this._IMG_ERROR
|
||
# LANG: EDSM Plugin - Error message from EDSM API
|
||
plug.show_error(tr.tl('Error: EDSM {MSG}').format(MSG=reply['msg']))
|
||
elif reply.get('systemCreated'):
|
||
this.system_link['image'] = this._IMG_NEW
|
||
else:
|
||
this.system_link['image'] = this._IMG_KNOWN
|