cocos2d-cli 1.4.0 → 1.5.0

package/src/commands/add.js

@@ -1,272 +1,176 @@
 /**
- * add 命令 - 添加节点（支持场景和预制体）
+ * add 命令 - 添加节点
  */
 
-const { loadScene, saveScene, buildMaps, refreshEditor, isPrefab, generateFileId, getPrefabRootIndex } = require('../lib/fire-utils');
-const { parseOptions, outputError, outputSuccess } = require('../lib/utils');
-const { createNodeData } = require('../lib/node-utils');
-const { createComponent } = require('../lib/components');
+const path = require('path');
+const fs = require('fs');
+const { CCNode, CCSceneAsset, CCPrefab, CCCanvas, CCWidget, CCSprite, CCLabel, CCButton } = require('../lib/cc');
+const { buildTree } = require('../lib/node-utils');
+const { loadScriptMap, isPrefab } = require('../lib/fire-utils');
+const { generateCompressedUUID } = require('../lib/utils');
 
 /**
- * 获取根节点 PrefabInfo 的索引
+ * 解析命令行选项
  */
-function getRootPrefabInfoIndex(data) {
-    for (let i = data.length - 1; i >= 0; i--) {
-        if (data[i].__type__ === 'cc.PrefabInfo') {
-            const rootRef = data[i].root?.__id__;
-            if (rootRef === 1) return i;
+function parseOptions(args, startIndex) {
+    const options = {};
+    for (let i = startIndex; i < args.length; i++) {
+        const arg = args[i];
+        if (arg.startsWith('--')) {
+            const eqIndex = arg.indexOf('=');
+            if (eqIndex > 0) {
+                const key = arg.substring(2, eqIndex);
+                let value = arg.substring(eqIndex + 1);
+                if (!isNaN(value) && value !== '') {
+                    value = parseFloat(value);
+                }
+                options[key] = value;
+            }
         }
     }
-    return data.length - 1;
+    return options;
 }
 
 /**
- * 获取节点子树的结束索引
+ * 创建组件
  */
-function getSubtreeEndIndex(data, nodeIndex) {
-    const node = data[nodeIndex];
-    if (!node) return nodeIndex;
-    
-    let lastIndex = nodeIndex;
-    
-    if (node._children) {
-        for (const childRef of node._children) {
-            const childEnd = getSubtreeEndIndex(data, childRef.__id__);
-            lastIndex = Math.max(lastIndex, childEnd);
-        }
-    }
-    
-    if (node._components) {
-        for (const compRef of node._components) {
-            lastIndex = Math.max(lastIndex, compRef.__id__);
-        }
-    }
-    
-    if (node._prefab && nodeIndex !== 1) {
-        lastIndex = Math.max(lastIndex, node._prefab.__id__);
+function createComponent(type) {
+    switch (type.toLowerCase()) {
+        case 'canvas': return new CCCanvas();
+        case 'widget': return new CCWidget();
+        case 'sprite': return new CCSprite();
+        case 'label': return new CCLabel();
+        case 'button': return new CCButton();
+        default: return null;
     }
-    
-    return lastIndex;
 }
 
 /**
- * 重建所有 __id__ 引用
+ * 查找节点
  */
-function rebuildReferencesForInsert(data, insertIndex, count) {
-    const indexMap = {};
+function findNode(root, path) {
+    if (!path) return root;
     
-    for (let oldIndex = 0; oldIndex < data.length; oldIndex++) {
-        if (oldIndex < insertIndex) {
-            indexMap[oldIndex] = oldIndex;
-        } else {
-            indexMap[oldIndex] = oldIndex + count;
-        }
-    }
+    const parts = path.split('/').filter(p => p);
+    if (parts.length === 0) return root;
     
-    function updateRef(obj) {
-        if (!obj || typeof obj !== 'object') return;
-        
-        if (obj.__id__ !== undefined) {
-            const oldId = obj.__id__;
-            if (indexMap[oldId] !== undefined) {
-                obj.__id__ = indexMap[oldId];
-            }
-        } else {
-            for (const key of Object.keys(obj)) {
-                updateRef(obj[key]);
-            }
-        }
+    let current = root;
+    
+    // 如果路径以根节点名称开始，跳过
+    if (parts[0] === root._name) {
+        parts.shift();
     }
     
-    for (const item of data) {
-        updateRef(item);
+    for (const part of parts) {
+        if (!current._children || current._children.length === 0) return null;
+        const found = current._children.find(c => c._name === part);
+        if (!found) return null;
+        current = found;
     }
     
-    return indexMap;
+    return current;
 }
 
 function run(args) {
     if (args.length < 3) {
-        outputError('用法: cocos2d-cli add <场景.fire | 预制体.prefab> <父节点索引> <节点名称> [选项]');
+        console.log(JSON.stringify({ error: '用法: cocos2d-cli add <场景.fire|预制体.prefab> <父节点路径> <节点名称> [--type=组件类型] [--x=N] [--y=N]' }));
         return;
     }
     
     const filePath = args[0];
-    const parentRef = args[1];
+    const parentPath = args[1];
     const nodeName = args[2];
-    
     const options = parseOptions(args, 3);
     
-    // 转换数值类型
-    if (options.x) options.x = parseFloat(options.x) || 0;
-    if (options.y) options.y = parseFloat(options.y) || 0;
-    if (options.width) options.width = parseFloat(options.width) || 0;
-    if (options.height) options.height = parseFloat(options.height) || 0;
-    if (options.at !== undefined) options.at = parseInt(options.at);
-    if (options.active !== undefined) options.active = options.active !== 'false';
-    if (options.fontSize) options.fontSize = parseInt(options.fontSize) || 40;
+    const ext = path.extname(filePath).toLowerCase();
     
     try {
-        const data = loadScene(filePath);
-        const { prefab } = buildMaps(data);
+        let root;
+        let asset;
+        const json = JSON.parse(fs.readFileSync(filePath, 'utf8'));
         
-        if (!/^\d+$/.test(parentRef)) {
-            outputError('父节点必须使用数字索引，请先用 tree 命令查看节点索引');
+        if (ext === '.fire') {
+            asset = CCSceneAsset.fromJSON(json);
+            root = asset._scene;
+        } else if (ext === '.prefab') {
+            asset = CCPrefab.fromJSON(json);
+            root = asset._root;
+        } else {
+            console.log(JSON.stringify({ error: '不支持的文件类型，仅支持 .fire 和 .prefab' }));
             return;
         }
         
-        const parentIndex = parseInt(parentRef);
+        const parent = findNode(root, parentPath);
         
-        if (parentIndex < 0 || parentIndex >= data.length || !data[parentIndex]) {
-            outputError(`无效的节点索引: ${parentRef}`);
+        if (!parent) {
+            console.log(JSON.stringify({ error: `父节点不存在: ${parentPath}` }));
             return;
         }
         
-        const parentNode = data[parentIndex];
-        const isRootChild = prefab && parentIndex === 1;
-        
-        // 确定插入位置
-        let insertIndex;
+        // 创建节点
+        const node = new CCNode(nodeName);
         
-        if (!parentNode._children || parentNode._children.length === 0) {
-            insertIndex = parentIndex + 1;
-        } else {
-            const targetPosition = options.at >= 0 ? options.at : parentNode._children.length;
-            
-            if (targetPosition === 0) {
-                insertIndex = parentNode._children[0].__id__;
-            } else if (targetPosition >= parentNode._children.length) {
-                const lastChildRef = parentNode._children[parentNode._children.length - 1];
-                insertIndex = getSubtreeEndIndex(data, lastChildRef.__id__) + 1;
-            } else {
-                insertIndex = parentNode._children[targetPosition].__id__;
-            }
+        // 应用属性
+        if (options.x !== undefined) node._trs.array[0] = parseFloat(options.x);
+        if (options.y !== undefined) node._trs.array[1] = parseFloat(options.y);
+        if (options.width !== undefined) node._contentSize.width = parseFloat(options.width);
+        if (options.height !== undefined) node._contentSize.height = parseFloat(options.height);
+        if (options.scaleX !== undefined) node._trs.array[7] = parseFloat(options.scaleX);
+        if (options.scaleY !== undefined) node._trs.array[8] = parseFloat(options.scaleY);
+        if (options.rotation !== undefined) {
+            node._trs.array[5] = parseFloat(options.rotation) * Math.PI / 180;
+            node._eulerAngles.z = parseFloat(options.rotation);
+        }
+        if (options.active !== undefined) {
+            node._active = options.active !== 'false' && options.active !== false;
         }
         
-        let newNodeIndex;
-        
-        if (prefab) {
-            // 预制体模式
-            const rootPrefabInfoOldIdx = getRootPrefabInfoIndex(data);
-            const rootPrefabInfo = data[rootPrefabInfoOldIdx];
-            
-            // 创建节点
-            const nodeData = createNodeData(nodeName, parentIndex, options);
-            
-            // 创建组件
-            let compData = null;
-            if (options.type) {
-                compData = createComponent(options.type, insertIndex);
-                if (compData) {
-                    if (options.type === 'label') {
-                        if (options.fontSize) {
-                            compData._fontSize = options.fontSize;
-                            compData._lineHeight = options.fontSize;
-                        }
-                        if (options.string) {
-                            compData._string = options.string;
-                            compData._N$string = options.string;
-                        }
+        // 添加组件
+        if (options.type) {
+            const comp = createComponent(options.type);
+            if (comp) {
+                comp.node = node;
+                node._components = [comp];
+                
+                // 组件特殊属性
+                if (options.type.toLowerCase() === 'label') {
+                    if (options.string) {
+                        comp._string = options.string;
+                        comp['_N$string'] = options.string;
                     }
-                }
-            }
-            
-            // 创建 PrefabInfo
-            const prefabInfo = {
-                "__type__": "cc.PrefabInfo",
-                "root": { "__id__": 1 },
-                "asset": { "__id__": 0 },
-                "fileId": generateFileId(),
-                "sync": false
-            };
-            
-            const itemsToInsert = [nodeData];
-            if (compData) itemsToInsert.push(compData);
-            itemsToInsert.push(prefabInfo);
-            
-            if (isRootChild) {
-                if (insertIndex > rootPrefabInfoOldIdx) {
-                    insertIndex--;
-                }
-                data.splice(rootPrefabInfoOldIdx, 1);
-            }
-            
-            for (let i = 0; i < itemsToInsert.length; i++) {
-                data.splice(insertIndex + i, 0, itemsToInsert[i]);
-            }
-            
-            rebuildReferencesForInsert(data, insertIndex, itemsToInsert.length);
-            
-            newNodeIndex = insertIndex;
-            
-            if (compData) {
-                compData.node = { "__id__": newNodeIndex };
-                nodeData._components.push({ "__id__": newNodeIndex + 1 });
-                nodeData._prefab = { "__id__": newNodeIndex + 2 };
-            } else {
-                nodeData._prefab = { "__id__": newNodeIndex + 1 };
-            }
-            
-            if (isRootChild) {
-                data.push(rootPrefabInfo);
-                data[1]._prefab = { "__id__": data.length - 1 };
-            }
-            
-        } else {
-            // 场景模式
-            const newNode = createNodeData(nodeName, parentIndex, options);
-            
-            const itemsToInsert = [newNode];
-            
-            let compData = null;
-            if (options.type) {
-                compData = createComponent(options.type, insertIndex);
-                if (compData) {
-                    if (options.type === 'label') {
-                        if (options.fontSize) {
-                            compData._fontSize = options.fontSize;
-                            compData._lineHeight = options.fontSize;
-                        }
-                        if (options.string) {
-                            compData._string = options.string;
-                            compData._N$string = options.string;
-                        }
+                    if (options.fontSize) {
+                        comp._fontSize = parseInt(options.fontSize);
+                        comp._lineHeight = parseInt(options.fontSize);
                     }
                 }
-                if (compData) itemsToInsert.push(compData);
-            }
-            
-            for (let i = 0; i < itemsToInsert.length; i++) {
-                data.splice(insertIndex + i, 0, itemsToInsert[i]);
-            }
-            
-            rebuildReferencesForInsert(data, insertIndex, itemsToInsert.length);
-            
-            newNodeIndex = insertIndex;
-            
-            if (compData) {
-                compData.node = { "__id__": newNodeIndex };
-                newNode._components.push({ "__id__": newNodeIndex + 1 });
             }
         }
         
-        // 更新父节点的 _children
-        if (!parentNode._children) parentNode._children = [];
-        const insertPosition = options.at >= 0 ? options.at : parentNode._children.length;
-        parentNode._children.splice(insertPosition, 0, { "__id__": newNodeIndex });
+        // 添加节点到父节点
+        parent._children = parent._children || [];
+        parent._children.push(node);
+        node._parent = parent;
+        
+        // 场景节点需要 _id，预制体需要 PrefabInfo
+        if (ext === '.fire') {
+            node._id = generateCompressedUUID();
+        } else if (ext === '.prefab') {
+            const { CCPrefabInfo } = require('../lib/cc');
+            node._prefab = new CCPrefabInfo();
+        }
         
-        saveScene(filePath, data);
-        refreshEditor(filePath);
+        // 保存
+        const data = asset.toJSON();
+        fs.writeFileSync(filePath, JSON.stringify(data, null, 2), 'utf8');
         
-        outputSuccess({ 
-            nodeIndex: newNodeIndex,
-            name: nodeName,
-            parent: parentRef,
-            type: prefab ? 'prefab' : 'scene'
-        });
+        // 输出节点树
+        const scriptMap = loadScriptMap(filePath);
+        const prefab = isPrefab(data);
+        const startIndex = prefab ? 0 : 1;
+        console.log(buildTree(data, scriptMap, startIndex).trim());
         
     } catch (err) {
-        outputError(err.message);
+        console.log(JSON.stringify({ error: err.message }));
     }
 }