Skip to content

Commit

Permalink
Format code and Improve project name finding (#4)
Browse files Browse the repository at this point in the history 
* fix: Set last_heartbeat initially to 0

* feat: If no project name is defined, search for a git archive

* chore: Format the code for developer sanity

* fix: Change class to be capitalized

* fix: fallback to directory name if no git repo is found

* fix: Only return dirname when no git repo is found
  • Loading branch information
@lajp
lajp authored Jun 23, 2023
1 parent e46f93c commit 66e03e2
Showing 1 changed file with 122 additions and 96 deletions.
218 changes: 122 additions & 96 deletions testaustime.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5,51 +5,53 @@
import time
import os
import threading
import subprocess

PLUGIN_SETTINGS_KEY = "testaustime.sublime-settings"
API_SETTINGS_KEY = "api_key"
ENDPOINT_SETTINGS_KEY = "endpoint_url"

last_heartbeat = time.time()
last_heartbeat = 0.0
can_show_missing_key_popup = True



class testaustime:
class Testaustime:
def __init__(self):
self.settings = sublime.load_settings(PLUGIN_SETTINGS_KEY)

if not self.get_endpoint_url():
print("no endpoint url defined, setting the value to https://api.testaustime.fi")
self.save_endpoint_url("https://api.testaustime.fi")
print("no endpoint url defined, setting the value to https://api.testaustime.fi")
self.save_endpoint_url("https://api.testaustime.fi")

if not self.get_api_key():
self.missing_api_key_popup()
self.missing_api_key_popup()

def missing_api_key_popup(self):
window = sublime.active_window()
global can_show_missing_key_popup
if can_show_missing_key_popup:
sublime.message_dialog("You haven't set the API key yet.\nSet it from 'Preferences -> testaustime -> Set API key")
can_show_missing_key_popup = False
sublime.active_window()
global can_show_missing_key_popup

if can_show_missing_key_popup:
sublime.message_dialog("You haven't set the API key yet.\nSet it from 'Preferences -> testaustime -> Set API key")
can_show_missing_key_popup = False

def save_api_key(self, api_key):
self.settings.set(API_SETTINGS_KEY, api_key)
sublime.save_settings(PLUGIN_SETTINGS_KEY)

def save_endpoint_url(self, endpoint_url):
self.settings.set(ENDPOINT_SETTINGS_KEY, endpoint_url)
sublime.save_settings(PLUGIN_SETTINGS_KEY)
self.settings.set(ENDPOINT_SETTINGS_KEY, endpoint_url)
sublime.save_settings(PLUGIN_SETTINGS_KEY)

def get_api_key(self):
key = self.settings.get(API_SETTINGS_KEY, None)
if key:
return str(key)
return str(key)

def get_endpoint_url(self):
endpoint = self.settings.get(ENDPOINT_SETTINGS_KEY, None)
if endpoint:
return str(endpoint)
endpoint = self.settings.get(ENDPOINT_SETTINGS_KEY, None)
if endpoint:
return str(endpoint)

if testaustime().get_api_key():
if Testaustime().get_api_key():
sublime.status_message(" testaustime is ready. happy coding")
else:
sublime.status_message(" testaustime API token not set")
Expand All @@ -60,114 +62,138 @@ def run(self):

def _on_input_done(self, user_input):
sublime.message_dialog("API key set")
testaustime().save_api_key(user_input)


Testaustime().save_api_key(user_input)

class prompt_url_endpoint(sublime_plugin.WindowCommand):
def run(self):
self.window.show_input_panel("Endpoint URL:", "", self._on_input_done, None, None)
def run(self):
self.window.show_input_panel("Endpoint URL:", "", self._on_input_done, None, None)

def _on_input_done(self, user_input):
if not user_input.startswith("https://") and not user_input.startswith("http://"):
user_input = "https://" + user_input

if user_input.endswith("/"):
user_input = user_input[:-1]

def _on_input_done(self, user_input):
sublime.message_dialog("Endpoint set")
if not user_input.startswith("https://") and not user_input.startswith("http://"):
user_input = "https://" + user_input
if user_input.endswith("/"):
user_input = user_input[:-1]
testaustime().save_endpoint_url(user_input)
Testaustime().save_endpoint_url(user_input)
sublime.message_dialog("Endpoint set")

class ApiCredHandler(sublime_plugin.TextCommand):
def retrieve_api_key(self, edit):
api_key = prompt_api_key()
testaustime.save_api_key(api_key)
Testaustime.save_api_key(api_key)

class get_project_name(sublime_plugin.TextCommand):
def run(self, edit):
get_current_project_name()


def run(self, edit):
get_current_project_name()


def AsyncApiCall(self, timeout, endpoint, has_body):
threads = []
err = ""

try:
if assemble_data() and assemble_headers():
if has_body:
request = urllib.request.Request(testaustime().get_endpoint_url() + endpoint, data=assemble_data(), headers=assemble_headers())
else:
request = urllib.request.Request(testaustime().get_endpoint_url() + endpoint, headers=assemble_headers())
response = urllib.request.urlopen(request)
response_text = response.read().decode('utf-8')
return response_text
except (urllib.error.HTTPError) as e:
err = '%s: HTTP error %s contacting API' % (__name__, str(e.code))
print(err)
sublime.message_dialog(err)
except (urllib.error.URLError) as e:
err = '%s: URL error %s contacting API' % (__name__, str(e.reason))
print(err)
sublime.message_dialog(err)
try:
if assemble_data() and assemble_headers():
if has_body:
request = urllib.request.Request(Testaustime().get_endpoint_url() + endpoint,
data=assemble_data(), headers=assemble_headers())
else:
request = urllib.request.Request(Testaustime().get_endpoint_url() + endpoint,
headers=assemble_headers())

response = urllib.request.urlopen(request)
response_text = response.read().decode('utf-8')

return response_text

except (urllib.error.HTTPError) as e:
err = '%s: HTTP error %s contacting API' % (__name__, str(e.code))
print(err)
sublime.message_dialog(err)

except (urllib.error.URLError) as e:
err = '%s: URL error %s contacting API' % (__name__, str(e.reason))
print(err)
sublime.message_dialog(err)

"""
FUNCTIONS
"""

def get_current_syntax():
view = sublime.active_window().active_view()
syntax = view.settings().get('syntax')
syntax = syntax.split('/')[-1].split('.')[0]
return syntax
view = sublime.active_window().active_view()
syntax = view.settings().get('syntax')
syntax = syntax.split('/')[-1].split('.')[0]
return syntax

def show_project():
view = sublime.active_window().active_view()
if view.window() is None:
return 'None'
project_file = view.window ().project_file_name ()
if project_file is not None:
project_name = os.path.splitext (os.path.basename (project_file))[0]
return project_name
view = sublime.active_window().active_view()
if view.window() is None:
return

project_file = view.window().project_file_name()
if project_file is not None:
project_name = os.path.splitext(os.path.basename(project_file))[0]
return project_name

filename = view.file_name()

if filename is None:
return

git_project = git_root(filename)
if len(git_project) > 0:
return str(os.path.basename(git_project))

return str(os.path.basename(os.path.dirname(filename)))

def git_root(file):
proc = subprocess.Popen(["git", "rev-parse", "--show-toplevel"], cwd=os.path.dirname(file), stdout=subprocess.PIPE)
proc.wait()

repo = proc.stdout.read().split(b"\n")[0]
return os.path.basename(repo)

def assemble_data():
data = {
'language': get_current_syntax(),
'hostname': os.uname()[1],
'editor_name': 'SublimeText',
'project_name': str(show_project())
}
data = json.dumps(data).encode('utf-8')
return data
data = {
'language': get_current_syntax(),
'hostname': os.uname()[1],
'editor_name': 'SublimeText',
'project_name': show_project()
}

data = json.dumps(data).encode('utf-8')
return data

def assemble_headers():
if testaustime().get_api_key():
headers = {
'Content-Type': 'application/json',
'Authorization': 'Bearer ' + testaustime().get_api_key(),
}
return(headers)
if Testaustime().get_api_key():
headers = {
'Content-Type': 'application/json',
'Authorization': 'Bearer ' + Testaustime().get_api_key(),
}

return headers

def get_user_data():
thread = threading.Thread(target=AsyncApiCall, args=(AsyncApiCall,5,'/users/@me', False))
thread.start()
thread = threading.Thread(target=AsyncApiCall, args=(AsyncApiCall,5,'/users/@me', False))
thread.start()

def heart_beat():
thread = threading.Thread(target=AsyncApiCall, args=(AsyncApiCall,5,'/activity/update', True))
thread.start()
def heartbeat():
thread = threading.Thread(target=AsyncApiCall, args=(AsyncApiCall,5,'/activity/update', True))
thread.start()

def flush():
thread = threading.Thread(target=AsyncApiCall, args=(AsyncApiCall,5,'/activity/flush', True))
thread.start()
thread = threading.Thread(target=AsyncApiCall, args=(AsyncApiCall,5,'/activity/flush', True))
thread.start()

class IdleHandler(sublime_plugin.EventListener):
def on_modified(view, event):
global last_activity
global last_heartbeat
now = time.time()
last_activity = now
if now - last_heartbeat > 30:
sublime.set_timeout_async(heart_beat(), 0)
last_heartbeat = now
def on_modified(view, event):
global last_activity
global last_heartbeat

now = time.time()
last_activity = now

if now - last_heartbeat > 30:
sublime.set_timeout_async(heartbeat(), 0)
last_heartbeat = now

class ExitHandler(sublime_plugin.EventListener):
def on_pre_close(self, view):
Expand Down

0 comments on commit 66e03e2

Please sign in to comment.