setup-branch.ts

import { ContentstackClient, getBranchFromAlias, log } from '@contentstack/cli-utilities';
import { ImportConfig } from 'src/types';
import { validateBranch } from './common-helper';

export const setupBranchConfig = async (
  config: ImportConfig,
  stackAPIClient: ReturnType<ContentstackClient['stack']>,
): Promise<void> => {

  if (config.branchName) {
    await validateBranch(stackAPIClient, config, config.branchName);
    return;
  }

  if (config.branchAlias) {
    config.branchName = await getBranchFromAlias(stackAPIClient, config.branchAlias);
    return;
  }

    try {
      const branches = await stackAPIClient
        .branch()
        .query()
        .find()
        .then(({ items }) => items);
      if (branches.length) {
        log.info(`The stack is branch-enabled, and branches exist. By default, content will be imported into the main branch.`);

        config.branchName = 'main';
        log.debug(`Setting default target branch to 'main'`);
      }
    } catch (error) {
      log.debug('Failed to fetch branches', { error });
    }
};