closeheat-editor/src/html_analizer.coffee

Edit deployed websites.

_ = require 'lodash'
jsdom = require('jsdom')
require('string_score')

module.exports =
class HTMLAnalizer
  constructor: (@file, @event) ->
    return unless @supportedFile()

    vdom = jsdom.jsdom @file.content,
      features:
        FetchExternalResources: false
        ProcessExternalResources: false

    @dom = $(vdom)
    @selector_parts = @event.selector.split(' > ')

  combinations: ->
    return [] unless @supportedFile()

    @scoredCombinations()

  urlFileMatchScore: ->
    max_selector_scale = 12
    @pathname().score(@file.path) * max_selector_scale

  pathname: ->
    return 'index.html' if @event.pathname == '/'

    FIRST_SLASH_REGEX = /^\//
    @event.pathname.replace(FIRST_SLASH_REGEX, '')

  supportedFile: ->
    SUPPORTED_EXTENSION_REGEX = /(\.|\/)(html|htm)$/
    @file.path.match(SUPPORTED_EXTENSION_REGEX)

  scoredCombinations: (all_combinations) ->
    _.map @allCombinations(), (combination) =>
      combination.type = 'html'
      combination.dom = @dom
      combination.string_score = @stringScore(combination)
      combination.url_file_match_score = @urlFileMatchScore()
      combination.score = combination.string_score * 0.8 + combination.selector_score * 0.2 + combination.url_file_match_score * 0.2
      combination.file_path = @file.path
      combination.original_text = @event.text
      combination

  stringScore: (combination) ->
    max_selector_scale = 12
    _.trim(@event.text).score(_.trim(combination.text)) * max_selector_scale

  allCombinations: ->
    result = []
    NTH_CHILD_REGEX = /:nth\-child\(\d\)/

    _.times @selector_parts.length + 1, (i) =>
      selector = _.takeRight(@selector_parts, i).join(' > ')
      selector_element = @dom.find(selector)[0]
      return true unless selector_element

      _.each @nodes(selector_element), (node) ->
        result.push
          selector: selector
          selector_score: i
          selector_element: selector_element
          node: node
          text: node.nodeValue

    bare_selector = _.last(@selector_parts).replace(NTH_CHILD_REGEX, '')
    _.each @dom.find(bare_selector), (selector_element) =>
      _.each @nodes(selector_element), (node) ->
        result.push
          selector: bare_selector
          selector_score: 0.1
          selector_element: selector_element
          node: node
          text: node.nodeValue

    no_selector = "*:contains('#{_.trim(@event.text)}')"
    no_selector_result = []
    _.each @dom.find(no_selector), (selector_element) =>
      _.each @nodes(selector_element), (node) ->
        no_selector_result.push
          selector: no_selector
          selector_score: 0
          selector_element: selector_element
          node: node
          text: node.nodeValue

    # would not be able to distiguish between elements without selector
    # in the same file
    # in the future maybe js analysis can help
    # or just allow to edit both (or select from source code)
    # or record user actions from the last rebuild
    # then change text, rebuild, see if dom changed in the right place,
    # then decide
    if no_selector_result.length == 1
      result = result.concat(no_selector_result)

    result

  # matching with inject script
  nodes: (target) ->
    WHITESPACE_REGEX = /^\s*$/

    result = []

    for node in target.childNodes
      if node.nodeName == "#text" && !(WHITESPACE_REGEX.test(node.nodeValue))
        result.push node

    NO_CONTENT_TAGS = ['INPUT', 'BUTTON', 'IMG']
    if _.includes(NO_CONTENT_TAGS, target.nodeName) && !target.childNodes.length
      result.push target

    result
    # TODO: handle "hello<a>some</a> super" editing super

    # console.log 'here'
    # console.log node.nodeValue
    # debugger if result[0].toString() == 'true'
    # 'Lenz'