ExportSeedCommand.php 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152
  1. <?php
  2. namespace Dcat\Admin\Console;
  3. use Illuminate\Console\Command;
  4. class ExportSeedCommand extends Command
  5. {
  6. /**
  7. * The console command name.
  8. *
  9. * @var string
  10. */
  11. protected $signature = 'admin:export-seed {classname=AdminTablesSeeder}
  12. {--users : add to seed users tables}';
  13. /**
  14. * The console command description.
  15. *
  16. * @var string
  17. */
  18. protected $description = 'Export seed a dcat-admin database tables menu, roles and permissions';
  19. /**
  20. * Execute the console command.
  21. *
  22. * @return void
  23. */
  24. public function handle()
  25. {
  26. $name = $this->argument('classname');
  27. $exceptFields = [];
  28. $exportUsers = $this->option('users');
  29. $namespace = version_compare(app()->version(), '8.0.0', '<') ? 'seeds' : 'seeders';
  30. $seedFile = $this->laravel->databasePath().'/'.$namespace.'/'.$name.'.php';
  31. $contents = $this->getStub('AdminTablesSeeder');
  32. $replaces = [
  33. 'DummyNamespace' => ucwords($namespace),
  34. 'DummyClass' => $name,
  35. 'ClassMenu' => config('admin.database.menu_model'),
  36. 'ClassPermission' => config('admin.database.permissions_model'),
  37. 'ClassRole' => config('admin.database.roles_model'),
  38. 'TableRoleMenu' => config('admin.database.role_menu_table'),
  39. 'TableRolePermissions' => config('admin.database.role_permissions_table'),
  40. 'ArrayMenu' => $this->getTableDataArrayAsString(config('admin.database.menu_table'), $exceptFields),
  41. 'ArrayPermission' => $this->getTableDataArrayAsString(config('admin.database.permissions_table'), $exceptFields),
  42. 'ArrayRole' => $this->getTableDataArrayAsString(config('admin.database.roles_table'), $exceptFields),
  43. 'ArrayPivotRoleMenu' => $this->getTableDataArrayAsString(config('admin.database.role_menu_table'), $exceptFields),
  44. 'ArrayPivotRolePermissions' => $this->getTableDataArrayAsString(config('admin.database.role_permissions_table'), $exceptFields),
  45. ];
  46. if ($exportUsers) {
  47. $replaces = array_merge($replaces, [
  48. 'ClassUsers' => config('admin.database.users_model'),
  49. 'TableRoleUsers' => config('admin.database.role_users_table'),
  50. 'TablePermissionsUsers' => config('admin.database.user_permissions_table'),
  51. 'ArrayUsers' => $this->getTableDataArrayAsString(config('admin.database.users_table'), $exceptFields),
  52. 'ArrayPivotRoleUsers' => $this->getTableDataArrayAsString(config('admin.database.role_users_table'), $exceptFields),
  53. 'ArrayPivotPermissionsUsers' => $this->getTableDataArrayAsString(config('admin.database.user_permissions_table'), $exceptFields),
  54. ]);
  55. } else {
  56. $contents = preg_replace('/\/\/ users tables[\s\S]*?(?=\/\/ finish)/mu', '', $contents);
  57. }
  58. $contents = str_replace(array_keys($replaces), array_values($replaces), $contents);
  59. $this->laravel['files']->put($seedFile, $contents);
  60. $this->line('<info>Admin tables seed file was created:</info> '.str_replace(base_path(), '', $seedFile));
  61. $this->line("Use: <info>php artisan db:seed --class={$name}</info>");
  62. }
  63. /**
  64. * Get data array from table as string result var_export.
  65. *
  66. * @param string $table
  67. * @param array $exceptFields
  68. *
  69. * @return string
  70. */
  71. protected function getTableDataArrayAsString($table, $exceptFields = [])
  72. {
  73. $fields = \DB::getSchemaBuilder()->getColumnListing($table);
  74. $fields = array_diff($fields, $exceptFields);
  75. $array = \DB::table($table)->get($fields)->map(function ($item) {
  76. return (array) $item;
  77. })->all();
  78. return $this->varExport($array, str_repeat(' ', 12));
  79. }
  80. /**
  81. * Get stub contents.
  82. *
  83. * @param $name
  84. *
  85. * @return string
  86. */
  87. protected function getStub($name)
  88. {
  89. return $this->laravel['files']->get(__DIR__."/stubs/$name.stub");
  90. }
  91. /**
  92. * Custom var_export for correct work with \r\n.
  93. *
  94. * @param $var
  95. * @param string $indent
  96. *
  97. * @return string
  98. */
  99. protected function varExport($var, $indent = '')
  100. {
  101. switch (gettype($var)) {
  102. case 'string':
  103. return '"'.addcslashes($var, "\\\$\"\r\n\t\v\f").'"';
  104. case 'array':
  105. $indexed = array_keys($var) === range(0, count($var) - 1);
  106. $r = [];
  107. foreach ($var as $key => $value) {
  108. $r[] = "$indent "
  109. .($indexed ? '' : $this->varExport($key).' => ')
  110. .$this->varExport($value, "{$indent} ");
  111. }
  112. return "[\n".implode(",\n", $r)."\n".$indent.']';
  113. case 'boolean':
  114. return $var ? 'true' : 'false';
  115. case 'integer':
  116. case 'double':
  117. return $var;
  118. default:
  119. return var_export($var, true);
  120. }
  121. }
  122. }