From 4f95b90b4d765212afcf6c062ab0e923b10582b2 Mon Sep 17 00:00:00 2001
From: stysmmaker <mmaker@mmaker.moe>
Date: Sat, 12 Jun 2021 11:06:57 -0400
Subject: [PATCH] Cleaned up logic

---
 .gitignore  |  2 --
 __init__.py | 34 +++++++++++++++++++++++-----------
 2 files changed, 23 insertions(+), 13 deletions(-)

diff --git a/.gitignore b/.gitignore
index f545b9c..6525d15 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,5 +1,3 @@
 .vscode
 *.code-workspace
-.vimrc
-*.swp
 __pycache__
\ No newline at end of file
diff --git a/__init__.py b/__init__.py
index 0bcca45..522f534 100644
--- a/__init__.py
+++ b/__init__.py
@@ -85,25 +85,23 @@ class FCurveHandleCopyValue(bpy.types.Operator):
 			G.selected_keys = []
 
 			if (len(fcurves) > 1):
-				self.report({"WARNING"}, "Only select one curve when copying an ease!")
+				self.report({"WARNING"}, "Please only select one curve when copying an ease.")
 				return {'CANCELLED'}
 			
 			for keyframe in fcurves[0].keyframe_points:
 				if (len(G.selected_keys) > 2):
-					self.report({"WARNING"}, "Please select exactly two keyframes when copying an ease!")
+					self.report({"WARNING"}, "Please select exactly two keyframes when copying an ease.")
 					return {'CANCELLED'}
 				
 				if (keyframe.select_control_point):
 					G.selected_keys.append(keyframe)
 		
 		if (len(G.selected_keys) != 2):
-			self.report({"WARNING"}, "Please select exactly two keyframes when copying an ease!")
+			self.report({"WARNING"}, "Please select exactly two keyframes when copying an ease.")
 			return {'CANCELLED'}
 		
 		G.bezier = convert_handles_to_bezier(G.selected_keys)
 
-		print("Done")
-
 		return {'FINISHED'}
 
 class FCurveHandlePasteValue(bpy.types.Operator):
@@ -117,12 +115,26 @@ class FCurveHandlePasteValue(bpy.types.Operator):
 
 			for fcurve in fcurves:
 				keys = fcurve.keyframe_points
-				for i in range(0, len(keys) - 1):
-					new_handles = generate_new_handles(keys[i], keys[i + 1])
-					keys[i].handle_right_type = 'FREE'
-					keys[i + 1].handle_left_type = 'FREE'
-					keys[i].handle_right    = new_handles[0]
-					keys[i + 1].handle_left = new_handles[1]
+				selected_keys = []
+				for i in range(0, len(keys)):
+					if (keys[i].select_control_point):
+						selected_keys.append(i)
+
+				if (len(selected_keys) == 0):
+					self.report({"WARNING"}, "Please select some keyframes to paste an ease to.")
+					return {'CANCELLED'}
+				if (len(selected_keys) == 1):
+					# TODO: Implement logic for this soon
+					pass
+				else:
+					selected_keys.pop() # TODO: Related to above, implement soon
+					for i in selected_keys:
+						if (i < len(keys) - 1):
+							new_handles = generate_new_handles(keys[i], keys[i + 1])
+							keys[i].handle_right_type = 'FREE'
+							keys[i + 1].handle_left_type = 'FREE'
+							keys[i].handle_right    = new_handles[0]
+							keys[i + 1].handle_left = new_handles[1]
 
 		return {'FINISHED'}