import { useCallback } from 'react';
import { notifications } from '@mantine/notifications';
import { pullChanges, commitAndPush } from '../api/git';
import { useWorkspaceData } from '../contexts/WorkspaceDataContext';
import type { CommitHash } from '@/types/models';

interface UseGitOperationsResult {
  handlePull: () => Promise<boolean>;
  handleCommitAndPush: (message: string) => Promise<void>;
}

export const useGitOperations = (): UseGitOperationsResult => {
  const { currentWorkspace } = useWorkspaceData();

  const handlePull = useCallback(async (): Promise<boolean> => {
    if (
      !currentWorkspace ||
      !currentWorkspace.gitEnabled ||
      !currentWorkspace.name
    )
      return false;

    try {
      const message = await pullChanges(currentWorkspace.name);
      notifications.show({
        title: 'Success',
        message: message || 'Successfully pulled latest changes',
        color: 'green',
      });
      return true;
    } catch (error) {
      console.error('Failed to pull latest changes:', error);
      notifications.show({
        title: 'Error',
        message: 'Failed to pull latest changes',
        color: 'red',
      });
      return false;
    }
  }, [currentWorkspace]);

  const handleCommitAndPush = useCallback(
    async (message: string): Promise<void> => {
      if (!currentWorkspace || !currentWorkspace.gitEnabled) return;

      try {
        const commitHash: CommitHash = await commitAndPush(
          currentWorkspace.name,
          message
        );
        notifications.show({
          title: 'Success',
          message: 'Successfully committed and pushed changes ' + commitHash,
          color: 'green',
        });
        return;
      } catch (error) {
        console.error('Failed to commit and push changes:', error);
        notifications.show({
          title: 'Error',
          message: 'Failed to commit and push changes',
          color: 'red',
        });
        return;
      }
    },
    [currentWorkspace]
  );

  return { handlePull, handleCommitAndPush };
};