ExportSeedCommand.php 6.0 KB

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