config/deployed-update.js

Rama main · réplica de lectura del repositorio original.

const { execSync } = require('child_process');
const { isDockerRunning, silentExit } = require('./helpers');

async function validateDockerRunning() {
  if (!isDockerRunning()) {
    console.red(
      'Error: Docker is not running. You will need to start Docker Desktop or if using linux/mac, run `sudo systemctl start docker`',
    );
    silentExit(1);
  }
}

function getCurrentBranch() {
  return execSync('git rev-parse --abbrev-ref HEAD', { encoding: 'utf8' }).trim();
}

const shouldRebase = process.argv.includes('--rebase');

(async () => {
  console.green(
    'Starting deployed update script, this may take a minute or two depending on your system and network.',
  );

  await validateDockerRunning();
  console.purple('Fetching the latest repo...');
  execSync('git fetch origin', { stdio: 'inherit' });

  if (!shouldRebase) {
    execSync('git checkout main', { stdio: 'inherit' });
    console.purple('Pulling the latest code from main...');
    execSync('git pull origin main', { stdio: 'inherit' });
  } else {
    const currentBranch = getCurrentBranch();
    console.purple(`Rebasing ${currentBranch} onto main...`);
    execSync('git rebase origin/main', { stdio: 'inherit' });
  }

  console.purple('Removing previously made Docker container...');
  const downCommand = 'sudo docker compose -f ./deploy-compose.yml down';
  console.orange(downCommand);
  execSync(downCommand, { stdio: 'inherit' });

  console.purple('Removing all tags for LibreChat `deployed` images...');
  const repositories = ['registry.librechat.ai/librechat-ai/librechat-dev-api', 'librechat-client'];
  repositories.forEach((repo) => {
    const imageRefs = execSync(`sudo docker images ${repo} --format "{{.Repository}}:{{.Tag}}"`, {
      encoding: 'utf8',
    })
      .split('\n')
      .filter(Boolean)
      .filter((ref) => !ref.includes('<none>'));
    imageRefs.forEach((imageRef) => {
      const removeImageCommand = `sudo docker rmi ${imageRef}`;
      console.orange(removeImageCommand);
      execSync(removeImageCommand, { stdio: 'inherit' });
    });
  });

  console.purple('Pulling latest LibreChat images...');
  const pullCommand = 'sudo docker compose -f ./deploy-compose.yml pull api';
  console.orange(pullCommand);
  execSync(pullCommand, { stdio: 'inherit' });

  /* The tag-removal above only covers the stock repositories; any overridden
   * `api` image (or other freshly pulled service) leaves its previous version
   * dangling — ~1.7GB per update that nothing reclaimed. Prune AFTER the pull
   * so the just-superseded layers are already untagged, mirroring update.js. */
  console.purple('Removing all unused dangling Docker images...');
  const pruneCommand = 'sudo docker image prune -f';
  console.orange(pruneCommand);
  execSync(pruneCommand, { stdio: 'inherit' });

  const startCommand = 'sudo docker compose -f ./deploy-compose.yml up -d';
  console.green('Your LibreChat app is now up to date! Start the app with the following command:');
  console.purple(startCommand);
  console.orange(
    "Note: it's also recommended to clear your browser cookies and localStorage for LibreChat to assure a fully clean installation.",
  );
  console.orange("Also: Don't worry, your data is safe :)");
})().catch((err) => {
  console.error('Update script failed:', err.message);
  process.exit(1);
});