fix: getLocationForJsonPath returns proper location for root path (#17)

package.json

@@ -50,7 +50,7 @@
   },
   "devDependencies": {
     "@stoplight/scripts": "~5.1",
-    "@stoplight/types": "^9.1.2",
+    "@stoplight/types": "^11.1.0",
     "@types/jest": "^24.0",
     "@types/js-yaml": "3.12.1",
     "@types/lodash": "^4.14.136",

src/__tests__/getLocationForJsonPath.spec.ts

@@ -14,6 +14,7 @@ describe('getLocationForJsonPath', () => {
 
     it.each`
       start      | end         | path
+      ${[0, 0]}  | ${[21, 0]}  | ${[]}
       ${[0, 0]}  | ${[0, 10]}  | ${['children', 0]}
       ${[2, 0]}  | ${[2, 31]}  | ${['children', 1, 'children', 0]}
       ${[2, 32]} | ${[2, 40]}  | ${['children', 1, 'children', 1, 'children', 0]}

src/getJsonPathForPosition.ts

@@ -1,7 +1,8 @@
 import { GetJsonPathForPosition, IPosition, JsonPath } from '@stoplight/types';
 import * as Unist from 'unist';
+import { MarkdownParserResult } from './types';
 
-export const getJsonPathForPosition: GetJsonPathForPosition<Unist.Node> = ({ ast }, position) => {
+export const getJsonPathForPosition: GetJsonPathForPosition<MarkdownParserResult> = ({ ast }, position) => {
   const path: JsonPath = [];
   findNodeAtPosition(ast, position, path);
   return path;

src/getLocationForJsonPath.ts

@@ -1,10 +1,11 @@
 import { GetLocationForJsonPath, JsonPath } from '@stoplight/types';
-import _get = require('lodash/get');
+import { get as _get } from 'lodash';
 import * as Unist from 'unist';
+import { MarkdownParserResult } from './types';
 
-export const getLocationForJsonPath: GetLocationForJsonPath<Unist.Node> = ({ ast }, path: JsonPath) => {
-  const data = _get(ast, path) as Unist.Node | undefined;
-  if (data === undefined) return;
+export const getLocationForJsonPath: GetLocationForJsonPath<MarkdownParserResult> = ({ ast }, path: JsonPath) => {
+  const data = path.length === 0 ? ast : (_get(ast, path) as Unist.Node | undefined);
+  if (data === void 0) return;
 
   return {
     range: {

yarn.lock

@@ -760,10 +760,10 @@
     tslib "1.9.3"
     typedoc "0.14.2"
 
-"@stoplight/types@^9.1.2":
-  version "9.1.2"
-  resolved "https://registry.yarnpkg.com/@stoplight/types/-/types-9.1.2.tgz#0b2ed2032ae901491543b9dd19309ff3a0acf12a"
-  integrity sha512-eNeWwTZ1E47mrprEdgtOutovVKxYTbm8k/B7Lh4JAusGoiHbMmzVT+IDdayBpOCzq+qKLzuP0z3IZo74IXaaEg==
+"@stoplight/types@^11.1.0":
+  version "11.1.0"
+  resolved "https://registry.yarnpkg.com/@stoplight/types/-/types-11.1.0.tgz#b29ecd8c8ad27bb564c7820c99f93c27f1d11bf1"
+  integrity sha512-z4VqnYIDTcj9R0f8AIDitBWkiAKKJ4W/Z5GJz/3fnBZBa/TD9Oak0xHbicXFXUvqTVCv/7D32PjPYiUX6W/EfQ==
   dependencies:
     "@types/json-schema" "^7.0.3"