linkgravity 1.0.0 → 1.0.2

package/bin/cli.js

@@ -176,7 +176,12 @@ if (cmd === 'start') {
             i++;
         } else if (args[i] === '-f') {
             isFollow = true;
-        } else if (args[i] === '-t' || args[i] === '--stamp' || args[i] === '--timestamp' || args[i] === '--timestamps') {
+        } else if (
+            args[i] === '-t' ||
+            args[i] === '--stamp' ||
+            args[i] === '--timestamp' ||
+            args[i] === '--timestamps'
+        ) {
             showStamps = true;
         } else {
             pm2Args.push(args[i]);

package/bin/setup.js

@@ -3,6 +3,7 @@ const fs = require('fs');
 const path = require('path');
 const os = require('os');
 const { spawnSync } = require('child_process');
+const { python: pythonExe } = require('../npm-scripts/venv-paths');
 
 const color = {
     reset: '\x1b[0m',
@@ -70,19 +71,37 @@ async function collectSessionScopes(existingScopes) {
         }
         isFirst = false;
 
-        const channelIds = await p.text({
-            message:
-                'Restrict to specific channel ID(s) in this server? Right-click a CHANNEL → Copy Channel ID. ' +
-                'Comma-separated, or leave empty to allow the WHOLE server:',
-        });
-        if (p.isCancel(channelIds)) {
-            p.cancel('Setup cancelled.');
-            process.exit(0);
+        const channelIds = [];
+        let isFirstChannel = true;
+        while (true) {
+            const channelId = await p.text({
+                message: isFirstChannel
+                    ? 'Restrict to a specific channel in this server? Right-click a CHANNEL → Copy Channel ID. ' +
+                      '(leave empty to allow the WHOLE server):'
+                    : 'Another channel ID to restrict to (leave empty if done):',
+            });
+            if (p.isCancel(channelId)) {
+                p.cancel('Setup cancelled.');
+                process.exit(0);
+            }
+            if (!channelId) break;
+            isFirstChannel = false;
+            channelIds.push(channelId.trim());
+
+            const addAnotherChannel = await p.confirm({
+                message: 'Add another channel?',
+                initialValue: false,
+            });
+            if (p.isCancel(addAnotherChannel)) {
+                p.cancel('Setup cancelled.');
+                process.exit(0);
+            }
+            if (!addAnotherChannel) break;
         }
 
         scopes.push({
             guild_id: guildId.trim(),
-            channel_ids: channelIds ? splitIds(channelIds) : [],
+            channel_ids: channelIds,
         });
 
         const addAnother = await p.confirm({ message: 'Add another server?', initialValue: false });
@@ -102,7 +121,7 @@ async function collectUserIds(existingIds) {
 
     p.note(
         'ONLY these users can use the bot (leave completely empty on first setup to allow EVERYONE). ' +
-            "Not related to DMs - this only gates the channel/threads configured above.",
+            'Not related to DMs - this only gates the channel/threads configured above.',
         'Allowed Discord Users',
     );
 
@@ -170,9 +189,9 @@ async function runSetup() {
     const userIds = await collectUserIds(existingUserIds);
 
     p.note(
-        'Wake words aren\'t set here anymore - they need a voice recording to register '
-            + '(so only your voice triggers them), which this terminal wizard can\'t do. '
-            + 'Set them from Discord with `/sound wake_words:<word>` once the bot is running.',
+        "Wake words aren't set here anymore - they need a voice recording to register " +
+            "(so only your voice triggers them), which this terminal wizard can't do. " +
+            'Set them from Discord with `/sound wake_words:<word>` once the bot is running.',
         'Wake Words',
     );
 
@@ -249,12 +268,40 @@ async function runSetup() {
     p.note('Configuration saved to lgy.json successfully!', 'Success');
 
     console.log(`${color.cyan}▶${color.reset} Restarting daemon to apply changes...`);
-    spawnSync('npx', ['-y', 'pm2', 'restart', 'lgy', '--update-env'], {
+    const restartResult = spawnSync('npx', ['-y', 'pm2', 'restart', 'lgy', '--update-env'], {
         stdio: 'pipe',
         env: { ...process.env, PYTHONUNBUFFERED: '1' },
     });
 
-    p.outro('Daemon restarted.');
+    if (restartResult.status === 0) {
+        p.outro('Daemon restarted.');
+    } else {
+        const stderr = (restartResult.stderr || '').toString();
+        if (stderr.includes('not found')) {
+            // Nothing to restart yet (first-ever setup, or pm2's process list was
+            // reset e.g. after a reboot without `pm2 save`) - start it instead of
+            // reporting a false "restarted" success.
+            const botPath = path.join(__dirname, '..', 'src', 'main.py');
+            const startResult = spawnSync(
+                'npx',
+                ['-y', 'pm2', 'start', botPath, '--interpreter', pythonExe, '--name', 'lgy'],
+                { stdio: 'pipe', env: { ...process.env, PYTHONUNBUFFERED: '1' } },
+            );
+            if (startResult.status === 0) {
+                p.outro("Daemon wasn't running yet - started it fresh instead.");
+            } else {
+                console.error((startResult.stderr || '').toString().trim());
+                p.outro(
+                    `${color.yellow}⚠${color.reset} Failed to start the daemon - run \`lgy start\` manually to see the full error.`,
+                );
+            }
+        } else {
+            console.error(stderr.trim());
+            p.outro(
+                `${color.yellow}⚠${color.reset} Failed to restart the daemon - run \`lgy restart\` manually to see the full error.`,
+            );
+        }
+    }
 }
 
 module.exports = runSetup;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "linkgravity",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "Discord bot bridge for the Antigravity (agy) CLI, with voice interaction support",
   "scripts": {
     "postinstall": "node npm-scripts/postinstall.js",

package/src/cogs/general_cog.py

@@ -7,7 +7,17 @@ import discord
 from discord import app_commands
 from discord.ext import commands
 
-from config import DATA_DIR, EMBED_COLOR, MODEL_CHOICES, allowed, is_allowed_session_channel, logger, session_manager
+from config import (
+    DATA_DIR,
+    EMBED_COLOR,
+    MODEL_CHOICES,
+    allowed,
+    bot_settings,
+    is_allowed_session_channel,
+    logger,
+    save_bot_settings,
+    session_manager,
+)
 from core.atomic_io import atomic_write_json, safe_load_json
 from utils.utils import get_default_cwd
 
@@ -116,7 +126,7 @@ class GeneralCog(commands.Cog):
         from utils.utils import get_current_model
 
         session = session_manager.get_session(str(interaction.channel_id)) or {}
-        current_model = session.get("model") or get_current_model()
+        current_model = session.get("model") or bot_settings.get("default_model") or get_current_model()
 
         choices = []
         if current_model:
@@ -151,6 +161,8 @@ class GeneralCog(commands.Cog):
             if not model:
                 await interaction.response.send_message(f"⚠️ Model matching `{model}` not found.", ephemeral=True)
                 return
+        else:
+            model = bot_settings.get("default_model") or None
 
         await interaction.response.defer(ephemeral=True)
         import uuid
@@ -197,8 +209,16 @@ class GeneralCog(commands.Cog):
             partial = next((m for m in cached_models if model.lower() in m.lower()), None)
             final_model = exact or partial or model
             session_manager.update_session(str(interaction.channel_id), "model", final_model)
+
+            bot_settings["default_model"] = final_model
+            save_bot_settings(bot_settings)
+
             await interaction.response.send_message(
-                embed=discord.Embed(description=f"🤖 Model changed: **{final_model}**", color=EMBED_COLOR)
+                embed=discord.Embed(
+                    description=f"🤖 Model changed: **{final_model}**\n"
+                    f"💾 Also set as the default for new `/new` sessions.",
+                    color=EMBED_COLOR,
+                )
             )
         except discord.Forbidden:
             logger.error(f"Missing permissions to start thread in channel {interaction.channel_id}")

package/src/cogs/voice/enrollment.py

@@ -17,7 +17,6 @@ NODE_REQUEST_TIMEOUT = aiohttp.ClientTimeout(total=5)
 
 
 class RecordingPromptView(discord.ui.View):
-
     def __init__(self, manager: "EnrollmentManager", user_id: str):
         super().__init__(timeout=120)
         self.manager = manager
@@ -48,7 +47,6 @@ class RecordingPromptView(discord.ui.View):
 
 
 class SampleConfirmView(discord.ui.View):
-
     def __init__(self, manager: "EnrollmentManager", user_id: str):
         super().__init__(timeout=60)
         self.manager = manager

package/src/cogs/voice_cog.py

@@ -6,7 +6,7 @@ import discord
 from discord import app_commands
 from discord.ext import commands, tasks
 
-from config import logger
+from config import allowed, logger
 
 from .voice.enrollment import EnrollmentManager
 from .voice.stt_session import SttSessionTracker
@@ -152,6 +152,10 @@ class VoiceCog(commands.Cog):
 
     @app_commands.command(name="join", description="Summon the bot to your current voice channel")
     async def cmd_join(self, interaction: discord.Interaction):
+        if not allowed(interaction.user.id):
+            await interaction.response.send_message("❌ Permission Denied", ephemeral=True)
+            return
+
         if not isinstance(interaction.channel, discord.Thread):
             await interaction.response.send_message(
                 "❌ This command can only be used inside a thread created with `/new`.", ephemeral=True
@@ -260,6 +264,10 @@ class VoiceCog(commands.Cog):
     ):
         import aiohttp
 
+        if not allowed(interaction.user.id):
+            await interaction.response.send_message("❌ Permission Denied", ephemeral=True)
+            return
+
         if (
             wake_word is None
             and active_times is None
@@ -330,6 +338,10 @@ class VoiceCog(commands.Cog):
 
     @app_commands.command(name="leave", description="Make the bot leave the voice channel")
     async def cmd_leave(self, interaction: discord.Interaction):
+        if not allowed(interaction.user.id):
+            await interaction.response.send_message("❌ Permission Denied", ephemeral=True)
+            return
+
         guild_id = interaction.guild_id
         await interaction.response.send_message("👋 Disconnected from voice channel.")
         try:
@@ -365,6 +377,12 @@ class VoiceCog(commands.Cog):
         try:
             guild_id = data.get("guild_id")
             user_id = data.get("user_id")
+
+            if not allowed(int(user_id)):
+                self.logger.debug(f"STT: ignoring speech from non-allowed user {user_id}")
+                await self.stt_session.clear_partial_msg(str(guild_id))
+                return
+
             # Node runs STT itself before calling this endpoint, so this is
             # already-recognized text, not raw audio - keeps STT off every VAD hit.
             text = data.get("text")

package/src/config.py

@@ -19,6 +19,9 @@ DEFAULT_LGY_CONFIG = {
     "voice_threshold": 3000,
     "tts_voice": "ko-KR-SunHiNeural",
     "tts_enabled": True,
+    # Sticky default for new /new sessions - set whenever /model succeeds,
+    # so a new thread doesn't fall back to agy's own settings.json model.
+    "default_model": "",
 }
 
 

package/src/core/agy_runner.py

@@ -126,9 +126,7 @@ async def run_agy(
                 libstdbuf_path = _find_libstdbuf()  # works around agy's output-truncation bug
                 if libstdbuf_path:
                     existing_preload = env.get("LD_PRELOAD", "")
-                    env["LD_PRELOAD"] = (
-                        f"{libstdbuf_path}:{existing_preload}" if existing_preload else libstdbuf_path
-                    )
+                    env["LD_PRELOAD"] = f"{libstdbuf_path}:{existing_preload}" if existing_preload else libstdbuf_path
                     env["_STDBUF_O"] = str(_STDOUT_BUFFER_SIZE)
 
                 kwargs = {

package/src/handlers/thread_reply.py

@@ -119,8 +119,9 @@ async def handle_thread_reply(bot, message: discord.Message):
     adapter = get_adapter()
     content = message.content.strip()
     if content.startswith("/new"):
-        await adapter.send_message(thread, 
-            "To start a new session, please use the `/new` slash command (not typed as text) - it works from inside a thread too and will open the new one in the right place."
+        await adapter.send_message(
+            thread,
+            "To start a new session, please use the `/new` slash command (not typed as text) - it works from inside a thread too and will open the new one in the right place.",
         )
         return
 

package/src/messengers/discord_adapter.py

@@ -187,6 +187,7 @@ class DiscordAdapter(MessengerAdapter):
             submit_btn.callback = submit_callback
             view.add_item(submit_btn)
         else:
+
             def make_option_callback(opt_text: str):
                 async def callback(interaction: discord.Interaction):
                     await _resolve(interaction, opt_text, "Selected")
@@ -200,6 +201,7 @@ class DiscordAdapter(MessengerAdapter):
                 view.add_item(btn)
 
         if allow_write_in:
+
             class WriteInModal(discord.ui.Modal, title="Write in"):
                 answer = discord.ui.TextInput(
                     label="Enter your response",