# Part of Odoo. See LICENSE file for full copyright and licensing details.

import json

from .common import SpreadsheetTestCommon, TEST_CONTENT
from odoo.exceptions import AccessError
from odoo.tests.common import new_test_user

class SpreadsheetTemplate(SpreadsheetTestCommon):

    def test_copy_template_without_name(self):
        template = self.env["spreadsheet.template"].create({
            "spreadsheet_data": TEST_CONTENT,
            "name": "Template name",
        })
        self.assertEqual(
            template.copy().name,
            "Template name (copy)",
            "It should mention the template is a copy"
        )

    def test_copy_template_with_name(self):
        template = self.env["spreadsheet.template"].create({
            "spreadsheet_data": TEST_CONTENT,
            "name": "Template name",
        })
        self.assertEqual(
            template.copy({"name": "New Name"}).name,
            "New Name",
            "It should have assigned the given name"
        )

    def test_file_name(self):
        template = self.env["spreadsheet.template"].create({
            "spreadsheet_data": TEST_CONTENT,
            "name": "Template name",
        })
        self.assertEqual(template.file_name, "Template name.osheet.json")

    def test_allow_write_on_own_template(self):
        template = self.env["spreadsheet.template"].with_user(self.spreadsheet_user)\
            .create({
                "spreadsheet_data": TEST_CONTENT,
                "name": "Template name",
            })
        template.write({"name": "bye"})
        self.assertEqual(
            template.name,
            "bye",
            "Document User can edit their own templates"
        )

    def test_forbid_write_on_others_template(self):
        template = self.env["spreadsheet.template"].create({
            "spreadsheet_data": TEST_CONTENT,
            "name": "Template name",
        })
        with self.assertRaises(
            AccessError, msg="Document User cannot edit other's templates"
        ):
            template.with_user(self.spreadsheet_user).write(
                {"name": "bye"}
            )

    def test_action_create_spreadsheet(self):
        template = self.env["spreadsheet.template"].create({
            "spreadsheet_data": TEST_CONTENT,
            "name": "Template name",
        })
        commands = self.new_revision_data(template)
        template.dispatch_spreadsheet_message(commands)
        action = template.action_create_spreadsheet()
        spreadsheet_id = action["params"]["spreadsheet_id"]
        document = self.env["documents.document"].browse(spreadsheet_id)
        self.assertTrue(document.exists())
        self.assertEqual(document.handler, "spreadsheet")
        self.assertEqual(document.mimetype, "application/o-spreadsheet")
        self.assertEqual(document.name, "Template name")
        self.assertEqual(document.spreadsheet_data, TEST_CONTENT)
        self.assertEqual(
            len(document.spreadsheet_revision_ids),
            2,
            "it should have copied the revision and added the user locale revision"
        )
        self.assertEqual(action["type"], "ir.actions.client")
        self.assertEqual(action["tag"], "action_open_spreadsheet")
        self.assertEqual(action["params"].get("convert_from_template"), None)

    def test_action_create_spreadsheet_non_admin(self):
        user = new_test_user(
            self.env, login="Jean", groups="documents.group_documents_user"
        )
        template = self.env["spreadsheet.template"].create({
            "spreadsheet_data": TEST_CONTENT,
            "name": "Template name",
        })
        commands = self.new_revision_data(template)
        template.dispatch_spreadsheet_message(commands)
        action = template.with_user(user).action_create_spreadsheet()
        spreadsheet_id = action["params"]["spreadsheet_id"]
        document = self.env["documents.document"].browse(spreadsheet_id)
        self.assertTrue(document.exists())
        self.assertEqual(
            len(document.spreadsheet_revision_ids),
            2,
            "it should have copied the revision and added the user locale revision"
        )

    def test_action_create_spreadsheet_in_folder(self):
        template = self.env["spreadsheet.template"].create({
            "spreadsheet_data": TEST_CONTENT,
            "name": "Template name",
        })
        action = template.action_create_spreadsheet({
            "folder_id": self.folder.id
        })
        spreadsheet_id = action["params"]["spreadsheet_id"]
        document = self.env["documents.document"].browse(spreadsheet_id)
        self.assertEqual(document.folder_id, self.folder)

    def test_action_create_spreadsheet_purges_comments(self):
        base_data = {
            "sheets": [{
                "comments": [
                    {"A1": {"threadId": 1, "isResolved": False}}
                ]
            }]
        }
        template = self.env["spreadsheet.template"].create({
            "spreadsheet_data": json.dumps(base_data),
            "name": "Template name",
        })
        template._dispatch_command(
            {
                "type": "ADD_COMMENT_THREAD",
                "sheetId": "sh1",
                "col": 0,
                "row": 1,
                "threadId": 2,
            }
        )

        action = template.action_create_spreadsheet()
        spreadsheet_id = action["params"]["spreadsheet_id"]
        document = self.env["documents.document"].browse(spreadsheet_id)
        self.assertEqual(document.spreadsheet_data, '{"sheets": [{"comments": {}}]}')
        self.assertEqual(
            len(document.spreadsheet_revision_ids),
            2,
            "it should have copied the revision and added the user locale revision"
        )
        comment_revision = json.loads(document.spreadsheet_revision_ids[0].commands)
        self.assertEqual(comment_revision["commands"], [])