oh-my-git/repository.gd

293 lines
6.4 KiB
GDScript3
Raw Normal View History

extends Control
2020-09-01 17:24:21 +02:00
onready var index = $VSplitContainer/RepoVis/Index
onready var nodes = $VSplitContainer/RepoVis/Nodes
onready var file_browser = $VSplitContainer/FileBrowser
onready var label_node = $VSplitContainer/RepoVis/Label
2020-09-03 18:10:09 +02:00
export var label: String setget set_label
2020-09-01 17:24:21 +02:00
export var path: String setget set_path, get_path
export var file_browser_active = true setget set_file_browser_active
2020-09-01 17:24:21 +02:00
var node = preload("res://node.tscn")
var shell = Shell.new()
var objects = {}
2020-09-15 18:41:06 +02:00
var mouse_inside = false
2020-09-14 16:03:01 +02:00
var _simplified_view = false
2020-09-01 17:24:21 +02:00
func _ready():
file_browser.shell = shell
# Trigger these again because nodes were not ready before.
set_label(label)
set_file_browser_active(file_browser_active)
2020-09-01 17:24:21 +02:00
2020-09-08 20:26:14 +02:00
func _process(_delta):
nodes.rect_pivot_offset = nodes.rect_size / 2
if path:
apply_forces()
func _input(event):
if mouse_inside:
if event.is_action_pressed("zoom_out") and nodes.rect_scale.x > 0.3:
nodes.rect_scale -= Vector2(0.05, 0.05)
if event.is_action_pressed("zoom_in") and nodes.rect_scale.x < 2:
nodes.rect_scale += Vector2(0.05, 0.05)
func there_is_a_git():
return shell.run("test -d .git && echo yes || echo no") == "yes\n"
func update_everything():
file_browser.update()
if there_is_a_git():
update_head()
update_refs()
update_index()
update_objects()
remove_gone_stuff()
2020-09-18 11:15:09 +02:00
else:
index.text = ""
for o in objects:
objects[o].queue_free()
objects = {}
2020-09-01 17:24:21 +02:00
func set_path(new_path):
path = new_path
shell.cd(new_path)
for o in objects.values():
o.queue_free()
objects = {}
update_everything()
2020-09-01 17:24:21 +02:00
func get_path():
return path
2020-09-03 18:10:09 +02:00
func set_label(new_label):
label = new_label
if label_node:
label_node.text = new_label
2020-09-03 18:10:09 +02:00
2020-09-01 17:24:21 +02:00
func update_index():
index.text = git("ls-files -s --abbrev=8").replace("\t", " ")
2020-09-01 17:24:21 +02:00
func random_position():
return Vector2(rand_range(0, rect_size.x), rand_range(0, rect_size.y))
2020-09-01 17:24:21 +02:00
func update_objects():
var all = all_objects()
# Create new objects, if necessary.
for o in all:
2020-09-01 17:24:21 +02:00
if objects.has(o):
continue
2020-09-14 16:03:01 +02:00
var type = object_type(o)
if _simplified_view:
if type == "tree" or type == "blob":
continue
2020-09-01 17:24:21 +02:00
var n = node.instance()
n.id = o
n.type = object_type(o)
n.content = object_content(o)
n.repository = self
2020-09-14 16:03:01 +02:00
match type:
"blob":
pass
"tree":
n.children = tree_children(o)
n.content = n.content.replacen("\t", " ")
2020-09-14 16:03:01 +02:00
"commit":
var c = {}
c[commit_tree(o)] = ""
for p in commit_parents(o):
c[p] = ""
n.children = c
"tag":
n.children = tag_target(o)
2020-09-14 14:54:30 +02:00
n.position = find_position(n)
nodes.add_child(n)
2020-09-01 17:24:21 +02:00
objects[o] = n
func update_refs():
for r in all_refs():
if not objects.has(r):
var n = node.instance()
n.id = r
n.type = "ref"
n.content = ""
n.repository = self
objects[r] = n
2020-09-14 14:54:30 +02:00
n.children = {ref_target(r): ""}
n.position = find_position(n)
nodes.add_child(n)
2020-09-01 17:24:21 +02:00
var n = objects[r]
2020-09-13 21:55:24 +02:00
n.children = {ref_target(r): ""}
2020-09-01 17:24:21 +02:00
func apply_forces():
for o in objects.values():
2020-09-14 16:03:01 +02:00
if not o.visible:
continue
2020-09-01 17:24:21 +02:00
for o2 in objects.values():
2020-09-14 16:03:01 +02:00
if o == o2 or not o2.visible:
2020-09-01 17:24:21 +02:00
continue
var d = o.position.distance_to(o2.position)
var dir = (o.global_position - o2.global_position).normalized()
2020-09-14 14:54:30 +02:00
var f = 2000/pow(d+0.00001,1.5)
2020-09-01 17:24:21 +02:00
o.position += dir*f
o2.position -= dir*f
var center_of_gravity = rect_size/2
var d = o.position.distance_to(center_of_gravity)
var dir = (o.position - center_of_gravity).normalized()
2020-09-03 18:10:09 +02:00
var f = (d+0.00001)*Vector2(0.03, 0.01)
2020-09-01 17:24:21 +02:00
o.position -= dir*f
2020-09-14 14:54:30 +02:00
func find_position(n):
var position = Vector2.ZERO
var count = 0
for child in n.children:
if objects.has(child):
position += objects[child].position
count += 1
if count > 0:
position /= count
n.position = position + Vector2(0, -150)
else:
n.position = random_position()
return n.position
2020-09-01 17:24:21 +02:00
func git(args, splitlines = false):
var o = shell.run("git --no-replace-objects " + args)
2020-09-01 17:24:21 +02:00
if splitlines:
o = o.split("\n")
# Remove last empty line.
o.remove(len(o)-1)
else:
# Remove trailing newline.
o = o.substr(0,len(o)-1)
return o
func update_head():
if not objects.has("HEAD"):
var n = node.instance()
n.id = "HEAD"
n.type = "head"
n.content = ""
n.repository = self
2020-09-14 14:54:30 +02:00
n.position = find_position(n)
2020-09-01 17:24:21 +02:00
objects["HEAD"] = n
nodes.add_child(n)
2020-09-01 17:24:21 +02:00
var n = objects["HEAD"]
2020-09-13 21:55:24 +02:00
n.children = {ref_target("HEAD"): ""}
2020-09-01 17:24:21 +02:00
func all_objects():
var obj = git("cat-file --batch-check='%(objectname)' --batch-all-objects", true)
var dict = {}
for o in obj:
dict[o] = ""
return dict
2020-09-01 17:24:21 +02:00
func object_type(id):
return git("cat-file -t "+id)
func object_content(id):
return git("cat-file -p "+id)
func tree_children(id):
var children = git("cat-file -p "+id, true)
var ids = {}
for c in children:
var a = c.split(" ")
ids[a[2].split("\t")[0]] = a[2].split("\t")[1]
return ids
func commit_tree(id):
var c = git("cat-file -p "+id, true)
for cc in c:
var ccc = cc.split(" ", 2)
match ccc[0]:
"tree":
return ccc[1]
return null
func commit_parents(id):
var parents = []
var c = git("cat-file -p "+id, true)
for cc in c:
var ccc = cc.split(" ", 2)
match ccc[0]:
"parent":
parents.push_back(ccc[1])
return parents
2020-09-13 21:55:24 +02:00
func tag_target(id):
var c = git("rev-parse %s^{}" % id)
return {c: ""}
2020-09-01 17:24:21 +02:00
func all_refs():
var refs = {}
2020-09-08 20:26:14 +02:00
# If there are no refs, show-ref will have exit code 1. We don't care.
for line in git("show-ref || true", true):
2020-09-01 17:24:21 +02:00
line = line.split(" ")
2020-09-08 20:26:14 +02:00
var _id = line[0]
2020-09-01 17:24:21 +02:00
var name = line[1]
refs[name] = ""
2020-09-01 17:24:21 +02:00
return refs
2020-09-13 21:55:24 +02:00
func ref_target(ref):
# Test whether this is a symbolic ref.
var ret = git("symbolic-ref -q "+ref+" || true")
# If it's not, it's probably a regular ref.
if ret == "":
if ref == "HEAD":
ret = git("show-ref --head "+ref).split(" ")[0]
else:
ret = git("show-ref "+ref).split(" ")[0]
2020-09-13 21:55:24 +02:00
return ret
2020-09-14 16:03:01 +02:00
func simplify_view(pressed):
_simplified_view = pressed
for o in objects:
var obj = objects[o]
if obj.type == "tree" or obj.type == "blob":
obj.visible = not pressed
if there_is_a_git():
update_objects()
func remove_gone_stuff():
# FIXME: Cache the result of all_objects.
var all = {}
for o in all_objects():
all[o] = ""
for o in all_refs():
all[o] = ""
all["HEAD"] = ""
# Delete objects, if they disappeared.
for o in objects.keys():
if not all.has(o):
objects[o].queue_free()
objects.erase(o)
2020-09-15 18:41:06 +02:00
func _on_mouse_entered():
mouse_inside = true
func _on_mouse_exited():
mouse_inside = false
func set_file_browser_active(active):
file_browser_active = active
if file_browser:
file_browser.visible = active