Support Kotlin dependencies
This commit is contained in:
parent
8fcfcb49d7
commit
8f62162c2a
3 changed files with 93 additions and 0 deletions
22
tests/samples/codefiles/kotlin.kt
Normal file
22
tests/samples/codefiles/kotlin.kt
Normal file
|
@ -0,0 +1,22 @@
|
||||||
|
package com.wakatime
|
||||||
|
|
||||||
|
import java.io.*
|
||||||
|
import alpha.time.Some
|
||||||
|
import bravo.charlie.something.Other
|
||||||
|
|
||||||
|
<#assign licenseFirst = "/*">
|
||||||
|
<#assign licensePrefix = " * ">
|
||||||
|
<#assign licenseLast = " */">
|
||||||
|
<#include "${project.licensePath}">
|
||||||
|
<#if package?? && package != "">
|
||||||
|
package ${package}
|
||||||
|
</#if>
|
||||||
|
|
||||||
|
import delta.io.*
|
||||||
|
|
||||||
|
class MainActivity : AppCompatActivity() {
|
||||||
|
|
||||||
|
override fun onCreate(savedInstanceState: Bundle?) {
|
||||||
|
super.onCreate(savedInstanceState)
|
||||||
|
}
|
||||||
|
}
|
|
@ -448,3 +448,15 @@ class DependenciesTestCase(TestCase):
|
||||||
expected_lines=21,
|
expected_lines=21,
|
||||||
entity='rust.rs',
|
entity='rust.rs',
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def test_kotlin_dependencies_detected(self):
|
||||||
|
self.shared(
|
||||||
|
expected_dependencies=[
|
||||||
|
'alpha.time',
|
||||||
|
'bravo.charlie',
|
||||||
|
'delta.io',
|
||||||
|
],
|
||||||
|
expected_language='Kotlin',
|
||||||
|
expected_lines=22,
|
||||||
|
entity='kotlin.kt',
|
||||||
|
)
|
||||||
|
|
|
@ -96,6 +96,65 @@ class JavaParser(TokenParser):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class KotlinParser(TokenParser):
|
||||||
|
state = None
|
||||||
|
exclude = [
|
||||||
|
r'^java\.',
|
||||||
|
]
|
||||||
|
|
||||||
|
def parse(self):
|
||||||
|
for index, token, content in self.tokens:
|
||||||
|
self._process_token(token, content)
|
||||||
|
return self.dependencies
|
||||||
|
|
||||||
|
def _process_token(self, token, content):
|
||||||
|
if self.partial(token) == 'Keyword':
|
||||||
|
self._process_keyword(token, content)
|
||||||
|
elif self.partial(token) == 'Text':
|
||||||
|
self._process_text(token, content)
|
||||||
|
elif self.partial(token) == 'Namespace':
|
||||||
|
self._process_namespace(token, content)
|
||||||
|
else:
|
||||||
|
self._process_other(token, content)
|
||||||
|
|
||||||
|
def _process_keyword(self, token, content):
|
||||||
|
self.state = content
|
||||||
|
|
||||||
|
def _process_text(self, token, content):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def _process_namespace(self, token, content):
|
||||||
|
if self.state == 'import':
|
||||||
|
self.append(self._format(content))
|
||||||
|
self.state = None
|
||||||
|
|
||||||
|
def _process_other(self, token, content):
|
||||||
|
self.state = None
|
||||||
|
|
||||||
|
def _format(self, content):
|
||||||
|
content = content.split(u('.'))
|
||||||
|
|
||||||
|
if len(content) == 0:
|
||||||
|
return None
|
||||||
|
|
||||||
|
if len(content) == 1:
|
||||||
|
return content[0]
|
||||||
|
|
||||||
|
if len(content[0]) == 3:
|
||||||
|
content = content[1:]
|
||||||
|
if content[-1] == u('*'):
|
||||||
|
content = content[:len(content) - 1]
|
||||||
|
|
||||||
|
if len(content) == 0:
|
||||||
|
return None
|
||||||
|
elif len(content) == 1:
|
||||||
|
content = content[0]
|
||||||
|
elif len(content) > 1:
|
||||||
|
content = u('.').join(content[:2])
|
||||||
|
|
||||||
|
return content
|
||||||
|
|
||||||
|
|
||||||
class ScalaParser(TokenParser):
|
class ScalaParser(TokenParser):
|
||||||
state = None
|
state = None
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue