This commit is contained in:
@@ -0,0 +1,172 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
module Rouge
|
||||
module Guessers
|
||||
class Disambiguation < Guesser
|
||||
include Util
|
||||
include Lexers
|
||||
|
||||
def initialize(filename, source)
|
||||
@filename = File.basename(filename)
|
||||
@source = source
|
||||
end
|
||||
|
||||
def filter(lexers)
|
||||
return lexers if lexers.size == 1
|
||||
return lexers if lexers.size == Lexer.all.size
|
||||
|
||||
@analyzer = TextAnalyzer.new(get_source(@source))
|
||||
|
||||
self.class.disambiguators.each do |disambiguator|
|
||||
next unless disambiguator.match?(@filename)
|
||||
|
||||
filtered = disambiguator.decide!(self)
|
||||
return filtered if filtered
|
||||
end
|
||||
|
||||
return lexers
|
||||
end
|
||||
|
||||
def contains?(text)
|
||||
return @analyzer.include?(text)
|
||||
end
|
||||
|
||||
def matches?(re)
|
||||
return !!(@analyzer =~ re)
|
||||
end
|
||||
|
||||
@disambiguators = []
|
||||
def self.disambiguate(*patterns, &decider)
|
||||
@disambiguators << Disambiguator.new(patterns, &decider)
|
||||
end
|
||||
|
||||
def self.disambiguators
|
||||
@disambiguators
|
||||
end
|
||||
|
||||
class Disambiguator
|
||||
include Util
|
||||
|
||||
def initialize(patterns, &decider)
|
||||
@patterns = patterns
|
||||
@decider = decider
|
||||
end
|
||||
|
||||
def decide!(guesser)
|
||||
out = guesser.instance_eval(&@decider)
|
||||
case out
|
||||
when Array then out
|
||||
when nil then nil
|
||||
else [out]
|
||||
end
|
||||
end
|
||||
|
||||
def match?(filename)
|
||||
@patterns.any? { |p| test_glob(p, filename) }
|
||||
end
|
||||
end
|
||||
|
||||
disambiguate '*.cfg' do
|
||||
next CiscoIos if matches?(/\A\s*(version|banner|interface)\b/)
|
||||
|
||||
INI
|
||||
end
|
||||
|
||||
disambiguate '*.pl' do
|
||||
next Perl if contains?('my $')
|
||||
next Prolog if contains?(':-')
|
||||
next Prolog if matches?(/\A\w+(\(\w+\,\s*\w+\))*\./)
|
||||
end
|
||||
|
||||
disambiguate '*.h' do
|
||||
next ObjectiveC if matches?(/@(end|implementation|protocol|property)\b/)
|
||||
next ObjectiveC if contains?('@"')
|
||||
next Cpp if matches?(/^\s*(?:catch|class|constexpr|namespace|private|
|
||||
protected|public|template|throw|try|using)\b/x)
|
||||
|
||||
C
|
||||
end
|
||||
|
||||
disambiguate '*.m' do
|
||||
next ObjectiveC if matches?(/@(end|implementation|protocol|property)\b/)
|
||||
next ObjectiveC if contains?('@"')
|
||||
|
||||
# Objective-C dereferenced pointers and Mathematica comments are similar.
|
||||
# Disambiguate for Mathematica by looking for any amount of whitespace (or no whitespace)
|
||||
# followed by "(*" (e.g. `(* comment *)`).
|
||||
next Mathematica if matches?(/^\s*\(\*/)
|
||||
|
||||
# Disambiguate for objc by looking for a deref'd pointer in a statement (e.g. `if (*foo == 0)`).
|
||||
# This pattern is less specific than the Mathematica pattern, so its positioned after it.
|
||||
next ObjectiveC if matches?(/^\s*(if|while|for|switch|do)\s*\([^)]*\*[^)]*\)/)
|
||||
|
||||
next Mathematica if contains?(':=')
|
||||
|
||||
next Mason if matches?(/<%(def|method|text|doc|args|flags|attr|init|once|shared|perl|cleanup|filter)([^>]*)(>)/)
|
||||
|
||||
next Matlab if matches?(/^\s*?%/)
|
||||
# Matlab cell array creation: data = {
|
||||
next Matlab if matches?(/^\s*[a-zA-Z]\w*\s*=\s*\{/)
|
||||
|
||||
next Mason if matches? %r!(</?%|<&)!
|
||||
end
|
||||
|
||||
disambiguate '*.php' do
|
||||
# PHP always takes precedence over Hack
|
||||
PHP
|
||||
end
|
||||
|
||||
disambiguate '*.hh' do
|
||||
next Cpp if matches?(/^\s*#include/)
|
||||
next Hack if matches?(/^<\?hh/)
|
||||
next Hack if matches?(/(\(|, ?)\$\$/)
|
||||
|
||||
Cpp
|
||||
end
|
||||
|
||||
disambiguate '*.plist' do
|
||||
next XML if matches?(/\A<\?xml\b/)
|
||||
|
||||
Plist
|
||||
end
|
||||
|
||||
disambiguate '*.sc' do
|
||||
next Python if matches?(/^#/)
|
||||
next SuperCollider if matches?(/(?:^~|;$)/)
|
||||
|
||||
next Python
|
||||
end
|
||||
|
||||
disambiguate 'Messages' do
|
||||
next MsgTrans if matches?(/^[^\s:]+:[^\s:]+/)
|
||||
|
||||
next PlainText
|
||||
end
|
||||
|
||||
disambiguate '*.cls' do
|
||||
next TeX if matches?(/\A\s*(?:\\|%)/)
|
||||
next OpenEdge if matches?(/(no\-undo|BLOCK\-LEVEL|ROUTINE\-LEVEL|&ANALYZE\-SUSPEND)/i)
|
||||
next Apex
|
||||
end
|
||||
|
||||
disambiguate '*.pp' do
|
||||
next Puppet if matches?(/(::)?([a-z]\w*::)/)
|
||||
next Pascal if matches?(/^(function|begin|var)\b/)
|
||||
next Pascal if matches?(/\b(end(;|\.))/)
|
||||
|
||||
Puppet
|
||||
end
|
||||
|
||||
disambiguate '*.p' do
|
||||
next Prolog if contains?(':-')
|
||||
next Prolog if matches?(/\A\w+(\(\w+\,\s*\w+\))*\./)
|
||||
next OpenEdge
|
||||
end
|
||||
|
||||
disambiguate '*.st' do
|
||||
next IecST if matches?(/^\s*END_/i)
|
||||
next Smalltalk
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
Reference in New Issue
Block a user