Added extension restrictions

This commit is contained in:
Vighnesh Rege
2015-05-13 23:20:51 -07:00
parent 6d2aa601cd
commit 90b59876d0
3 changed files with 87 additions and 0 deletions

View File

@@ -254,6 +254,9 @@ module Linguist
#
# Returns true or false
def generated_apache_thrift?
return false unless ['.rb', '.py', '.go', '.js', '.m', '.java', '.h', '.cc', '.cpp'].include?(extname)
return false unless lines.count > 1
return lines[0].include?("Autogenerated by Thrift Compiler") || lines[1].include?("Autogenerated by Thrift Compiler")
end