/*
 * Copyright (c) 2025 Bytedance, Inc. and its affiliates.
 * SPDX-License-Identifier: Apache-2.0
 */
import { describe, it, expect } from 'vitest';
import { parseActionVlm } from '../src/actionParser';

describe('parseActionVlm', () => {
  // BC mode tests
  describe('BC mode', () => {
    it('should correctly parse input with Thought', () => {
      const input = `Thought: I need to click this button
Action: click(start_box='(100,200)')`;

      const result = parseActionVlm(input);

      expect(result).toEqual([
        {
          reflection: null,
          thought: 'I need to click this button',
          action_type: 'click',
          action_inputs: {
            start_box: '[0.1,0.2,0.1,0.2]',
          },
        },
      ]);
    });

    it('should correctly parse input with Thought custom factors', () => {
      const input = `Thought: I need to click this button
Action: click(start_box='(100,200)')`;

      const result = parseActionVlm(input, [1366, 768]);

      expect(result).toEqual([
        {
          reflection: null,
          thought: 'I need to click this button',
          action_type: 'click',
          action_inputs: {
            start_box:
              '[0.07320644216691069,0.2604166666666667,0.07320644216691069,0.2604166666666667]',
          },
        },
      ]);
    });

    it('should correctly parse input with Reflection and Action_Summary', () => {
      const input = `Reflection: This is a reflection
Action_Summary: This is a summary
Action: type(text='Hello', start_box='(300,400)')`;

      const result = parseActionVlm(input);

      expect(result).toEqual([
        {
          reflection: 'This is a reflection',
          thought: 'This is a summary',
          action_type: 'type',
          action_inputs: {
            text: 'Hello',
            start_box: '[0.3,0.4,0.3,0.4]',
          },
        },
      ]);
    });

    it('should handle multiple actions', () => {
      const input = `Thought: Perform multiple actions
Action: click(start_box='(100,200)')

type(text='Hello', start_box='(300,400)')`;

      const result = parseActionVlm(input);

      expect(result).toEqual([
        {
          thought: 'Perform multiple actions',
          reflection: null,
          action_type: 'click',
          action_inputs: {
            start_box: '[0.1,0.2,0.1,0.2]',
          },
        },
        {
          thought: 'Perform multiple actions',
          reflection: null,
          action_type: 'type',
          action_inputs: {
            text: 'Hello',
            start_box: '[0.3,0.4,0.3,0.4]',
          },
        },
      ]);
    });
  });

  // O1 mode tests
  describe('O1 mode', () => {
    it('should correctly parse O1 format input', () => {
      const input = `<Thought>I need to perform this action</Thought>
Action_Summary: Click and type text
Action: click(start_box='(100,200)')
</Output>`;

      const result = parseActionVlm(input, [1000, 1000], 'o1');

      expect(result).toEqual([
        {
          reflection: null,
          thought: 'I need to perform this action\n<Action_Summary>\nClick and type text',
          action_type: 'click',
          action_inputs: {
            start_box: '[0.1,0.2,0.1,0.2]',
          },
        },
      ]);
    });

    it('should handle complex O1 format input', () => {
      const input = `<Thought>Complex operation</Thought>
Action_Summary: Multiple sequential actions
Action: click(start_box='(100,200)')
</Output>`;

      const result = parseActionVlm(input, [1000, 1000], 'o1');

      expect(result).toEqual([
        {
          reflection: null,
          thought: 'Complex operation\n<Action_Summary>\nMultiple sequential actions',
          action_type: 'click',
          action_inputs: {
            start_box: '[0.1,0.2,0.1,0.2]',
          },
        },
      ]);
    });
  });

  // bc mode but new format output tests
  describe('bc mode but hallucination', () => {
    it('should correctly parse new format input', () => {
      const input = `Thought: 我看到当前屏幕显示的是一个电子表格软件和一个聊天窗口，而任务要求我需要在浏览器中搜索北京明天天气。我需要先点击任务栏上的浏览器图标来启动浏览器。
Action: click(start_box='<bbox>637 964 637 964</bbox>')`;

      const result = parseActionVlm(input, [1000, 1000], 'bc');

      expect(result).toEqual([
        {
          reflection: null,
          thought:
            '我看到当前屏幕显示的是一个电子表格软件和一个聊天窗口，而任务要求我需要在浏览器中搜索北京明天天气。我需要先点击任务栏上的浏览器图标来启动浏览器。',
          action_type: 'click',
          action_inputs: {
            start_box: '[0.637,0.964,0.637,0.964]',
          },
        },
      ]);
    });

    it('should correctly parse new format input', () => {
      const input = `Thought: 我看到当前屏幕显示的是一个电子表格软件和一个聊天窗口，而任务要求我需要在浏览器中搜索北京明天天气。我需要先点击任务栏上的浏览器图标来启动浏览器。
Action: click(start_box='<bbox>637 964 637 964</bbox>')`;

      const result = parseActionVlm(input, [1000, 1000], 'bc', {
        width: 2560,
        height: 1440,
      });

      expect(result).toEqual([
        {
          reflection: null,
          thought:
            '我看到当前屏幕显示的是一个电子表格软件和一个聊天窗口，而任务要求我需要在浏览器中搜索北京明天天气。我需要先点击任务栏上的浏览器图标来启动浏览器。',
          action_type: 'click',
          action_inputs: {
            start_box: '[0.637,0.964,0.637,0.964]',
            start_coords: [1630.72, 1388.16],
          },
        },
      ]);
    });

    it('should correctly parse new format input', () => {
      const input = `<think>
Okay, let's see. The user wants to search news, but the current search bar has "hi" in it. First, I need to clear that. The search bar has an 'X' button on the right, so clicking that will remove the existing text. Then, I can type "news" into the search bar. Wait, but maybe after clearing, I should enter the new search term. Let's check the steps again. Clear the search bar, type "news", then press enter or click the search icon. That should initiate a news search. Let's do that step by step.
</think>
Thought: To search for news, first clear the existing text "hi" from the Google search bar by clicking the 'X' icon. Then, type "news" into the search bar and press Enter to perform the search. This will display news-related results.
Action: click(point='<point>510 150</point>')`;

      const result = parseActionVlm(input, [1000, 1000], 'bc', {
        width: 2560,
        height: 1440,
      });

      expect(result).toEqual([
        {
          reflection: null,
          thought:
            'To search for news, first clear the existing text "hi" from the Google search bar by clicking the \'X\' icon. Then, type "news" into the search bar and press Enter to perform the search. This will display news-related results.',
          action_type: 'click',
          action_inputs: {
            start_box: '[0.51,0.15,0.51,0.15]',
            start_coords: [1305.6, 216],
          },
        },
      ]);
    });

    it('should correctly parse input with Reflection and Action_Summary', () => {
      const input = `Reflection: This is a reflection
Action_Summary: This is a summary
Action: type(text='Hello', start_box='<bbox>300 400</bbox>')`;

      const result = parseActionVlm(input);

      expect(result).toEqual([
        {
          reflection: 'This is a reflection',
          thought: 'This is a summary',
          action_type: 'type',
          action_inputs: {
            text: 'Hello',
            start_box: '[0.3,0.4,0.3,0.4]',
          },
        },
      ]);
    });

    it('should handle multiple actions', () => {
      const input = `Thought: Perform multiple actions
Action: click(start_box='<bbox>100 200</bbox>')

type(text='Hello', start_box='<bbox>300 400</bbox>')`;

      const result = parseActionVlm(input);

      expect(result).toEqual([
        {
          thought: 'Perform multiple actions',
          reflection: null,
          action_type: 'click',
          action_inputs: {
            start_box: '[0.1,0.2,0.1,0.2]',
          },
        },
        {
          thought: 'Perform multiple actions',
          reflection: null,
          action_type: 'type',
          action_inputs: {
            text: 'Hello',
            start_box: '[0.3,0.4,0.3,0.4]',
          },
        },
      ]);
    });

    it('should correctly parse new format input', () => {
      const input = `Thought: 我看到当前屏幕显示的是一个电子表格软件和一个聊天窗口，而任务要求我需要在浏览器中搜索北京明天天气。我需要先点击任务栏上的浏览器图标来启动浏览器。
Action: click(start_box='[637,964,637,964]')`;

      const result = parseActionVlm(input, [1000, 1000], 'bc');

      expect(result).toEqual([
        {
          reflection: null,
          thought:
            '我看到当前屏幕显示的是一个电子表格软件和一个聊天窗口，而任务要求我需要在浏览器中搜索北京明天天气。我需要先点击任务栏上的浏览器图标来启动浏览器。',
          action_type: 'click',
          action_inputs: {
            start_box: '[0.637,0.964,0.637,0.964]',
          },
        },
      ]);
    });

    it('should correctly parse new format input', () => {
      const input = `Thought: 我看到当前屏幕显示的是一个电子表格软件和一个聊天窗口，而任务要求我需要在浏览器中搜索北京明天天气。我需要先点击任务栏上的浏览器图标来启动浏览器。
Action: click(start_box='[637,964,637,964]')`;

      const result = parseActionVlm(input, [1000, 1000], 'bc', {
        width: 2560,
        height: 1440,
      });

      expect(result).toEqual([
        {
          reflection: null,
          thought:
            '我看到当前屏幕显示的是一个电子表格软件和一个聊天窗口，而任务要求我需要在浏览器中搜索北京明天天气。我需要先点击任务栏上的浏览器图标来启动浏览器。',
          action_type: 'click',
          action_inputs: {
            start_box: '[0.637,0.964,0.637,0.964]',
            start_coords: [1630.72, 1388.16],
          },
        },
      ]);
    });
  });

  describe('Box coordinates normalization', () => {
    it('should correctly normalize box with four coordinates', () => {
      const input = `Thought: I need to click on this element
Action: click(start_box='[130,226]')`;

      const result = parseActionVlm(input, [1000, 1000], 'bc', {
        width: 2560,
        height: 1440,
      });

      expect(result).toEqual([
        {
          action_inputs: {
            start_box: '[0.13,0.226,0.13,0.226]',
            start_coords: [332.8, 325.44],
          },
          action_type: 'click',
          reflection: null,
          thought: 'I need to click on this element',
        },
      ]);
    });

    it('should correctly normalize box with end_box coordinates', () => {
      const input = `Thought: I need to click on this element
Action: click(end_box='[130,226]')`;

      const result = parseActionVlm(input, [1000, 1000], 'bc', {
        width: 2560,
        height: 1440,
      });

      expect(result).toEqual([
        {
          action_inputs: {
            end_box: '[0.13,0.226,0.13,0.226]',
            end_coords: [332.8, 325.44],
          },
          action_type: 'click',
          reflection: null,
          thought: 'I need to click on this element',
        },
      ]);
    });

    it('should correctly normalize box with start_box and end_box coordinates', () => {
      const input = `Thought: I need to click on this element
Action: drag(start_box='[130,226]', end_box='[200,226]')`;

      const result = parseActionVlm(input, [1000, 1000], 'bc', {
        width: 2560,
        height: 1440,
      });

      expect(result).toEqual([
        {
          action_inputs: {
            start_box: '[0.13,0.226,0.13,0.226]',
            start_coords: [332.8, 325.44],
            end_box: '[0.2,0.226,0.2,0.226]',
            end_coords: [512, 325.44],
          },
          action_type: 'drag',
          reflection: null,
          thought: 'I need to click on this element',
        },
      ]);
    });

    it('should not normalize box with four coordinates', () => {
      expect(
        parseActionVlm(
          `Thought: I need to click on this element
Action: click(start_box='[]')`,
          [1000, 1000],
          'bc',
          {
            width: 2560,
            height: 1440,
          },
        ),
      ).toEqual([
        {
          action_inputs: {
            start_box: '[]',
            start_coords: [],
          },
          action_type: 'click',
          reflection: null,
          thought: 'I need to click on this element',
        },
      ]);

      expect(
        parseActionVlm(
          `Thought: I need to click on this element
Action: click(start_box='')`,
          [1000, 1000],
          'bc',
          {
            width: 2560,
            height: 1440,
          },
        ),
      ).toEqual([
        {
          action_inputs: {},
          action_type: 'click',
          reflection: null,
          thought: 'I need to click on this element',
        },
      ]);
    });

    it('should correctly normalize box with four coordinates with scale factor', () => {
      const input = `Thought: I need to click on this element
Action: click(start_box='[130,226]')`;

      const result = parseActionVlm(
        input,
        [1000, 1000],
        'bc',
        {
          width: 1280,
          height: 720,
        },
        2,
      );

      expect(result).toEqual([
        {
          action_inputs: {
            start_box: '[0.13,0.226,0.13,0.226]',
            start_coords: [332.8, 325.44],
          },
          action_type: 'click',
          reflection: null,
          thought: 'I need to click on this element',
        },
      ]);
    });
  });

  // Edge cases
  describe('Edge cases', () => {
    it('should handle input without Action keyword', () => {
      const input = 'click(start_box="(100,200)")';
      const result = parseActionVlm(input);

      expect(result).toEqual([
        {
          action_inputs: {
            start_box: '[0.1]',
          },
          action_type: 'click',
          reflection: null,
          thought: '',
        },
      ]);
    });

    it('should handle with Chinese colon', () => {
      const input = `Thought: I need to click this button
Action：click(start_box='(100,200)')`;

      const result = parseActionVlm(input, [1366, 768]);

      expect(result).toEqual([
        {
          reflection: null,
          thought: 'I need to click this button',
          action_type: 'click',
          action_inputs: {
            start_box:
              '[0.07320644216691069,0.2604166666666667,0.07320644216691069,0.2604166666666667]',
          },
        },
      ]);
    });

    it('should handle with Chinese colon in the middle of the text', () => {
      const input = `\n\nAction：click(start_box='<bbox>191 21 191 21</bbox>')`;

      const result = parseActionVlm(input, [1366, 768]);

      expect(result).toEqual([
        {
          reflection: null,
          thought: '',
          action_type: 'click',
          action_inputs: {
            start_box: '[0.1398243045387994,0.02734375,0.1398243045387994,0.02734375]',
          },
        },
      ]);
    });

    it('should handle empty action input', () => {
      const input = 'Thought: Empty action\nAction:';
      const result = parseActionVlm(input);

      expect(result).toEqual([
        {
          action_inputs: {},
          action_type: '',
          reflection: null,
          thought: 'Empty action',
        },
      ]);
    });
  });

  describe('Box coordinates normalization', () => {
    it('should correctly normalize box with four coordinates using custom factors', () => {
      const input = `Thought: I need to click on this element
Action: click(start_box='[348, 333, 928, 365]')`;

      const result = parseActionVlm(input, [1366, 768]);

      expect(result).toEqual([
        {
          reflection: null,
          thought: 'I need to click on this element',
          action_type: 'click',
          action_inputs: {
            // Verify that x1, y1, x2, y2 are all normalized correctly
            // x1 = 348/1366, y1 = 333/768, x2 = 928/1366, y2 = 365/768
            start_box: '[0.2547584187408492,0.43359375,0.6793557833089312,0.4752604166666667]',
          },
        },
      ]);
    });

    it('should handle real-world screen dimensions with four coordinates', () => {
      const input = `Thought: I need to click on this element in the browser
Action: click(start_box='[287, 111, 313, 124]')`;

      const result = parseActionVlm(input, [1280, 800]);

      expect(result).toEqual([
        {
          reflection: null,
          thought: 'I need to click on this element in the browser',
          action_type: 'click',
          action_inputs: {
            // Verify the normalized results at the actual screen size
            // x1 = 287/1280, y1 = 111/800, x2 = 313/1280, y2 = 124/800
            start_box: '[0.22421875,0.13875,0.24453125,0.155]',
          },
        },
      ]);
    });

    it('should handle zero coordinates correctly', () => {
      const input = `Thought: I need to click on the start button
Action: click(start_box='[0, 964, 10, 984]')`;

      const result = parseActionVlm(input, [1000, 1000]);

      expect(result).toEqual([
        {
          reflection: null,
          thought: 'I need to click on the start button',
          action_type: 'click',
          action_inputs: {
            start_box: '[0,0.964,0.01,0.984]',
          },
        },
      ]);
    });

    it('should handle zero coordinates correctly, v2', () => {
      const input = `Thought: I need to click on the start button
Action: click(start_box='[0, 964, 10, 984]')`;

      const result = parseActionVlm(input, [1000, 1000], 'bc', {
        width: 2560,
        height: 1440,
      });

      expect(result).toEqual([
        {
          reflection: null,
          thought: 'I need to click on the start button',
          action_type: 'click',
          action_inputs: {
            start_box: '[0,0.964,0.01,0.984]',
            start_coords: [12.8, 1402.56],
          },
        },
      ]);
    });
  });

  describe('Box coordinates normalization', () => {
    it('should parse omni format(1)', () => {
      const input = `<think_never_used_51bce0c785ca2f68081bfa7d91973934>Hmm, the previous action didn't work. Let's check again. The Google search page is open, and the search box is where we need to input. Wait, maybe the search box wasn't properly focused? Let me click on the search box first to make sure it's active, then type "UI TARS". That should work. So first, click the search input field, then type the query.</think_never_used_51bce0c785ca2f68081bfa7d91973934>
<computer_env>
Action: click(point='<point>400 435</point>')
</computer_env>`;

      const result = parseActionVlm(input, [1000, 1000]);

      expect(result).toEqual([
        {
          reflection: null,
          thought: `Hmm, the previous action didn't work. Let's check again. The Google search page is open, and the search box is where we need to input. Wait, maybe the search box wasn't properly focused? Let me click on the search box first to make sure it's active, then type "UI TARS". That should work. So first, click the search input field, then type the query.`,
          action_type: 'click',
          action_inputs: {
            start_box: '[0.4,0.435,0.4,0.435]',
          },
        },
      ]);
    });

    it('should parse omni format(2)', () => {
      const input = `<think>Hmm, the previous action didn't work. Let's check again. The Google search page is open, and the search box is where we need to input. Wait, maybe the search box wasn't properly focused? Let me click on the search box first to make sure it's active, then type "UI TARS". That should work. So first, click the search input field, then type the query.</think>
<computer_env>
Action: click(point='<point>400 435</point>')
</computer_env>`;

      const result = parseActionVlm(input, [1000, 1000]);

      expect(result).toEqual([
        {
          reflection: null,
          thought: `Hmm, the previous action didn't work. Let's check again. The Google search page is open, and the search box is where we need to input. Wait, maybe the search box wasn't properly focused? Let me click on the search box first to make sure it's active, then type "UI TARS". That should work. So first, click the search input field, then type the query.`,
          action_type: 'click',
          action_inputs: {
            start_box: '[0.4,0.435,0.4,0.435]',
          },
        },
      ]);
    });

    it('should parse omni format(3)', () => {
      const input = `<think_xxx>Hmm, the previous action didn't work. Let's check again. The Google search page is open, and the search box is where we need to input. Wait, maybe the search box wasn't properly focused? Let me click on the search box first to make sure it's active, then type "UI TARS". That should work. So first, click the search input field, then type the query.</think_xxx>
<computer_env>
Action: click(point='<point>400 435</point>')
</computer_env>`;

      const result = parseActionVlm(input, [1000, 1000]);

      expect(result).toEqual([
        {
          reflection: null,
          thought: `Hmm, the previous action didn't work. Let's check again. The Google search page is open, and the search box is where we need to input. Wait, maybe the search box wasn't properly focused? Let me click on the search box first to make sure it's active, then type "UI TARS". That should work. So first, click the search input field, then type the query.`,
          action_type: 'click',
          action_inputs: {
            start_box: '[0.4,0.435,0.4,0.435]',
          },
        },
      ]);
    });
  });
});
