antlink 0.0.2 → 0.0.4

package/README.md

@@ -12,17 +12,29 @@
 **Installing AntLink**
 
 ```shell
-$ npm install antlink -g
+$ npm install antlink -g   # 全局安装
 ```
 
-**Initialize service**
 
 ```shell
-cd E:\antwork
-ant init 
+cd E:\antwork          # 蚁巢位置路径
+ant init               # 建巢（获得安全码）
+ant reset [code]       # 重置蚁巢（需安全码）
 ```
 
 
+```shell
+ant nest start         # 启动蚁巢
+ant nest stop          # 停止蚁巢
+ant nest restart       # 重启蚁巢
+ant nest status        # 运行状态
+```
+
+
+
+
+
+
 
 [npm-v-1]: https://badgen.net/npm/v/antlink
 [npm-v-2]: https://img.shields.io/npm/v/antlink

package/index.js

@@ -1,4 +1,10 @@
 'use strict';
-module.exports = {
-	console.log('Hello, I am an intelligent agent');
-};
+const plain = require('./lib/utils/plain');
+module.exports = async()=>{
+	try{
+		const res = await plain(__dirname);
+		console.log(res);
+	}catch(err){
+		console.log(err);
+	}
+}

package/lib/comm.js

@@ -2,13 +2,14 @@
 fkgConmm();
 async function fkgConmm() {
     const arg = process.argv;
-    const {version} = require('../package.json');
-    if (arg.length < 3) console.log(version);
+    const {version} = require('../package.json');
+	const versionFunt = ()=>{console.log('v'+version);}
+    if (arg.length < 3) versionFunt();
     if (arg.length > 2) {
-        const arr = require('./json/command.json');
+        const arr = require('./json/cmds.json');
         const name = arg[2].toLowerCase(); // 字母转小写
         if (name=='-v') {
-            console.log(version);
+            versionFunt();
         } else if(arr.includes(name)){
             await require('./order/'+name)(arg.slice(3)); 
         }else {

package/lib/json/cmds.json

@@ -0,0 +1 @@
+["init","nest","reset","info"]

package/lib/order/init.js

@@ -1,4 +1,45 @@
 'use strict';
-module.exports = async(arg)=>{
-	console.log(arg);
+const oss = require('oss-sdk');
+const log = require('../utils/logs');
+module.exports = async()=>{
+	// 判断是否初始化
+	const dir = oss.dir(process.cwd());
+	const res = await dir.list();
+	if(res){
+		if(res.length>0){
+			// 当前文件夹存在文件，需要清空
+			log.r('The current folder is not empty')
+			log.r('This folder must be cleared!')
+		}else{
+			// 开始初始化
+			const dirAdd = await dir.add('nest','data','skills','scripts');
+			if(dirAdd){
+				// 新建配置文件
+				let obj = {};
+				const {version} = require('../../package.json');
+				obj.version = version;
+				obj.initTime = new Date().getTime();
+				const rrid = require('rrid');
+				obj.appId = rrid.key(32);
+				const scode = rrid.key(16);
+				obj.appKey = rrid.ekey(obj.appId+obj.initTime+scode,obj.appId+obj.initTime+scode);
+				const file = oss.file(process.cwd());
+				const fileAdd = await file.add('config.ant',JSON.stringify(obj));
+				if(fileAdd){
+					// 初始化成功
+					log.g('Initialization successful!');
+					log.y('Security Code：'+scode);
+					log.y('Keep the security code safe')
+				}else{
+					log.r('Initialization failed!');
+					await dir.clear();
+				}
+			}else{
+				//初始化失败：无法创建目录
+				log.r('Initialization failed: unable to create directory')
+			}
+		}
+	}else{
+		log.r('Unknown error!')
+	}
 }

package/lib/order/nest/index.js

@@ -0,0 +1,15 @@
+'use strict';
+const log = require('../../utils/logs');
+module.exports = async(arg)=>{
+	if(arg.length===0){
+		log.r('Missing parameter')
+	}else{
+		const cmd = arg[0];
+		const arr = ['start','stop','restart','status'];
+		if(arr.includes(cmd)) {
+			await require('./'+cmd)(arg.slice(1));
+		}else{
+			log.y('Command not found：ant nest '+cmd)
+		}
+	}
+}

package/lib/order/nest/restart.js

@@ -0,0 +1,4 @@
+'use strict';
+module.exports = async(arg)=>{
+	console.log(arg);
+}

package/lib/order/nest/start.js

@@ -0,0 +1,4 @@
+'use strict';
+module.exports = async()=>{
+	console.log(arg);
+}

package/lib/order/nest/status.js

@@ -0,0 +1,4 @@
+'use strict';
+module.exports = async(arg)=>{
+	console.log(arg);
+}

package/lib/order/nest/stop.js

@@ -0,0 +1,4 @@
+'use strict';
+module.exports = async(arg)=>{
+	console.log(arg);
+}

package/lib/order/reset.js

@@ -0,0 +1,17 @@
+'use strict';
+const oss = require('oss-sdk');
+const scode = require('../utils/scode');
+const log = require('../utils/logs');
+module.exports = async(arg)=>{
+	const res = await scode(arg[0]);
+	if(res){
+		// 安全码验证通过
+		const dir = oss.dir(process.cwd());
+		const clear = await dir.clear();
+		if(clear){
+			log.g('Reset successful!');
+		}else{
+			log.r('Reset failure!');
+		}
+	}
+}

package/lib/utils/input.js

@@ -0,0 +1,35 @@
+const logs = require('./logs')
+const readline = require('readline');
+module.exports = (...arg) => {
+	let arr = [];
+	return new Promise(async (resolve) => {
+		if (arg.length > 0) {
+			try{
+				const ps = readline.createInterface({
+				    input: process.stdin,  // 输入源：终端标准输入
+				    output: process.stdout // 输出源：终端标准输出
+				});
+				process.stdin.setEncoding('utf8');
+				const inputs = (text)=>{
+					return new Promise((resolves) => {
+						ps.question('\033[33m'+text, (answer) => {
+							console.log('\033[0m');
+							resolves(answer.trim());
+						});
+					});
+				}
+				for (let i = 0; i < arg.length; i++) {
+					arr.push(await inputs(arg[i]));
+					if(arg.length-1===i) ps.close();
+				}
+				resolve(arr);
+			}catch(err){
+				logs.r(err);
+				resolve(arr);
+			}
+		} else {
+			resolve(arr);
+			logs.r('Please enter parameters');
+		}
+	})
+}

package/lib/utils/logs.js

@@ -0,0 +1,43 @@
+const logs = (d)=> {console.log(d);}
+module.exports = {
+    b(...d){
+		for (var i = 0; i < d.length; i++) {
+			logs('\033[30m'+d[i]+'\033[0m');
+		}
+    },
+    r(...d){
+		for (var i = 0; i < d.length; i++) {
+			logs('\033[31m'+d[i]+'\033[0m');
+		}
+    },
+    g(...d){
+		for (var i = 0; i < d.length; i++) {
+			logs('\033[32m'+d[i]+'\033[0m');
+		}		
+    },
+    y(...d){
+		for (var i = 0; i < d.length; i++) {
+			logs('\033[33m'+d[i]+'\033[0m');
+		}
+    },
+    blue(...d){
+		for (var i = 0; i < d.length; i++) {
+			logs('\033[34m'+d[i]+'\033[0m');
+		}
+    },
+    p(...d){
+		for (var i = 0; i < d.length; i++) {
+			logs('\033[35m'+d[i]+'\033[0m');
+		}
+    },
+    sblue(...d){
+		for (var i = 0; i < d.length; i++) {
+			logs('\033[36m'+d[i]+'\033[0m');
+		}
+    },
+    w(...d){
+		for (var i = 0; i < d.length; i++) {
+			logs('\033[37m'+d[i]+'\033[0m');
+		}
+    }
+}

package/lib/utils/plain.js

@@ -0,0 +1,52 @@
+const path = require('path');
+const fs = require('fs').promises;
+
+/**
+ * 生成文件夹目录结构纯文本
+ * @param {string} dirPath - 目标文件夹路径
+ * @param {Array<string>} ignoreDirs - 需要忽略的文件夹列表
+ * @returns {Promise<string>} 目录结构纯文本
+ */
+module.exports = async (dirPath) => {
+  const result = [];
+  const ignoreDirs = ['node_modules','test.js','package-lock.json'];
+  // 递归遍历目录
+  async function traverse(currentPath, prefix = '') {
+    try {
+      const entries = await fs.readdir(currentPath, { withFileTypes: true });
+      
+      // 过滤 + 排序
+      const filteredEntries = entries
+        .filter(entry => !ignoreDirs.includes(entry.name))
+        .sort((a, b) => {
+          if (a.isDirectory() && !b.isDirectory()) return -1;
+          if (!a.isDirectory() && b.isDirectory()) return 1;
+          return a.name.localeCompare(b.name);
+        });
+
+      for (let i = 0; i < filteredEntries.length; i++) {
+        const entry = filteredEntries[i];
+        const isLast = i === filteredEntries.length - 1;
+        const connector = isLast ? '└── ' : '├── ';
+        const fullPath = path.join(currentPath, entry.name);
+
+        result.push(prefix + connector + entry.name);
+
+        if (entry.isDirectory()) {
+          const newPrefix = prefix + (isLast ? '    ' : '│   ');
+          await traverse(fullPath, newPrefix);
+        }
+      }
+    } catch (err) {
+      throw new Error(`遍历目录失败: ${err.message}`);
+    }
+  }
+
+  // 安全获取根目录名
+  const resolvedPath = path.resolve(dirPath);
+  const rootName = path.basename(resolvedPath);
+  result.push(rootName);
+
+  await traverse(dirPath);
+  return result.join('\n');
+};

package/lib/utils/scode.js

@@ -0,0 +1,37 @@
+'use strict';
+const oss = require('oss-sdk');
+const log = require('./logs');
+const input = require('./input');
+module.exports = async(scode)=>{
+	let ok = null;
+	if(scode===undefined) {
+		scode = await input('Enter Security Code：');
+		scode = scode[0];
+	}
+	if(scode) {
+		// 验证安全码
+		const file = oss.file(process.cwd());
+		let conf = await file.get('config.ant');
+		if(conf){
+			try {
+				conf = JSON.parse(conf);
+				const rrid = require('rrid');
+				const key = rrid.ekey(conf.appId+conf.initTime+scode,conf.appId+conf.initTime+scode);
+				if(key===conf.appKey){
+					// 验证码通过
+					ok = conf;
+				}else{
+					log.r('Security code error');
+				}
+			}catch(err){
+				log.r(err);
+			}
+		}else{
+			// 配置文件不存在
+			log.r('The configuration file does not exist')
+		}
+	}else{
+		log.r('The security code does not exist');
+	}
+	return ok;
+}

package/package.json

@@ -1,17 +1,28 @@
 {
 	"name": "antlink",
-	"version": "0.0.2",
-	"description": "The framework has just been formed.",
+	"version": "0.0.4",
+	"description": "Ant Nest AI Framework.",
 	"main": "index.js",
 	"author": "weduu",
 	"license": "MIT",
 	"bin": {
 		"ant": "bin/ant"
 	},
+	"keywords": [
+		"ai",
+		"cli",
+		"agent", 
+		"tenant"
+	],
 	"files": [
 		"lib/",
 		"bin/",
 		"index.js",
+		"plain.md",
 		"README.md"
-	]
-}
+	],
+	"dependencies": {
+		"oss-sdk": "^1.0.5",
+		"rrid": "^1.0.7"
+	}
+}

package/plain.md

@@ -0,0 +1,29 @@
+AntLink
+├── bin
+│   └── ant
+├── lib
+│   ├── ants
+│   │   ├── models
+│   │   ├── plugins
+│   │   ├── skills
+│   │   └── tools
+│   ├── json
+│   │   └── cmds.json
+│   ├── order
+│   │   ├── nest
+│   │   │   └── index.js
+│   │   ├── init.js
+│   │   └── reset.js
+│   ├── utils
+│   │   ├── input.js
+│   │   ├── logs.js
+│   │   ├── plain.js
+│   │   └── scode.js
+│   ├── web
+│   │   ├── api
+│   │   └── h5
+│   └── comm.js
+├── index.js
+├── package.json
+├── plain.md
+└── README.md

package/lib/json/command.json

@@ -1 +0,0 @@
-["init","start","stop","kill","reset"]