happy-sql 1.0.1 → 1.1.0

package/ChangeLog

@@ -1,3 +1,8 @@
+2026.07.27, v1.1.0
+
+feature:
+- f6446c3 happy-sql: COUNT
+
 2026.07.27, v1.0.1
 
 feature:

package/README.md

@@ -17,6 +17,17 @@ SQL to JS AST parser and printer.
 npm i happy-sql --save
 ```
 
+## API
+
+```js
+const source = `select(count('*'), from(users))`;
+
+convertJsToSql(source);
+// returns
+'SELECT COUNT(*)\nFROM users\n');
+
+```
+
 ## License
 
 MIT

package/lib/parser/parser.js

@@ -40,14 +40,12 @@ export const parseSqlNode = (source) => {
     
     const elements = [];
     
-    for (let i = 0; i < cst.statements.length; i++) {
-        const stmt = cst.statements[i];
-        
-        if (stmt.type === 'empty')
+    for (const [i, statement] of cst.statements.entries()) {
+        if (statement.type === 'empty')
             continue;
         
-        const stmtNode = convertClause(stmt);
-        const leading = !i ? cst.leading : stmt.leading;
+        const stmtNode = convertClause(statement);
+        const leading = i ? statement.leading : cst.leading;
         
         elements.push(wrapSection(stmtNode, getSectionName(leading)));
     }

package/lib/parser/visitors/clause/count/count.js

@@ -0,0 +1,23 @@
+import {types} from '@putout/babel';
+
+const {
+    stringLiteral,
+    callExpression,
+    identifier,
+} = types;
+
+export const isCount = (column) => {
+    return column.type === 'func_call' && /count/i.test(column.name.name);
+};
+
+export const convertCount = (column) => {
+    const {items} = column.args.expr.args;
+    const args = [];
+    
+    for (const item of items) {
+        if (item.type === 'all_columns')
+            args.push(stringLiteral('*'));
+    }
+    
+    return callExpression(identifier('count'), args);
+};

package/lib/parser/visitors/clause/count/fixture/count-fix.js

@@ -0,0 +1,3 @@
+[
+    select(count('*'), from(users)),
+];

package/lib/parser/visitors/clause/count/fixture/count.sql

@@ -0,0 +1 @@
+SELECT COUNT(*) FROM users

package/lib/parser/visitors/clause/select/select.js

@@ -3,6 +3,7 @@ import {convertOperand} from '#parser/clause/operand';
 import {convertFrom} from '#parser/clause/from';
 import {convertWhere} from '#parser/clause/where';
 import {findClause} from '#parser/clause/find';
+import {convertCount, isCount} from '#parser/clause/count';
 
 const {
     identifier,
@@ -11,11 +12,14 @@ const {
     callExpression,
 } = types;
 
-const convertColumn = (col) => {
-    if (col.type === 'alias')
-        return assignmentExpression('=', identifier(col.alias.name), convertOperand(col.expr));
+const convertColumn = (column) => {
+    if (isCount(column))
+        return convertCount(column);
     
-    return identifier(col.name);
+    if (column.type === 'alias')
+        return assignmentExpression('=', identifier(column.alias.name), convertOperand(column.expr));
+    
+    return identifier(column.name);
 };
 
 const convertColumns = (columns) => {

package/lib/printer/visitors/call-expression/call-expression.js

@@ -2,9 +2,10 @@ import {clauses} from './clauses/clauses.js';
 
 export const CallExpression = (path, printer) => {
     const {name} = path.node.callee;
+    const visit = clauses[name];
     
-    if (!clauses[name])
+    if (!visit)
         throw Error(`${name} not supported yet`);
     
-    clauses[name](path, printer);
+    visit(path, printer);
 };

package/lib/printer/visitors/call-expression/clauses/clauses.js

@@ -9,10 +9,12 @@ import {update} from './update/update.js';
 import {set} from './set/set.js';
 import {del} from './delete/delete.js';
 import {section} from './section/section.js';
+import {count} from './count/count.js';
 
 export const clauses = {
-    select,
+    count,
     from,
+    select,
     where,
     insert,
     into,

package/lib/printer/visitors/call-expression/clauses/count/count.js

@@ -0,0 +1,7 @@
+export const count = (path, {write, traverse}) => {
+    const [first] = path.get('arguments');
+    write('COUNT');
+    write('(');
+    traverse(first);
+    write(')');
+};

package/lib/printer/visitors/call-expression/clauses/count/fixture/count.sql

@@ -0,0 +1,2 @@
+SELECT COUNT(*)
+FROM users

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "happy-sql",
-  "version": "1.0.1",
+  "version": "1.1.0",
   "type": "module",
   "description": "sql to js ast parser and printer",
   "exports": {
@@ -18,6 +18,7 @@
     "#parser/test": "./test/create-parser-test.js",
     "#parser/clause": "./lib/parser/visitors/clause/index.js",
     "#parser/clause/select": "./lib/parser/visitors/clause/select/select.js",
+    "#parser/clause/count": "./lib/parser/visitors/clause/count/count.js",
     "#parser/clause/from": "./lib/parser/visitors/clause/from/from.js",
     "#parser/clause/where": "./lib/parser/visitors/clause/where/where.js",
     "#parser/clause/insert": "./lib/parser/visitors/clause/insert/insert.js",
@@ -79,6 +80,7 @@
     "@putout/babel": "^5.7.3",
     "@putout/operator-json": "^3.4.0",
     "@putout/printer": "^18.17.0",
+    "just-camel-case": "^6.2.0",
     "redstd": "^1.0.2",
     "sql-parser-cst": "^0.42.1"
   }