import { MockFileSystem } from '@shopify/theme-check-common/src/test';
import { assert, beforeEach, describe, expect, it } from 'vitest';
import { TextDocumentEdit } from 'vscode-json-languageservice';
import { ApplyWorkspaceEditParams } from 'vscode-languageserver-protocol';
import { TextDocument } from 'vscode-languageserver-textdocument';
import { ClientCapabilities } from '../../ClientCapabilities';
import { DocumentManager } from '../../documents';
import { MockConnection, mockConnection } from '../../test/MockConnection';
import { RenameHandler } from '../RenameHandler';

describe('Module: AssetRenameHandler', () => {
  const mockRoot = 'mock-fs:';
  const findThemeRootURI = async () => mockRoot;
  let capabilities: ClientCapabilities;
  let documentManager: DocumentManager;
  let handler: RenameHandler;
  let connection: MockConnection;
  let fs: MockFileSystem;
  beforeEach(() => {
    connection = mockConnection(mockRoot);
    connection.spies.sendRequest.mockReturnValue(Promise.resolve(true));
    capabilities = new ClientCapabilities();
    fs = new MockFileSystem(
      {
        'assets/oldName.js': 'console.log("Hello, world!")',
        'sections/section.liquid': `<script src="{{ 'oldName.js' | asset_url }}" defer></script>`,
        'blocks/block.liquid': `{{ 'oldName.js' | asset_url | script_tag }} oldName.js`,
      },
      mockRoot,
    );
    documentManager = new DocumentManager(fs);
    handler = new RenameHandler(connection, capabilities, documentManager, findThemeRootURI);
  });

  describe('when the client does not support workspace/applyEdit', () => {
    beforeEach(() => {
      capabilities.setup({
        workspace: {
          applyEdit: false,
        },
      });
    });

    it('does nothing', async () => {
      await handler.onDidRenameFiles({
        files: [
          {
            oldUri: 'mock-fs:/assets/oldName.js',
            newUri: 'mock-fs:/assets/newName.js',
          },
        ],
      });
      expect(connection.spies.sendRequest).not.toHaveBeenCalled();
    });
  });

  describe('when the client supports workspace/applyEdit', () => {
    beforeEach(() => {
      capabilities.setup({
        workspace: {
          applyEdit: true,
        },
      });
    });

    it('returns a needConfirmation: false workspace edit for renaming an asset', async () => {
      await handler.onDidRenameFiles({
        files: [
          {
            oldUri: 'mock-fs:/assets/oldName.js',
            newUri: 'mock-fs:/assets/newName.js',
          },
        ],
      });

      const expectedTextEdit = {
        range: expect.any(Object),
        newText: 'newName.js',
      };

      expect(connection.spies.sendRequest).toHaveBeenCalledWith('workspace/applyEdit', {
        label: "Rename asset 'oldName.js' to 'newName.js'",
        edit: {
          changeAnnotations: {
            renameAsset: {
              label: `Rename asset 'oldName.js' to 'newName.js'`,
              needsConfirmation: false,
            },
          },
          documentChanges: [
            {
              textDocument: {
                uri: 'mock-fs:/sections/section.liquid',
                version: null,
              },
              edits: [expectedTextEdit],
              annotationId: 'renameAsset',
            },
            {
              textDocument: {
                uri: 'mock-fs:/blocks/block.liquid',
                version: null,
              },
              edits: [expectedTextEdit],
              annotationId: 'renameAsset',
            },
          ],
        },
      });
    });

    it('replaces the correct text in the documents', async () => {
      await handler.onDidRenameFiles({
        files: [
          {
            oldUri: 'mock-fs:/assets/oldName.js',
            newUri: 'mock-fs:/assets/newName.js',
          },
        ],
      });

      const params: ApplyWorkspaceEditParams = connection.spies.sendRequest.mock.calls[0][1];
      const expectedFs = new MockFileSystem(
        {
          'assets/oldName.js': 'console.log("Hello, world!")',
          'sections/section.liquid': `<script src="{{ 'newName.js' | asset_url }}" defer></script>`,
          'blocks/block.liquid': `{{ 'newName.js' | asset_url | script_tag }} oldName.js`,
        },
        'mock-fs:',
      );

      assert(params.edit);
      assert(params.edit.documentChanges);
      for (const docChange of params.edit.documentChanges) {
        assert(TextDocumentEdit.is(docChange));
        const uri = docChange.textDocument.uri;
        const edits = docChange.edits;
        const initialDoc = await fs.readFile(uri);
        const expectedDoc = await expectedFs.readFile(uri);
        expect(edits).to.applyEdits(initialDoc, expectedDoc);
      }
    });

    it('handles .js.liquid files', async (ext) => {
      await handler.onDidRenameFiles({
        files: [
          {
            oldUri: 'mock-fs:/assets/oldName.js',
            newUri: `mock-fs:/assets/newName.js.liquid`,
          },
        ],
      });

      const params: ApplyWorkspaceEditParams = connection.spies.sendRequest.mock.calls[0][1];
      const expectedFs = new MockFileSystem(
        {
          'assets/oldName.js': 'console.log("Hello, world!")',
          'sections/section.liquid': `<script src="{{ 'newName.js' | asset_url }}" defer></script>`,
          'blocks/block.liquid': `{{ 'newName.js' | asset_url | script_tag }} oldName.js`,
        },
        'mock-fs:',
      );

      assert(params.edit);
      assert(params.edit.documentChanges);
      for (const docChange of params.edit.documentChanges) {
        assert(TextDocumentEdit.is(docChange));
        const uri = docChange.textDocument.uri;
        const edits = docChange.edits;
        const initialDoc = await fs.readFile(uri);
        const expectedDoc = await expectedFs.readFile(uri);
        expect(edits).to.applyEdits(initialDoc, expectedDoc);
      }
    });

    it('handles .css.liquid files', async () => {
      fs = new MockFileSystem(
        {
          'assets/oldName.css.liquid': 'body { color: red; }',
          'sections/section.liquid': `{% echo 'oldName.css' | asset_url | stylesheet_tag %}`,
        },
        'mock-fs:',
      );
      documentManager = new DocumentManager(fs);
      handler = new RenameHandler(connection, capabilities, documentManager, findThemeRootURI);

      await handler.onDidRenameFiles({
        files: [
          {
            oldUri: 'mock-fs:/assets/oldName.css.liquid',
            newUri: `mock-fs:/assets/newName.css.liquid`,
          },
        ],
      });

      const params: ApplyWorkspaceEditParams = connection.spies.sendRequest.mock.calls[0][1];
      const expectedFs = new MockFileSystem(
        {
          'assets/newName.css.liquid': 'body { color: red; }',
          'sections/section.liquid': `{% echo 'newName.css' | asset_url | stylesheet_tag %}`,
        },
        'mock-fs:',
      );

      assert(params.edit);
      assert(params.edit.documentChanges);
      for (const docChange of params.edit.documentChanges) {
        assert(TextDocumentEdit.is(docChange));
        const uri = docChange.textDocument.uri;
        const edits = docChange.edits;
        const initialDoc = await fs.readFile(uri);
        const expectedDoc = await expectedFs.readFile(uri);
        expect(edits).to.applyEdits(initialDoc, expectedDoc);
      }
    });
  });
});