Как посчитать количество функций в коде JS?

Question

[Antonio Solo]

Столкнулся с тестовым заданием (знакомому прислал рекрутер), которое поставило меня в тупик.

Выполните задание на языке JavaScript или на любом другом языке программирования.

Программа должна получать на вход *.js-файл (пример файла
https://github.com/SAP-samples/s4hana-ext-deploy-c...)

2a. Обработать входной файл и вывести на экран наименования функций, у которых есть оператор return. Не учитывать вложенные функции (определенные внутри тела функции в качестве переменных или callback-функций).

2b. Обработать входной файл и вывести на экран наименования только тех функций, у которых есть входные параметры, с указанием их количества. Не учитывать вложенные функции (определенные внутри тела функции в качестве переменных или callback-функций).

Пример вывода для функции _deleteOneEntity: function (sPath, fnSuccess, fnFailed) :

_deleteOneEntity : 3

я представляю как это сделать, написав парсер JS при помощи Lex-Yacc, но вряд-ли такое дали на тест, учитывая что два других вопроса это "посчитать количество нулей в факториале" и "написать запрос к API".

что подкажете?

Comments

[Rerurk]

*.js это обычный *.txt => string

[Antonio Solo]

Rerurk, капитан очевидность ?

[Danny Arty]

интересное задание, я бы попробовал решить задачу с помощью esprima

Answer 1

[Дмитрий Беляев]

Такие задачи однозначно решаются обходом AST. Возьмите общеизвестный парсер и инструмент для обхода его дерева, например связка из @babel/parser и @babel/traverse подойдет.
Так же можно посмотреть в какие ноды что парсится с помощью https://astexplorer.net/ - в нем несколько языков, а для js/ts на выбор есть несколько парсеров.
Вам по сути нужно найти FunctionDeclaration содержащий в body ReturnStatement и извлечь из него Identifier соответствующий имени. А для второй части - FunctionDeclaration с ArgumentsList не нулевой длины.
Ну и так как в условии сказано, что внутрь не заходить, то можно пройтись лишь по верхнему уровню, тогда @babel/traverse и не нужен даже

Comments

[Antonio Solo]

уточню вопрос :

два других вопроса в тесте - это "посчитать количество нулей в факториале" и "написать запрос к API" - на 10 минут каждое

но обходить дерево, хоть и уже построенное, будет отнюдь не простым делом. даже зная основы потребуется разбираться с теми структурами, которые строит бабель. к тому же что это очень узкая тема, и можно много лет пользоваться JS и бабелем, не задумываясь как это работает.

может быть есть более простое решение ? что-нибудь вроде сonsole.dir() ?

[Дмитрий Беляев]

Antonio Solo, обход дерева ничуть не сложнее "посчитать количество нулей в факториале"

[Kovalsky]

Antonio Solo, да ну вы что, попробуйте, это весело. Возьмите эсприму в качестве парсера, потренируйтесь на astexplorer.net. После этого вызываете esprima.parse(your_lovely_js_code) и ищете нужные объекты

Answer 2

[Stalker_RED]

можно даже потоково это посчитать.

заводишь три флага:
1. мы внутри функции
2. внутри строки
3. внутри вложенной функции

читаешь посимвольно пока не наткнешься на function или => объявление переменной, если попал в функцию запоминаешь имя (можно еще номер строки и символа), перебираешь дальше в поисках return или конца конца функции.

если попал внутрь вложенной функции - ищешь ее конец, return и другие вложенные игнорируешь.
если попал внутрь строки - ищешь конец строки, в строке игнорируешь вообще все.

А когда у тебя есть файл, все проще) есть indexOf, slice, replace с регулярками.

Comments

[Antonio Solo]

регулярками

это не так просто как кажется. потребуется лексер (токенизатор), а как их делать есть специальная теория. ну и взять готовый @babel/parser (как советуют) будет проще, чем изобретать самостоятельно.

[Дмитрий Беляев]

регулярками

когда в руках молоток все подозрительно выглядит как гвозди...

[shurshur]

console.log("закрываю все функции :}}}}");

[Rerurk]

Дмитрий Беляев, Дмитрий, я может и сильно ошибаюсь, но что мешает ловить в стринге "function()" И "return", никто же не сказал о классах и методах

[Дмитрий Беляев]

function f() {
  console.log(g());
  function g() {
    return 1;
  }
}

Rerurk, по условиям задачи, такую функцию надо пропустить, как такое на строках адекватно решать?

[Rerurk]

Дмитрий Беляев, и? g и return Дмитрий Беляев,

[Дмитрий Беляев]

Rerurk, а теперь почитайте внимательно, в вопросе процитирована задача, которую решает автор, и там есть такое:

Не учитывать вложенные функции

[Дмитрий Беляев]

Rerurk, а учитывая, что это тестовое к собеседованию, то использование babel/esprima покажет сразу 2 положительных моментов: "умеет гуглить/искать на npm" и "умеет работать с деревьями"
А вот дай мне решение с обработкой кода на строках, я бы лично резюмировал так: "не использовал инженерного мышления, решил задачу в лоб, не проверил крайние случаи". И да, какое бы решение Вы не написали, работающее на строках - я найду пример кода, на котором оно отработает некорректно.

[Rerurk]

Дмитрий Беляев, я согласен что быстрее и проще, с помощью инструментов, Но как понять в Вашем случае, умеет ли человек составлять алгоритмы

[Kovalsky]

Rerurk, https://stackoverflow.com/questions/1732348/regex-...

У страха глаза велики. Оп справится с разбором аст за полчаса максимум) главное начать

[Дмитрий Беляев]

Rerurk, для джуна не нужно составлять алгоритмы, достаточно знать какие они бывают и где взять их реализацию. Составлять алгоритмы - это инженерная задача, которая нужна далеко не всем, так многие задачи решаются известными алгоритмами.
Ну а что касается текущего примера - обход дерева покажет владение алгоритмами гораздо лучше, чем извращения с методами строки.
Опять же, если бы хотели алгоритмы, запросили бы тот же поиск в строке без встроенных методах, да чтоб был быстрее O(nm), а не более приближенный к реальным задачам анализ кода.