85 lines
2.3 KiB
Python
85 lines
2.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
wakatime.dependencies.objective
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
Parse dependencies from Objective-C and Swift code.
|
|
|
|
:copyright: (c) 2018 Alan Hamlett.
|
|
:license: BSD, see LICENSE for more details.
|
|
"""
|
|
|
|
import re
|
|
|
|
from . import TokenParser
|
|
|
|
|
|
class SwiftParser(TokenParser):
|
|
state = None
|
|
exclude = [
|
|
r'^foundation$',
|
|
]
|
|
|
|
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) == 'Declaration':
|
|
self._process_declaration(token, content)
|
|
elif self.partial(token) == 'Class':
|
|
self._process_class(token, content)
|
|
else:
|
|
self._process_other(token, content)
|
|
|
|
def _process_declaration(self, token, content):
|
|
if self.state is None:
|
|
self.state = content
|
|
|
|
def _process_class(self, token, content):
|
|
if self.state == 'import':
|
|
self.append(content)
|
|
self.state = None
|
|
|
|
def _process_other(self, token, content):
|
|
pass
|
|
|
|
|
|
class ObjectiveCParser(TokenParser):
|
|
state = None
|
|
extension = re.compile(r'\.[mh]$')
|
|
|
|
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) == 'Preproc':
|
|
self._process_preproc(token, content)
|
|
else:
|
|
self._process_other(token, content)
|
|
|
|
def _process_preproc(self, token, content):
|
|
if self.state:
|
|
self._process_import(token, content)
|
|
|
|
self.state = content
|
|
|
|
def _process_import(self, token, content):
|
|
if self.state == '#' and content.startswith('import '):
|
|
self.append(self._format(content))
|
|
self.state = None
|
|
|
|
def _process_other(self, token, content):
|
|
pass
|
|
|
|
def _format(self, content):
|
|
content = content.strip().lstrip('import ').strip()
|
|
content = content.strip('"').strip("'").strip()
|
|
content = content.strip('<').strip('>').strip()
|
|
content = content.split('/')[0]
|
|
content = self.extension.sub('', content, count=1)
|
|
return content
|