matrix_appservice_kakaotalk/commands/auth.py

@@ -69,30 +69,10 @@ async def login(evt: CommandEvent) -> None:
         await evt.reply("You're already logged in")
         return
 
-    num_args = len(evt.args)
-    save = num_args > 0 and evt.args[0] == "--save"
-    # TODO Once web login is implemented, don't make <email> a mandatory argument
-    if not save and num_args != 1:
-        await evt.reply("**Usage:** `$cmdprefix+sp login [--save] <email>`")
-        return
-    email = evt.args[0 if not save else 1] if num_args > 0 else None
+    save = len(evt.args) > 0 and evt.args[0] == "--save"
+    email = evt.args[0 if not save else 1] if len(evt.args) > 0 else None
 
     if email:
-        try:
-            creds = await LoginCredential.get_by_mxid(evt.sender.mxid)
-        except:
-            evt.log.exception("Exception while looking for saved password")
-            creds = None
-        if creds and creds.email == email:
-            await evt.reply("Logging in with saved password")
-            evt.sender.command_status = {
-                "action": "Login with saved password",
-                "room_id": evt.room_id,
-                "save": True,
-            }
-            await _login_with_password(evt, email, creds.password, evt.sender.force_login)
-            return
-
         evt.sender.command_status = {
             "action": "Login",
             "room_id": evt.room_id,
@@ -101,6 +81,20 @@ async def login(evt: CommandEvent) -> None:
             "save": save,
             "forced": evt.sender.force_login,
         }
