1 files changed, 72 insertions, 0 deletions
diff --git a/utils/filament.go b/utils/filament.go
new file mode 100755
index 0000000..a2145fc
--- /dev/null
+++ b/utils/filament.go
@@ -0,0 +1,72 @@
+package utils
+
+import (
+	// "errors"
+
+	"github.com/gotk3/gotk3/gtk"
+	"github.com/Z-Bolt/OctoScreen/logger"
+	"github.com/Z-Bolt/OctoScreen/octoprintApis"
+	// "github.com/Z-Bolt/OctoScreen/octoprintApis/dataModels"
+)
+
+
+func Extrude(
+	client					*octoprintApis.Client,
+	isForward				bool,
+	extruderId				string,
+	parentWindow			*gtk.Window,
+	flowRatePercentage		int,
+	length					int,
+) {
+	var action string
+	if isForward {
+		action = "extrude"
+	} else {
+		action = "retract"
+	}
+
+	if CheckIfHotendTemperatureIsTooLow(client, extruderId, action, parentWindow) {
+		logger.Error("filament.Extrude() - temperature is too low")
+		// No need to display an error - CheckIfHotendTemperatureIsTooLow() displays an error to the user
+		// if the temperature is too low.
+		return
+	}
+
+	logger.Infof("filament.Extrude() - setting flow rate percentage of %d", flowRatePercentage)
+	if err := SetFlowRate(client, flowRatePercentage); err != nil {
+		logger.LogError("filament.Extrude()", "SetFlowRate()", err)
+		// TODO: display error?
+		return
+	}
+
+	cmd := &octoprintApis.ToolExtrudeRequest{}
+	if isForward {
+		cmd.Amount = length
+	} else {
+		cmd.Amount = -length
+	}
+
+	logger.Infof("filament.Extrude() - sending extrude request with length of: %d", cmd.Amount)
+	if err := cmd.Do(client); err != nil {
+		logger.LogError("filament.Extrude()", "Do(ToolExtrudeRequest)", err)
+		// TODO: display error?
+		return
+	}
+}
+
+
+func SetFlowRate(
+	client					*octoprintApis.Client,
+	flowRatePercentage		int,
+) error {
+	cmd := &octoprintApis.ToolFlowRateRequest{}
+	cmd.Factor = flowRatePercentage
+
+	logger.Infof("filament.SetFlowRate() - changing flow rate to %d%%", cmd.Factor)
+	if err := cmd.Do(client); err != nil {
+		logger.LogError("filament.SetFlowRate()", "Go(ToolFlowRateRequest)", err)
+		return err
+	}
+
+	return nil
+}