ExportSeedCommand.php 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149
  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. $seedFile = $this->laravel->databasePath().'/seeds/'.$name.'.php';
  30. $contents = $this->getStub('AdminTablesSeeder');
  31. $replaces = [
  32. 'DummyClass' => $name,
  33. 'ClassMenu' => config('admin.database.menu_model'),
  34. 'ClassPermission' => config('admin.database.permissions_model'),
  35. 'ClassRole' => config('admin.database.roles_model'),
  36. 'TableRoleMenu' => config('admin.database.role_menu_table'),
  37. 'TableRolePermissions' => config('admin.database.role_permissions_table'),
  38. 'ArrayMenu' => $this->getTableDataArrayAsString(config('admin.database.menu_table'), $exceptFields),
  39. 'ArrayPermission' => $this->getTableDataArrayAsString(config('admin.database.permissions_table'), $exceptFields),
  40. 'ArrayRole' => $this->getTableDataArrayAsString(config('admin.database.roles_table'), $exceptFields),
  41. 'ArrayPivotRoleMenu' => $this->getTableDataArrayAsString(config('admin.database.role_menu_table'), $exceptFields),
  42. 'ArrayPivotRolePermissions' => $this->getTableDataArrayAsString(config('admin.database.role_permissions_table'), $exceptFields),
  43. ];
  44. if ($exportUsers) {
  45. $replaces = array_merge($replaces, [
  46. 'ClassUsers' => config('admin.database.users_model'),
  47. 'TableRoleUsers' => config('admin.database.role_users_table'),
  48. 'TablePermissionsUsers' => config('admin.database.user_permissions_table'),
  49. 'ArrayUsers' => $this->getTableDataArrayAsString(config('admin.database.users_table'), $exceptFields),
  50. 'ArrayPivotRoleUsers' => $this->getTableDataArrayAsString(config('admin.database.role_users_table'), $exceptFields),
  51. 'ArrayPivotPermissionsUsers' => $this->getTableDataArrayAsString(config('admin.database.user_permissions_table'), $exceptFields),
  52. ]);
  53. } else {
  54. $contents = preg_replace('/\/\/ users tables[\s\S]*?(?=\/\/ finish)/mu', '', $contents);
  55. }
  56. $contents = str_replace(array_keys($replaces), array_values($replaces), $contents);
  57. $this->laravel['files']->put($seedFile, $contents);
  58. $this->line('<info>Admin tables seed file was created:</info> '.str_replace(base_path(), '', $seedFile));
  59. $this->line("Use: <info>php artisan db:seed --class={$name}</info>");
  60. }
  61. /**
  62. * Get data array from table as string result var_export.
  63. *
  64. * @param string $table
  65. * @param array $exceptFields
  66. *
  67. * @return string
  68. */
  69. protected function getTableDataArrayAsString($table, $exceptFields = [])
  70. {
  71. $fields = \DB::getSchemaBuilder()->getColumnListing($table);
  72. $fields = array_diff($fields, $exceptFields);
  73. $array = \DB::table($table)->get($fields)->map(function ($item) {
  74. return (array) $item;
  75. })->all();
  76. return $this->varExport($array, str_repeat(' ', 12));
  77. }
  78. /**
  79. * Get stub contents.
  80. *
  81. * @param $name
  82. *
  83. * @return string
  84. */
  85. protected function getStub($name)
  86. {
  87. return $this->laravel['files']->get(__DIR__."/stubs/$name.stub");
  88. }
  89. /**
  90. * Custom var_export for correct work with \r\n.
  91. *
  92. * @param $var
  93. * @param string $indent
  94. *
  95. * @return string
  96. */
  97. protected function varExport($var, $indent = '')
  98. {
  99. switch (gettype($var)) {
  100. case 'string':
  101. return '"'.addcslashes($var, "\\\$\"\r\n\t\v\f").'"';
  102. case 'array':
  103. $indexed = array_keys($var) === range(0, count($var) - 1);
  104. $r = [];
  105. foreach ($var as $key => $value) {
  106. $r[] = "$indent "
  107. .($indexed ? '' : $this->varExport($key).' => ')
  108. .$this->varExport($value, "{$indent} ");
  109. }
  110. return "[\n".implode(",\n", $r)."\n".$indent.']';
  111. case 'boolean':
  112. return $var ? 'true' : 'false';
  113. case 'integer':
  114. case 'double':
  115. return $var;
  116. default:
  117. return var_export($var, true);
  118. }
  119. }
  120. }