format type

Author: coderosh

README.md

@@ -1 +1,13 @@
 # JSON FROM TABLE
+
+Converts html tables into JSON
+
+## Usage
+
+Install via npm
+
+```npm
+    npm i jsonfromtable
+```
+
+You can convert

index.js

@@ -6,11 +6,12 @@ const toJson = require('./utils/tojson')
 const defaultOptions = {
   url: undefined,
   html: undefined,
-  selector: 'table'
+  selector: 'table',
+  format: 'jsobject'
 }
 
-const html = async (options, headers) => {
-  const { url, html, selector } = { ...defaultOptions, ...options }
+const htmlTableToJson = async (options, headers) => {
+  const { url, html, selector, format } = { ...defaultOptions, ...options }
 
   let data
 
@@ -23,27 +24,26 @@ const html = async (options, headers) => {
 
   const $ = cheerio.load(data)
 
-  let body = toJson($, 'table', headers)
+  if ($('table').html() === null) {
+    throw new Error(
+      `Please provide ${url ? 'url' : 'html'} which contains table`
+    )
+  }
+
+  let body = toJson($, selector, headers)
 
-  console.log(body)
+  if (format === 'json') {
+    return JSON.stringify(body)
+  } else {
+    return body
+  }
 }
 
-html(
-  {
-    html: `
-<table style="width:100%">
-  <tr>
-    <td>Jill</td>
-    <td>Smith</td>
-    <td>50</td>
-  </tr>
-  <tr>
-    <td>Eve</td>
-    <td>Jackson</td>
-    <td>94</td>
-  </tr>
-</table>
-`
-  },
-  ['Name', 'LastName', 'Age']
-)
+htmlTableToJson({
+  url: 'https://www.worldometers.info/coronavirus/',
+  format: 'json'
+}).then(data => {
+  console.log(data)
+})
+
+module.exports = htmlTableToJson

utils/tojson.js

@@ -3,6 +3,12 @@ const toJson = ($, tableSelector, _header) => {
   const body = []
   let header = []
 
+  if ($(tableSelector).html() === null) {
+    throw new Error(
+      `${tableSelector} is not a valid selector for table, please try again`
+    )
+  }
+
   // Add headers from table to header array
   $(`${tableSelector} th`).each((_, el) => {
     header.push(
@@ -26,7 +32,7 @@ const toJson = ($, tableSelector, _header) => {
   let d = {},
     j = 0
 
-  // For tds in table
+  // loop all tds in table
   $(`${tableSelector} td`).each((_, el) => {
     let val = $(el)
       .text()