ExportSeedCommand.php 5.0 KB

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