✨ 代码生成时可以指定生成的文件

ballcat-codegen/ballcat-codegen-frontend/src/api/gen/templateinfo.js

@@ -7,6 +7,13 @@ export function getObj(id) {
   })
 }
 
+export function getList(groupId) {
+  return axios({
+    url: '/gen/template/info/list/' + groupId,
+    method: 'get'
+  })
+}
+
 export function putObj(obj) {
   return axios({
     url: '/gen/template/info',

ballcat-codegen/ballcat-codegen-frontend/src/main.js

@@ -20,7 +20,8 @@ import {
   Breadcrumb,
   Modal,
   Radio,
-  Descriptions
+  Descriptions,
+  Checkbox
 } from 'ant-design-vue'
 import App from './App.vue'
 import router from './router'
@@ -47,6 +48,7 @@ Vue.use(Dropdown)
 Vue.use(Breadcrumb)
 Vue.use(Modal)
 Vue.use(Radio)
+Vue.use(Checkbox)
 Vue.use(Descriptions)
 Vue.prototype.$message = message
 

ballcat-codegen/ballcat-codegen-frontend/src/mixins/formMixin.js

@@ -70,7 +70,6 @@ export default {
       // 钩子函数 处理提交之前处理的事件
       this.beforeStartSubmit()
       const reqFunction = this.reqFunctions[this.formAction]
-      console.log(reqFunction)
       this.form.validateFields((err, values) => {
         if (!err) {
           this.submitLoading = true

ballcat-codegen/ballcat-codegen-frontend/src/views/gen/codegen/GenerateModal.vue

@@ -28,6 +28,28 @@
               </a-select-option>
             </a-select>
           </a-form-item>
+          <a-form-item label="生成文件选择">
+            <div :style="{ borderBottom: '1px solid #E9E9E9' }">
+              <a-checkbox :indeterminate="indeterminate" :checked="checkAll" @change="onCheckAllChange">
+                Check all
+              </a-checkbox>
+            </div>
+            <a-checkbox-group
+              v-decorator="['templateFileIds', { initialValue: templateFileIds }]"
+              @change="onTemplateFileIdsChange"
+              style="width: 100%;"
+            >
+              <a-row>
+                <template v-for="item in templateFiles">
+                  <a-col :span="12">
+                    <a-checkbox :value="item.directoryEntryId">
+                      {{ item.title }}
+                    </a-checkbox>
+                  </a-col>
+                </template>
+              </a-row>
+            </a-checkbox-group>
+          </a-form-item>
         </a-col>
         <a-col :span="16">
           <a-divider orientation="left">
@@ -61,6 +83,7 @@
 import { FormModalMixin } from '@/mixins'
 import { getSelectData } from '@/api/gen/templategroup'
 import { getProperties } from '@/api/gen/templateproperty'
+import { getList as getTemplateFiles } from '@/api/gen/templateinfo'
 import { generate } from '@/api/gen/generate'
 
 export default {
@@ -84,7 +107,13 @@ export default {
       defaultTemplateGroupId: 1,
       tableNames: [],
       templateGroupSelectData: [],
-      properties: []
+      properties: [],
+
+      templateFiles: [],
+      templateFileIds: [],
+      checkedList: [],
+      indeterminate: false,
+      checkAll: true
     }
   },
   mounted() {
@@ -104,6 +133,10 @@ export default {
       getProperties(templateGroupId).then(res => {
         this.properties = res.data
       })
+      getTemplateFiles(templateGroupId).then(res => {
+        this.templateFiles = res.data
+        this.templateFileIds = this.templateFiles.map(x => x.directoryEntryId)
+      })
     },
     handleOk() {
       // 钩子函数 处理提交之前处理的事件
@@ -141,6 +174,19 @@ export default {
     submitDataProcess(data) {
       data.tableNames = this.tableNames
       return data
+    },
+    onTemplateFileIdsChange(checkedList) {
+      this.indeterminate = !!checkedList.length && checkedList.length < this.templateFileIds.length
+      this.checkAll = checkedList.length === this.templateFileIds.length
+    },
+    onCheckAllChange(e) {
+      this.indeterminate = false
+      this.checkAll = e.target.checked
+      setTimeout(() => {
+        this.$nextTick(function() {
+          this.form.setFieldsValue({ templateFileIds: this.checkAll ? this.templateFileIds : [] })
+        })
+      }, 0)
     }
   }
 }

ballcat-codegen/ballcat-codegen-frontend/src/views/gen/codegen/GeneratePage.vue

@@ -147,9 +147,8 @@ export default {
     },
     multiGenerate() {
       const tableNames = this.selectedRowKeys
-      console.log(tableNames)
       if (tableNames && tableNames.length > 0) {
-        this.handleGenerate(tableNames)
+        this.$refs.generateModal.show(tableNames)
       } else {
         this.$message.warning('至少选中一张表')
       }