+        try:
+            creds = await LoginCredential.get_by_mxid(evt.sender.mxid)
+        except:
+            evt.log.exception("Exception while looking for saved password")
+            creds = None
+        if creds and creds.email == email:
+            await evt.reply("Logging in with saved password")
+            await _login_with_password(
+                evt,
+                evt.sender.command_status.pop("email"),
+                creds.password,
+                evt.sender.command_status.pop("forced"),
+            )
+            return
 
     """ TODO Implement web login
     if evt.bridge.public_website:

matrix_appservice_kakaotalk/config.py

@@ -117,7 +117,6 @@ class Config(BaseBridgeConfig):
         else:
             copy("rpc.connection.host")
             copy("rpc.connection.port")
-        copy("rpc.logging_keys")
 
     def _get_permissions(self, key: str) -> tuple[bool, bool, bool, str]:
         level = self["bridge.permissions"].get(key, "")

matrix_appservice_kakaotalk/example-config.yaml

@@ -252,11 +252,6 @@ rpc:
         # Only for type: tcp
         host: localhost
         port: 29392
-    # Command arguments to print in logs. Optional.
-    # TODO Support nested arguments, like channel_props.ktid
-    logging_keys:
-    - mxid
-    #- channel_props
 
 # Python logging configuration.
 #

matrix_appservice_kakaotalk/kt/client/client.py

@@ -109,10 +109,6 @@ class Client:
             await cls._rpc_client.connect()
             await cls._rpc_client.wait_for_disconnection()
 
-    @classmethod
-    def wait_for_connection(cls) -> Awaitable[None]:
-        return cls._rpc_client.wait_for_connection()
-
     @classmethod
     def stop_cls(cls) -> None:
         """Stop and disconnect from the Node backend."""

matrix_appservice_kakaotalk/portal.py

@@ -1881,7 +1881,7 @@ class Portal(DBPortal, BasePortal):
         if not self.is_direct:
             self._main_intent = self.az.intent
         else:
-            # TODO Save kt_sender in DB instead? Only do that if keeping a unique DM portal for each receiver
+            # TODO Save kt_sender in DB instead? Depends on if DM channels are shared...
             user = await u.User.get_by_ktid(self.kt_receiver)
             assert user, f"Found no user for this portal's receiver of {self.kt_receiver}"
             if self.kt_type == KnownChannelType.MemoChat:
@@ -1920,7 +1920,7 @@ class Portal(DBPortal, BasePortal):
         create: bool = True,
         kt_type: ChannelType | None = None,
     ) -> Portal | None:
-        # TODO Direct chats are shared, so can remove kt_receiver if DM portals should be shared
+        # TODO Find out if direct channels are shared. If so, don't need kt_receiver!
         if kt_type:
             kt_receiver = kt_receiver if KnownChannelType.is_direct(kt_type) else 0
         ktid_full = (ktid, kt_receiver)

matrix_appservice_kakaotalk/rpc/rpc.py

@@ -86,7 +86,6 @@ class RPCClient:
     _is_connected: CancelableEvent
     _is_disconnected: CancelableEvent
     _connection_lock: asyncio.Lock
-    _logging_keys: list[str]
 
     def __init__(self, config: Config, register_config_key: str) -> None:
         self.config = config
@@ -106,7 +105,6 @@ class RPCClient:
         self._is_disconnected = CancelableEvent(self.loop)
         self._is_disconnected.set()
         self._connection_lock = asyncio.Lock()
-        self._logging_keys = config["rpc.logging_keys"]
 
     async def connect(self) -> None:
         async with self._connection_lock:
@@ -149,8 +147,7 @@ class RPCClient:
         self._read_task = self.loop.create_task(self._try_read_loop())
         await self._raw_request("register",
                                 peer_id=self.config["appservice.address"],
-                                register_config=self.config[self.register_config_key],
-                                logging_keys=self._logging_keys)
+                                register_config=self.config[self.register_config_key])
         self._is_connected.set()
         self._is_disconnected.clear()
 
@@ -305,10 +302,7 @@ class RPCClient:
         req_id = self._next_req_id
         future = self._response_waiters[req_id] = self.loop.create_future()
         req = {"id": req_id, "command": command, **data}
-        self.log.debug("Request %d: %s", req_id,
-                       ', '.join(
-                           [command] +
-                           [f"{k}: {data[k]}" for k in self._logging_keys if k in data]))
+        self.log.debug("Request %d: %s", req_id, command)
         assert self._writer is not None
         self._writer.write(json.dumps(req).encode("utf-8"))
         self._writer.write(b"\n")

matrix_appservice_kakaotalk/user.py

@@ -317,8 +317,6 @@ class User(DBUser, BaseUser):
                         oauth_credential = await Client.login(uuid=uuid, form=form, forced=True)
                     except OAuthException as e:
                         latest_exc = e
-                else:
-                    return False
             if oauth_credential:
                 self.oauth_credential = oauth_credential
                 await self.save()
@@ -390,9 +388,6 @@ class User(DBUser, BaseUser):
     ) -> None:
         try:
             if not await self._load_session(is_startup=is_startup) and self.has_state:
-                self.log.debug("reload_session failure: wait for connection to Node module before prompting for manual login")
-                await Client.wait_for_connection()
-                self.log.debug("reload_session failure: now connected to Node module")
                 await self.send_bridge_notice(
                     "Logged out of KakaoTalk. Must use the `login` command to log back in.",
                     important=True,
@@ -434,6 +429,7 @@ class User(DBUser, BaseUser):
 
     async def logout(self, *, remove_ktid: bool = True, reset_device: bool = False) -> None:
         if self._client:
+            # TODO Look for a logout API call
             await self._client.stop()
         if remove_ktid:
             await self.push_bridge_state(BridgeStateEvent.LOGGED_OUT)
@@ -750,10 +746,7 @@ class User(DBUser, BaseUser):
                 reason_suffix = "To reconnect, use the `sync` command."
             else:
                 reason_suffix = "You are now logged out. To log back in, use the `login` command."
-            await self.send_bridge_notice(
-                f"Disconnected from KakaoTalk: {reason_str} {reason_suffix}",
-                important=True,
-            )
+            await self.send_bridge_notice(f"Disconnected from KakaoTalk: {reason_str} {reason_suffix}")
 
     async def on_error(self, error: JSON) -> None:
         await self.send_bridge_notice(

node/README.md

@@ -2,6 +2,3 @@
 If `type` is `unix`, `path` is the path where to create the socket, and `force` is whether to overwrite the socket file if it already exists.
 
 If `type` is `tcp`, `port` and `host` are the host/port where to listen.
-
-### Register timeout
-`register_timeout` is the amount of time (in milliseconds) that a connecting peer must send a "register" command after initiating a connection.

node/example-config.json

@@ -3,6 +3,5 @@
 		"type": "unix",
 		"path": "/var/run/matrix-appservice-kakaotalk/rpc.sock",
 		"force": false
-	},
-	"register_timeout": 3000
+	}
 }

node/src/client.js

@@ -422,10 +422,9 @@ export default class PeerClient {
 		this.stopped = false
 		this.notificationID = 0
 		this.maxCommandID = 0
-		this.peerID = ""
+		this.peerID = null
 		this.deviceName = "KakaoTalk Bridge"
-		/** @type {[string]} */
-		this.loggingKeys = []
+
 		/** @type {Map<string, UserClient>} */
 		this.userClients = new Map()
 	}
@@ -456,10 +455,10 @@ export default class PeerClient {
 
 		setTimeout(() => {
 			if (!this.peerID && !this.stopped) {
-				this.log(`Didn't receive register request within ${this.manager.registerTimeout/1000} seconds, terminating`)
+				this.log("Didn't receive register request within 3 seconds, terminating")
 				this.stop("Register request timeout")
 			}
-		}, this.manager.registerTimeout)
+		}, 3000)
 	}
 
 	async stop(error = null) {
@@ -483,11 +482,11 @@ export default class PeerClient {
 		if (this.peerID && this.manager.clients.get(this.peerID) === this) {
 			this.manager.clients.delete(this.peerID)
 		}
-		this.log(`Connection closed (peer: ${this.peerID || "unknown peer"})`)
+		this.log(`Connection closed (peer: ${this.peerID})`)
 	}
 
 	#closeUsers() {
-		this.log(`Closing all API clients for ${this.peerID || "unknown peer"}`)
+		this.log("Closing all API clients for", this.peerID)
 		for (const userClient of this.userClients.values()) {
 			userClient.disconnect()
 		}
@@ -527,6 +526,7 @@ export default class PeerClient {
 	 * request failed, its status is stored here.
 	 */
 	handleLogin = async (req) => {
+		// TODO Look for a logout API call
 		const authClient = await this.#createAuthClient(req.uuid)
 		const loginRes = await authClient.login(req.form, req.forced)
 		if (loginRes.status === KnownAuthStatusCode.DEVICE_NOT_REGISTERED) {
@@ -1166,12 +1166,10 @@ export default class PeerClient {
 	 * @param {string} req.peer_id
 	 * @param {Object} req.register_config
 	 * @param {string} req.register_config.device_name
-	 * @param {?[string]} req.logging_keys
 	 */
 	handleRegister = async (req) => {
 		this.peerID = req.peer_id
 		this.deviceName = req.register_config.device_name || this.deviceName
-		this.loggingKeys = req.logging_keys || this.loggingKeys
 		this.log(`Registered socket ${this.connID} -> ${this.peerID}`)
 		if (this.manager.clients.has(this.peerID)) {
 			const oldClient = this.manager.clients.get(this.peerID)
@@ -1202,12 +1200,7 @@ export default class PeerClient {
 			this.log("Ignoring old request", req.id)
 			return
 		}
-		this.log(
-			`Request ${req.id}:`,
-			[req.command].concat(
-				this.loggingKeys.filter(k => k in req).map(k => `${k}: ${JSON.stringify(req[k], this.#writeReplacer)}`))
-			.join(', ')
-		)
+		this.log("Received request", req.id, "with command", req.command)
 		this.maxCommandID = req.id
 		let handler
 		if (!this.peerID) {

node/src/clientmanager.js

@@ -22,9 +22,8 @@ import { promisify } from "./util.js"
 
 
 export default class ClientManager {
-	constructor(listenConfig, registerTimeout) {
+	constructor(listenConfig) {
 		this.listenConfig = listenConfig
-		this.registerTimeout = registerTimeout
 		this.server = net.createServer(this.acceptConnection)
 		this.connections = []
 		this.clients = new Map()

node/src/main.js

@@ -32,7 +32,7 @@ const configPath = args["--config"] || "config.json"
 console.log("[Main] Reading config from", configPath)
 const config = JSON.parse(fs.readFileSync(configPath).toString())
 
-const manager = new ClientManager(config.listen, config.register_timeout)
+const manager = new ClientManager(config.listen)
 
 function stop() {
 	manager.stop().then(() => {