import { Command } from "commander";
import { OllamaAction } from "./ollama";

export function initializeUpdateCommands(program: Command) {
  const updateCommand = program
    .command("update")
    .description("Update resources like models or configurations");

  updateCommand
    .command("ollama")
    .description("Manage Ollama models (update or remove)")
    .option("--model [model-name]", "Specify the model to update or remove", '')
    .option("--remove", "Remove the specified model instead of updating")
    .action(async (options) => {
      const ollamaAction = new OllamaAction();

      if (options.remove) {
        await ollamaAction.removeModel(options.model);
      } else {
        await ollamaAction.updateModel(options.model);
      }
    });

  return program;
}