import { context, events } from "@budibase/backend-core"
import {
  AutoFieldSubType,
  Datasource,
  BBReferenceFieldSubType,
  FieldType,
  INTERNAL_TABLE_SOURCE_ID,
  InternalTable,
  RelationshipType,
  Row,
  SaveTableRequest,
  Table,
  TableSourceType,
  User,
  ViewCalculation,
  ViewV2Enriched,
} from "@budibase/types"
import { checkBuilderEndpoint } from "./utilities/TestFunctions"
import * as setup from "./utilities"
import * as uuid from "uuid"

import { generator } from "@budibase/backend-core/tests"
import { DatabaseName, getDatasource } from "../../../integrations/tests/utils"
import { tableForDatasource } from "../../../tests/utilities/structures"
import timekeeper from "timekeeper"

const { basicTable } = setup.structures
const ISO_REGEX_PATTERN = /^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}\.\d{3}Z$/

describe.each([
  ["internal", undefined],
  [DatabaseName.POSTGRES, getDatasource(DatabaseName.POSTGRES)],
  [DatabaseName.MYSQL, getDatasource(DatabaseName.MYSQL)],
  [DatabaseName.SQL_SERVER, getDatasource(DatabaseName.SQL_SERVER)],
  [DatabaseName.MARIADB, getDatasource(DatabaseName.MARIADB)],
])("/tables (%s)", (_, dsProvider) => {
  const isInternal: boolean = !dsProvider
  let datasource: Datasource | undefined
  let config = setup.getConfig()

  afterAll(setup.afterAll)

  beforeAll(async () => {
    await config.init()
    if (dsProvider) {
      datasource = await config.api.datasource.create(await dsProvider)
    }
  })

  describe("create", () => {
    beforeEach(() => {
      jest.clearAllMocks()
    })

    it("creates a table successfully", async () => {
      const name = generator.guid()
      const table = await config.api.table.save(
        tableForDatasource(datasource, { name })
      )
      expect(table.name).toEqual(name)
      expect(events.table.created).toHaveBeenCalledTimes(1)
      expect(events.table.created).toHaveBeenCalledWith(table)
    })

    it("creates a table via data import", async () => {
      const table: SaveTableRequest = basicTable()
      table.rows = [{ name: "test-name", description: "test-desc" }]

      const res = await config.api.table.save(table)

      expect(events.table.created).toHaveBeenCalledTimes(1)
      expect(events.table.created).toHaveBeenCalledWith(res)
      expect(events.table.imported).toHaveBeenCalledTimes(1)
      expect(events.table.imported).toHaveBeenCalledWith(res)
      expect(events.rows.imported).toHaveBeenCalledTimes(1)
      expect(events.rows.imported).toHaveBeenCalledWith(res, 1)
    })

    it("should apply authorization to endpoint", async () => {
      await checkBuilderEndpoint({
        config,
        method: "POST",
        url: `/api/tables`,
        body: basicTable(),
      })
    })
  })

  describe("update", () => {
    it("updates a table", async () => {
      const table = await config.api.table.save(
        tableForDatasource(datasource, {
          schema: {
            name: {
              type: FieldType.STRING,
              name: "name",
              constraints: {
                type: "string",
              },
            },
          },
        })
      )

      const updatedTable = await config.api.table.save({
        ...table,
        name: generator.guid(),
      })

      expect(events.table.updated).toHaveBeenCalledTimes(1)
      expect(events.table.updated).toHaveBeenCalledWith(updatedTable)
    })

    it("updates all the row fields for a table when a schema key is renamed", async () => {
      const testTable = await config.api.table.save(basicTable(datasource))
      await config.createLegacyView({
        name: "TestView",
        field: "Price",
        calculation: ViewCalculation.STATISTICS,
        tableId: testTable._id!,
        schema: {},
        filters: [],
      })

      const testRow = await config.api.row.save(testTable._id!, {
        name: "test",
      })

      const { name, ...otherColumns } = testTable.schema
      const updatedTable = await config.api.table.save({
        ...testTable,
        _rename: {
          old: "name",
          updated: "updatedName",
        },
        schema: {
          ...otherColumns,
          updatedName: {
            ...name,
            name: "updatedName",
          },
        },
      })

      expect(updatedTable.name).toEqual(testTable.name)

      const res = await config.api.row.get(testTable._id!, testRow._id!)
      expect(res.updatedName).toEqual("test")
      expect(res.name).toBeUndefined()
    })

    it("updates only the passed fields", async () => {
      await timekeeper.withFreeze(new Date(2021, 1, 1), async () => {
        const table = await config.api.table.save(
          tableForDatasource(datasource, {
            schema: {
              autoId: {
                name: "id",
                type: FieldType.NUMBER,
                subtype: AutoFieldSubType.AUTO_ID,
                autocolumn: true,
                constraints: {
                  type: "number",
                  presence: false,
                },
              },
            },
          })
        )

        const newName = generator.guid()

        const updatedTable = await config.api.table.save({
          ...table,
          name: newName,
        })

        let expected: Table = {
          ...table,
          name: newName,
          _id: expect.any(String),
        }
        if (isInternal) {
          expected._rev = expect.stringMatching(/^2-.+/)
        }

        expect(updatedTable).toEqual(expected)

        const persistedTable = await config.api.table.get(updatedTable._id!)
        expected = {
          ...table,
          name: newName,
          _id: updatedTable._id,
        }
        if (datasource?.isSQL) {
          expected.sql = true
        }
        if (isInternal) {
          expected._rev = expect.stringMatching(/^2-.+/)
        }
        expect(persistedTable).toEqual(expected)
      })
    })

    describe("user table", () => {
      isInternal &&
        it("should add roleId and email field when adjusting user table schema", async () => {
          const table = await config.api.table.save({
            ...basicTable(datasource),
            _id: "ta_users",
          })
          expect(table.schema.email).toBeDefined()
          expect(table.schema.roleId).toBeDefined()
        })
    })

    it("should add a new column for an internal DB table", async () => {
      const saveTableRequest: SaveTableRequest = {
        ...basicTable(),
      }

      const response = await config.api.table.save(saveTableRequest)

      const expectedResponse = {
        ...saveTableRequest,
        _rev: expect.stringMatching(/^\d-.+/),
        _id: expect.stringMatching(/^ta_.+/),
        createdAt: expect.stringMatching(ISO_REGEX_PATTERN),
        updatedAt: expect.stringMatching(ISO_REGEX_PATTERN),
        views: {},
      }
      expect(response).toEqual(expectedResponse)
    })
  })

  describe("import", () => {
    it("imports rows successfully", async () => {
      const name = generator.guid()
      const table = await config.api.table.save(
        basicTable(datasource, { name })
      )
      const importRequest = {
        schema: table.schema,
        rows: [{ name: "test-name", description: "test-desc" }],
      }

      jest.clearAllMocks()

      await config.api.table.import(table._id!, importRequest)

      expect(events.table.created).not.toHaveBeenCalled()
      expect(events.rows.imported).toHaveBeenCalledTimes(1)
      expect(events.rows.imported).toHaveBeenCalledWith(
        expect.objectContaining({
          name,
          _id: table._id,
        }),
        1
      )
    })
  })

  describe("fetch", () => {
    let testTable: Table

    beforeEach(async () => {
      testTable = await config.api.table.save(
        basicTable(datasource, { name: generator.guid() })
      )
    })

    it("returns all tables", async () => {
      const res = await config.api.table.fetch()
      const table = res.find(t => t._id === testTable._id)
      expect(table).toBeDefined()
      expect(table!.name).toEqual(testTable.name)
      expect(table!.type).toEqual("table")
      expect(table!.sourceType).toEqual(testTable.sourceType)
    })

    it("should apply authorization to endpoint", async () => {
      await checkBuilderEndpoint({
        config,
        method: "GET",
        url: `/api/tables`,
      })
    })

    it("should enrich the view schemas", async () => {
      const viewV2 = await config.api.viewV2.create({
        tableId: testTable._id!,
        name: generator.guid(),
      })
      const legacyView = await config.api.legacyView.save({
        tableId: testTable._id!,
        name: generator.guid(),
        filters: [],
        schema: {},
      })

      const res = await config.api.table.fetch()

      const table = res.find(t => t._id === testTable._id)
      expect(table).toBeDefined()
      expect(table!.views![viewV2.name]).toBeDefined()

      const expectedViewV2: ViewV2Enriched = {
        ...viewV2,
        schema: {
          description: {
            constraints: {
              type: "string",
            },
            name: "description",
            type: FieldType.STRING,
            visible: false,
          },
          name: {
            constraints: {
              type: "string",
            },
            name: "name",
            type: FieldType.STRING,
            visible: false,
          },
        },
      }

      if (!isInternal) {
        expectedViewV2.schema!.id = {
          name: "id",
          type: FieldType.NUMBER,
          visible: false,
          autocolumn: true,
        }
      }

      expect(table!.views![viewV2.name!]).toEqual(expectedViewV2)

      if (isInternal) {
        expect(table!.views![legacyView.name!]).toBeDefined()
        expect(table!.views![legacyView.name!]).toEqual({
          ...legacyView,
          schema: {
            description: {
              constraints: {
                type: "string",
              },
              name: "description",
              type: "string",
            },
            name: {
              constraints: {
                type: "string",
              },
              name: "name",
              type: "string",
            },
          },
        })
      }
    })
  })

  describe("get", () => {
    it("returns a table", async () => {
      const table = await config.api.table.save(
        basicTable(datasource, { name: generator.guid() })
      )
      const res = await config.api.table.get(table._id!)
      expect(res).toEqual(table)
    })
  })

  describe("indexing", () => {
    it("should be able to create a table with indexes", async () => {
      await context.doInAppContext(config.getAppId(), async () => {
        const db = context.getAppDB()
        const indexCount = (await db.getIndexes()).total_rows
        const table = basicTable()
        table.indexes = ["name"]
        const res = await config.api.table.save(table)
        expect(res._id).toBeDefined()
        expect(res._rev).toBeDefined()
        expect((await db.getIndexes()).total_rows).toEqual(indexCount + 1)
        // update index to see what happens
        table.indexes = ["name", "description"]
        await config.api.table.save({
          ...table,
          _id: res._id,
          _rev: res._rev,
        })
        // shouldn't have created a new index
        expect((await db.getIndexes()).total_rows).toEqual(indexCount + 1)
      })
    })
  })

  describe("destroy", () => {
    let testTable: Table

    beforeEach(async () => {
      testTable = await config.createTable()
    })

    it("returns a success response when a table is deleted.", async () => {
      await config.api.table.destroy(testTable._id!, testTable._rev!, {
        body: { message: `Table ${testTable._id} deleted.` },
      })
      expect(events.table.deleted).toHaveBeenCalledTimes(1)
      expect(events.table.deleted).toHaveBeenCalledWith({
        ...testTable,
        tableId: testTable._id,
      })
    })

    it("deletes linked references to the table after deletion", async () => {
      const linkedTable = await config.createTable({
        name: "LinkedTable",
        type: "table",
        schema: {
          name: {
            type: FieldType.STRING,
            name: "name",
            constraints: {
              type: "string",
            },
          },
          TestTable: {
            type: FieldType.LINK,
            relationshipType: RelationshipType.ONE_TO_MANY,
            name: "TestTable",
            fieldName: "TestTable",
            tableId: testTable._id!,
            constraints: {
              type: "array",
            },
          },
        },
      })

      await config.api.table.destroy(testTable._id!, testTable._rev!, {
        body: { message: `Table ${testTable._id} deleted.` },
      })
      const dependentTable = await config.api.table.get(linkedTable._id!)
      expect(dependentTable.schema.TestTable).not.toBeDefined()
    })

    it("should apply authorization to endpoint", async () => {
      await checkBuilderEndpoint({
        config,
        method: "DELETE",
        url: `/api/tables/${testTable._id}/${testTable._rev}`,
      })
    })
  })

  describe("migrate", () => {
    let users: User[]
    beforeAll(async () => {
      users = await Promise.all([
        config.createUser({ email: `${uuid.v4()}@example.com` }),
        config.createUser({ email: `${uuid.v4()}@example.com` }),
        config.createUser({ email: `${uuid.v4()}@example.com` }),
      ])
    })

    it("should successfully migrate a one-to-many user relationship to a user column", async () => {
      const table = await config.api.table.save({
        name: "table",
        type: "table",
        sourceId: INTERNAL_TABLE_SOURCE_ID,
        sourceType: TableSourceType.INTERNAL,
        schema: {
          "user relationship": {
            type: FieldType.LINK,
            fieldName: "test",
            name: "user relationship",
            constraints: {
              type: "array",
              presence: false,
            },
            relationshipType: RelationshipType.ONE_TO_MANY,
            tableId: InternalTable.USER_METADATA,
          },
        },
      })

      const rows = await Promise.all(
        users.map(u =>
          config.api.row.save(table._id!, { "user relationship": [u] })
        )
      )

      await config.api.table.migrate(table._id!, {
        oldColumn: table.schema["user relationship"],
        newColumn: {
          name: "user column",
          type: FieldType.BB_REFERENCE,
          subtype: BBReferenceFieldSubType.USER,
        },
      })

      const migratedTable = await config.api.table.get(table._id!)
      expect(migratedTable.schema["user column"]).toBeDefined()
      expect(migratedTable.schema["user relationship"]).not.toBeDefined()

      const migratedRows = await config.api.row.fetch(table._id!)

      rows.sort((a, b) => a._id!.localeCompare(b._id!))
      migratedRows.sort((a, b) => a._id!.localeCompare(b._id!))

      for (const [i, row] of rows.entries()) {
        const migratedRow = migratedRows[i]
        expect(migratedRow["user column"]).toBeDefined()
        expect(migratedRow["user relationship"]).not.toBeDefined()
        expect(row["user relationship"][0]._id).toEqual(
          migratedRow["user column"][0]._id
        )
      }
    })

    it("should succeed when the row is created from the other side of the relationship", async () => {
      // We found a bug just after releasing this feature where if the row was created from the
      // users table, not the table linking to it, the migration would succeed but lose the data.
      // This happened because the order of the documents in the link was reversed.
      const table = await config.api.table.save({
        name: "table",
        type: "table",
        sourceId: INTERNAL_TABLE_SOURCE_ID,
        sourceType: TableSourceType.INTERNAL,
        schema: {
          "user relationship": {
            type: FieldType.LINK,
            fieldName: "test",
            name: "user relationship",
            constraints: {
              type: "array",
              presence: false,
            },
            relationshipType: RelationshipType.MANY_TO_ONE,
            tableId: InternalTable.USER_METADATA,
          },
        },
      })

      let testRow = await config.api.row.save(table._id!, {})

      await Promise.all(
        users.map(u =>
          config.api.row.patch(InternalTable.USER_METADATA, {
            tableId: InternalTable.USER_METADATA,
            _rev: u._rev!,
            _id: u._id!,
            test: [testRow],
          })
        )
      )

      await config.api.table.migrate(table._id!, {
        oldColumn: table.schema["user relationship"],
        newColumn: {
          name: "user column",
          type: FieldType.BB_REFERENCE,
          subtype: BBReferenceFieldSubType.USERS,
        },
      })

      const migratedTable = await config.api.table.get(table._id!)
      expect(migratedTable.schema["user column"]).toBeDefined()
      expect(migratedTable.schema["user relationship"]).not.toBeDefined()

      const migratedRow = await config.api.row.get(table._id!, testRow._id!)

      expect(migratedRow["user column"]).toBeDefined()
      expect(migratedRow["user relationship"]).not.toBeDefined()
      expect(migratedRow["user column"]).toHaveLength(3)
      expect(migratedRow["user column"].map((u: Row) => u._id)).toEqual(
        expect.arrayContaining(users.map(u => u._id))
      )
    })

    it("should successfully migrate a many-to-many user relationship to a users column", async () => {
      const table = await config.api.table.save({
        name: "table",
        type: "table",
        sourceId: INTERNAL_TABLE_SOURCE_ID,
        sourceType: TableSourceType.INTERNAL,
        schema: {
          "user relationship": {
            type: FieldType.LINK,
            fieldName: "test",
            name: "user relationship",
            constraints: {
              type: "array",
              presence: false,
            },
            relationshipType: RelationshipType.MANY_TO_MANY,
            tableId: InternalTable.USER_METADATA,
          },
        },
      })

      const row1 = await config.api.row.save(table._id!, {
        "user relationship": [users[0], users[1]],
      })

      const row2 = await config.api.row.save(table._id!, {
        "user relationship": [users[1], users[2]],
      })

      await config.api.table.migrate(table._id!, {
        oldColumn: table.schema["user relationship"],
        newColumn: {
          name: "user column",
          type: FieldType.BB_REFERENCE,
          subtype: BBReferenceFieldSubType.USERS,
        },
      })

      const migratedTable = await config.api.table.get(table._id!)
      expect(migratedTable.schema["user column"]).toBeDefined()
      expect(migratedTable.schema["user relationship"]).not.toBeDefined()

      const row1Migrated = await config.api.row.get(table._id!, row1._id!)
      expect(row1Migrated["user relationship"]).not.toBeDefined()
      expect(row1Migrated["user column"].map((r: Row) => r._id)).toEqual(
        expect.arrayContaining([users[0]._id, users[1]._id])
      )

      const row2Migrated = await config.api.row.get(table._id!, row2._id!)
      expect(row2Migrated["user relationship"]).not.toBeDefined()
      expect(row2Migrated["user column"].map((r: Row) => r._id)).toEqual(
        expect.arrayContaining([users[1]._id, users[2]._id])
      )
    })

    it("should successfully migrate a many-to-one user relationship to a users column", async () => {
      const table = await config.api.table.save({
        name: "table",
        type: "table",
        sourceId: INTERNAL_TABLE_SOURCE_ID,
        sourceType: TableSourceType.INTERNAL,
        schema: {
          "user relationship": {
            type: FieldType.LINK,
            fieldName: "test",
            name: "user relationship",
            constraints: {
              type: "array",
              presence: false,
            },
            relationshipType: RelationshipType.MANY_TO_ONE,
            tableId: InternalTable.USER_METADATA,
          },
        },
      })

      const row1 = await config.api.row.save(table._id!, {
        "user relationship": [users[0], users[1]],
      })

      const row2 = await config.api.row.save(table._id!, {
        "user relationship": [users[2]],
      })

      await config.api.table.migrate(table._id!, {
        oldColumn: table.schema["user relationship"],
        newColumn: {
          name: "user column",
          type: FieldType.BB_REFERENCE,
          subtype: BBReferenceFieldSubType.USERS,
        },
      })

      const migratedTable = await config.api.table.get(table._id!)
      expect(migratedTable.schema["user column"]).toBeDefined()
      expect(migratedTable.schema["user relationship"]).not.toBeDefined()

      const row1Migrated = await config.api.row.get(table._id!, row1._id!)
      expect(row1Migrated["user relationship"]).not.toBeDefined()
      expect(row1Migrated["user column"].map((r: Row) => r._id)).toEqual(
        expect.arrayContaining([users[0]._id, users[1]._id])
      )

      const row2Migrated = await config.api.row.get(table._id!, row2._id!)
      expect(row2Migrated["user relationship"]).not.toBeDefined()
      expect(row2Migrated["user column"].map((r: Row) => r._id)).toEqual([
        users[2]._id,
      ])
    })

    describe("unhappy paths", () => {
      let table: Table
      beforeAll(async () => {
        table = await config.api.table.save(
          tableForDatasource(datasource, {
            schema: {
              "user relationship": {
                type: FieldType.LINK,
                fieldName: "test",
                name: "user relationship",
                constraints: {
                  type: "array",
                  presence: false,
                },
                relationshipType: RelationshipType.MANY_TO_ONE,
                tableId: InternalTable.USER_METADATA,
              },
              num: {
                type: FieldType.NUMBER,
                name: "num",
                constraints: {
                  type: "number",
                  presence: false,
                },
              },
            },
          })
        )
      })

      it("should fail if the new column name is blank", async () => {
        await config.api.table.migrate(
          table._id!,
          {
            oldColumn: table.schema["user relationship"],
            newColumn: {
              name: "",
              type: FieldType.BB_REFERENCE,
              subtype: BBReferenceFieldSubType.USERS,
            },
          },
          { status: 400 }
        )
      })

      it("should fail if the new column name is a reserved name", async () => {
        await config.api.table.migrate(
          table._id!,
          {
            oldColumn: table.schema["user relationship"],
            newColumn: {
              name: "_id",
              type: FieldType.BB_REFERENCE,
              subtype: BBReferenceFieldSubType.USERS,
            },
          },
          { status: 400 }
        )
      })

      it("should fail if the new column name is the same as an existing column", async () => {
        await config.api.table.migrate(
          table._id!,
          {
            oldColumn: table.schema["user relationship"],
            newColumn: {
              name: "num",
              type: FieldType.BB_REFERENCE,
              subtype: BBReferenceFieldSubType.USERS,
            },
          },
          { status: 400 }
        )
      })

      it("should fail if the old column name isn't a column in the table", async () => {
        await config.api.table.migrate(
          table._id!,
          {
            oldColumn: {
              name: "not a column",
              type: FieldType.BB_REFERENCE,
              subtype: BBReferenceFieldSubType.USERS,
            },
            newColumn: {
              name: "new column",
              type: FieldType.BB_REFERENCE,
              subtype: BBReferenceFieldSubType.USERS,
            },
          },
          { status: 400 }
        )
      })
    })
  })
})