// Copyright 2024 Dolthub, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package _go

import (
	"testing"

	"github.com/dolthub/go-mysql-server/sql"
)

func TestCreateViewStatements(t *testing.T) {
	RunScripts(t, createViewStmts)
}

var createViewStmts = []ScriptTest{
	{
		Name: "basic create view statements",
		SetUpScript: []string{
			"create table t1 (pk int);",
			"insert into t1 values (1), (2), (3), (1);",
		},
		Assertions: []ScriptTestAssertion{
			{
				Query:    "create view v as select * from t1 order by pk;",
				Expected: []sql.Row{},
			},
			{
				Query:    "select * from v order by pk;",
				Expected: []sql.Row{{1}, {1}, {2}, {3}},
			},
		},
	},
	{
		Name: "views on different schemas",
		SetUpScript: []string{
			"CREATE SCHEMA testschema;",
			"SET search_path TO testschema;",
			"CREATE TABLE testing (pk INT primary key, v2 TEXT);",
			"INSERT INTO testing VALUES (1,'a'), (2,'b'), (3,'c');",
			"CREATE VIEW testview AS SELECT * FROM testing;",
			"CREATE SCHEMA myschema;",
			"SET search_path TO myschema;",
			"CREATE TABLE mytable (pk INT primary key, v1 INT);",
			"INSERT INTO mytable VALUES (1,4), (2,5), (3,6);",
			"CREATE VIEW myview AS SELECT * FROM mytable;",
		},
		Assertions: []ScriptTestAssertion{
			{
				Query:    "SHOW search_path;",
				Expected: []sql.Row{{"myschema"}},
			},
			{
				Query:    "select v1 from myview order by pk;",
				Expected: []sql.Row{{4}, {5}, {6}},
			},
			{
				Query:       "select v2 from testview order by pk;",
				ExpectedErr: "table not found: testview",
			},
			{
				Query:    "select v2 from testschema.testview order by pk;",
				Expected: []sql.Row{{"a"}, {"b"}, {"c"}},
			},
			{
				Query:    "select name from dolt_schemas;",
				Expected: []sql.Row{{"myview"}},
			},
			{
				Query: "SET search_path = 'testschema';",
			},
			{
				Query:    "SHOW search_path;",
				Expected: []sql.Row{{"testschema"}},
			},
			{
				Query:       "select * from myview order by pk; /* err */",
				ExpectedErr: "table not found: myview",
			},
			{
				Query:    "select v1 from myschema.myview order by pk;",
				Expected: []sql.Row{{4}, {5}, {6}},
			},
			{
				Query:    "select v2 from testview order by pk;",
				Expected: []sql.Row{{"a"}, {"b"}, {"c"}},
			},
			{
				Query:    "select name from dolt_schemas;",
				Expected: []sql.Row{{"testview"}},
			},
			{
				Query:    "SET search_path = testschema, myschema;",
				Expected: []sql.Row{},
			},
			{
				Query:    "SHOW search_path;",
				Expected: []sql.Row{{"testschema, myschema"}},
			},
			{
				Skip:     true, // TODO: Should be able to resolve views from all schema in search_path
				Query:    "select v1 from myview order by pk;",
				Expected: []sql.Row{{4}, {5}, {6}},
			},
			{
				Query:    "select v2 from testview order by pk;",
				Expected: []sql.Row{{"a"}, {"b"}, {"c"}},
			},
			{
				Skip:     true, // TODO: Should be able to resolve views from all schema in search_path
				Query:    "select name from dolt_schemas;",
				Expected: []sql.Row{{"testview"}, {"myview"}},
			},
		},
	},
	{
		Name: "create view from view",
		SetUpScript: []string{
			"create table t1 (pk int);",
			"insert into t1 values (1), (2), (3), (1);",
			"create view v as select * from t1 where pk > 1;",
		},
		Assertions: []ScriptTestAssertion{
			{
				Query:    "create view v1 as select * from v order by pk;",
				Expected: []sql.Row{},
			},
			{
				Query:    "select * from v1 order by pk;",
				Expected: []sql.Row{{2}, {3}},
			},
		},
	},
	{
		Name: "view with expression name",
		SetUpScript: []string{
			"create view v as select 2+2",
		},
		Assertions: []ScriptTestAssertion{
			{
				Query:    "SELECT * from v;",
				Expected: []sql.Row{{4}},
			},
		},
	},
	{
		Name: "view with column names",
		SetUpScript: []string{
			`CREATE TABLE xy (x int primary key, y int);`,
			`insert into xy values (1, 4), (4, 9)`,
		},
		Assertions: []ScriptTestAssertion{
			{
				Query:    "create view v_today(today) as select 2",
				Expected: []sql.Row{},
			},
			{
				Query:    "CREATE VIEW xyv (u,v) AS SELECT * from xy",
				Expected: []sql.Row{},
			},
			{
				Query:    "SELECT v from xyv;",
				Expected: []sql.Row{{4}, {9}},
			},
			{
				Query:    "SELECT today from v_today;",
				Expected: []sql.Row{{2}},
			},
		},
	},
	{
		Skip: true, // TODO: getting subquery alias not supported error
		Name: "nested view",
		SetUpScript: []string{
			"create table t1 (pk int);",
			"insert into t1 values (1), (2), (3), (4);",
		},
		Assertions: []ScriptTestAssertion{
			{
				Query:    "create view unionView as (select * from t1 order by pk desc limit 1) union all (select * from t1 order by pk limit 1)",
				Expected: []sql.Row{},
			},
			{
				Query:    "select * from unionView order by pk;",
				Expected: []sql.Row{{1}, {4}},
			},
		},
	},
	{
		Name: "cast (postgres-specific syntax)",
		SetUpScript: []string{
			"create table t1 (pk int);",
			"insert into t1 values (1), (2), (3), (4);",
		},
		Assertions: []ScriptTestAssertion{
			{
				Query:    "CREATE VIEW v AS SELECT pk::INT2 FROM t1 ORDER BY pk;",
				Expected: []sql.Row{},
			},
			{
				Query:    "select * from v order by pk;",
				Expected: []sql.Row{{1}, {2}, {3}, {4}},
			},
			{
				Query:    "CREATE VIEW v_text AS SELECT pk::int2, (pk)::text AS pk_text FROM t1;",
				Expected: []sql.Row{},
			},
			{
				Query:    "select pk_text from v_text order by pk;",
				Expected: []sql.Row{{"1"}, {"2"}, {"3"}, {"4"}},
			},
		},
	},
	{
		Name: "not yet supported create view queries",
		Assertions: []ScriptTestAssertion{
			{
				Query: "CREATE TEMPORARY VIEW v AS SELECT 1;",
				Skip:  true,
			},
			{
				Query: "CREATE RECURSIVE VIEW v AS SELECT 1;",
				Skip:  true,
			},
			{
				Query: "CREATE VIEW v AS SELECT 1 WITH LOCAL CHECK OPTION;",
				Skip:  true,
			},
			{
				Query: "CREATE VIEW v WITH (check_option = 'local') AS SELECT 1;",
				Skip:  true,
			},
			{
				Query: "CREATE VIEW v WITH (security_barrier = true) AS SELECT 1;",
				Skip:  true,
			},
		},
	},
	{
		Name: "create view with CTE",
		SetUpScript: []string{
			"CREATE TABLE public.t1 (id integer NOT NULL);",
		},
		Assertions: []ScriptTestAssertion{
			{
				Query:    `create view public.v1 as with table1 as (select * from t1) select id from table1;`,
				Expected: []sql.Row{},
			},
		},
	},
	{
		Name: "create view with custom type in its select statement",
		SetUpScript: []string{
			"CREATE TYPE e AS ENUM ('sched', 'busy', 'final', 'help');",
			"CREATE TABLE t (id integer NOT NULL, t e);",
			"INSERT INTO t VALUES (1, 'busy'), (2, 'final'), (3, 'busy'), (4, 'help');",
		},
		Assertions: []ScriptTestAssertion{
			{
				Query:    `create view v as select * from t where (t = 'busy'::e);`,
				Expected: []sql.Row{},
			},
			{
				Query:    `select * from v;`,
				Expected: []sql.Row{{1, "busy"}, {3, "busy"}},
			},
		},
	},
	{
		Name: "View names must be unique across all relation types",
		SetUpScript: []string{
			"CREATE TABLE tbl1 (pk int PRIMARY KEY, v1 int);",
			"CREATE SEQUENCE seq1;",
			"CREATE VIEW existing_view AS SELECT pk FROM tbl1;",
			"CREATE INDEX idx1 ON tbl1 (v1);",
		},
		Assertions: []ScriptTestAssertion{
			{
				Query:       "CREATE VIEW tbl1 AS SELECT pk FROM tbl1;",
				ExpectedErr: `relation "tbl1" already exists`,
			},
			{
				Query:       "CREATE OR REPLACE VIEW tbl1 AS SELECT pk FROM tbl1;",
				ExpectedErr: `"tbl1" is not a view`,
			},
			{
				Query:       "CREATE VIEW seq1 AS SELECT pk FROM tbl1;",
				ExpectedErr: `relation "seq1" already exists`,
			},
			{
				Query:       "CREATE OR REPLACE VIEW seq1 AS SELECT pk FROM tbl1;",
				ExpectedErr: `"seq1" is not a view`,
			},
			{
				Query:       "CREATE VIEW existing_view AS SELECT pk FROM tbl1;",
				ExpectedErr: `relation "existing_view" already exists`,
			},
			{
				Query:    "CREATE OR REPLACE VIEW existing_view AS SELECT pk FROM tbl1;",
				Expected: []sql.Row{},
			},
			{
				Query:       "CREATE VIEW idx1 AS SELECT pk FROM tbl1;",
				ExpectedErr: `relation "idx1" already exists`,
			},
			{
				Query:       "CREATE OR REPLACE VIEW idx1 AS SELECT pk FROM tbl1;",
				ExpectedErr: `"idx1" is not a view`,
			},
		},
	},
}