feat: install DB,Orm,Schema,Seeder facades

database/setup/setup.go

@@ -10,23 +10,32 @@ import (
 )
 
 func main() {
-	// config, err := supportfile.GetFrameworkContent("database/setup/config/database.go")
-	// if err != nil {
-	// 	panic(err)
-	// }
+	stubs := Stubs{}
 
 	packages.Setup(os.Args).
 		Install(
 			modify.GoFile(path.Config("app.go")).
 				Find(match.Imports()).Modify(modify.AddImport(packages.GetModulePath())).
 				Find(match.Providers()).Modify(modify.Register("&database.ServiceProvider{}")),
-			// modify.File(path.Config("database.go")).Overwrite(config),
+			modify.File(path.Config("database.go")).Overwrite(stubs.Config(packages.GetModuleNameFromArgs(os.Args))),
+			modify.File(path.Database("kernel.go")).Overwrite(stubs.Kernel(packages.GetModuleNameFromArgs(os.Args))),
+			modify.WhenFacade("DB", modify.File(path.Facades("db.go")).Overwrite(stubs.DBFacade())),
+			modify.WhenFacade("Orm", modify.File(path.Facades("orm.go")).Overwrite(stubs.OrmFacade())),
+			modify.WhenFacade("Schema", modify.File(path.Facades("schema.go")).Overwrite(stubs.SchemaFacade())),
+			modify.WhenFacade("Seeder", modify.File(path.Facades("seeder.go")).Overwrite(stubs.SeederFacade())),
 		).
 		Uninstall(
-			modify.GoFile(path.Config("app.go")).
-				Find(match.Providers()).Modify(modify.Unregister("&database.ServiceProvider{}")).
-				Find(match.Imports()).Modify(modify.RemoveImport(packages.GetModulePath())),
-			// modify.File(path.Config("database.go")).Remove(),
+			modify.WhenNoFacades([]string{"DB", "Orm", "Schema", "Seeder"},
+				modify.GoFile(path.Config("app.go")).
+					Find(match.Providers()).Modify(modify.Unregister("&database.ServiceProvider{}")).
+					Find(match.Imports()).Modify(modify.RemoveImport(packages.GetModulePath())),
+				modify.File(path.Config("database.go")).Remove(),
+				modify.File(path.Database("kernel.go")).Remove(),
+			),
+			modify.WhenFacade("DB", modify.File(path.Facades("db.go")).Remove()),
+			modify.WhenFacade("Orm", modify.File(path.Facades("orm.go")).Remove()),
+			modify.WhenFacade("Schema", modify.File(path.Facades("schema.go")).Remove()),
+			modify.WhenFacade("Seeder", modify.File(path.Facades("seeder.go")).Remove()),
 		).
 		Execute()
 }

database/setup/config/database.go → database/setup/stubs.go

@@ -1,14 +1,23 @@
-package config
+package main
 
 import (
-	"github.com/goravel/framework/facades"
+	"strings"
+)
+
+type Stubs struct{}
+
+func (s Stubs) Config(module string) string {
+	content := `package config
+
+import (
+	"DummyModule/app/facades"
 )
 
 func init() {
 	config := facades.Config()
 	config.Add("database", map[string]any{
 		// Default database connection name
-		"default": config.Env("DB_CONNECTION", "postgres"),
+		"default": config.Env("DB_CONNECTION"),
 
 		// Database connections
 		"connections": map[string]any{},
@@ -61,3 +70,82 @@ func init() {
 		},
 	})
 }
+`
+
+	return strings.ReplaceAll(content, "DummyModule", module)
+}
+
+func (s Stubs) DBFacade() string {
+	return `package facades
+
+import (
+	"github.com/goravel/framework/contracts/database/db"
+)
+
+func DB() db.DB {
+	return App().MakeDB()
+}
+`
+}
+
+func (s Stubs) OrmFacade() string {
+	return `package facades
+
+import (
+	"github.com/goravel/framework/contracts/database/orm"
+)
+
+func Orm() orm.Orm {
+	return App().MakeOrm()
+}
+`
+}
+
+func (s Stubs) SchemaFacade() string {
+	return `package facades
+
+import (
+	"github.com/goravel/framework/contracts/database/schema"
+)
+
+func Schema() schema.Schema {
+	return App().MakeSchema()
+}
+`
+}
+
+func (s Stubs) SeederFacade() string {
+	return `package facades
+
+import (
+	"github.com/goravel/framework/contracts/database/seeder"
+)
+
+func Seeder() seeder.Facade {
+	return App().MakeSeeder()
+}
+`
+}
+
+func (s Stubs) Kernel(module string) string {
+	content := `package database
+
+import (
+	"github.com/goravel/framework/contracts/database/schema"
+	"github.com/goravel/framework/contracts/database/seeder"
+)
+
+type Kernel struct {
+}
+
+func (kernel Kernel) Migrations() []schema.Migration {
+	return []schema.Migration{}
+}
+
+func (kernel Kernel) Seeders() []seeder.Seeder {
+	return []seeder.Seeder{}
+}
+`
+
+	return strings.ReplaceAll(content, "DummyModule", module)
+}