Implement methods on kittycad.rs types

.gitignore

@@ -1,2 +1,3 @@
 /target
 .vscode
+.idea

kittycad/src/lib.rs

@@ -99,6 +99,8 @@ pub mod hidden;
 /// FROM: <https://docs.kittycad.io/api/meta>
 #[cfg(feature = "requests")]
 pub mod meta;
+#[cfg(feature = "requests")]
+mod methods;
 /// Modeling API for updating your 3D files using the KittyCAD engine.
 ///
 /// FROM: <https://docs.kittycad.io/api/modeling>

kittycad/src/methods.rs

@@ -0,0 +1,150 @@
+use crate::types::{Angle, Point2D, Point3D, UnitAngle};
+use std::ops::{Add, Div, Mul, Sub};
+
+impl Angle {
+    /// Make a new angle in degrees
+    pub fn from_degrees(size: f64) -> Self {
+        Self {
+            unit: UnitAngle::Degrees,
+            value: size,
+        }
+    }
+    /// Make a new angle in radians
+    pub fn from_radians(size: f64) -> Self {
+        Self {
+            unit: UnitAngle::Radians,
+            value: size,
+        }
+    }
+
+    /// Get the size of the angle, in radians
+    pub fn radians(&self) -> f64 {
+        match self.unit {
+            UnitAngle::Radians => self.value,
+            UnitAngle::Degrees => self.value.to_radians(),
+        }
+    }
+
+    /// Get the size of the angle, in degrees
+    pub fn degrees(&self) -> f64 {
+        match self.unit {
+            UnitAngle::Degrees => self.value,
+            UnitAngle::Radians => self.value.to_degrees(),
+        }
+    }
+}
+
+impl Add for Point2D {
+    type Output = Self;
+
+    fn add(self, rhs: Self) -> Self::Output {
+        Self {
+            x: self.x + rhs.x,
+            y: self.y + rhs.y,
+        }
+    }
+}
+
+impl Sub for Point2D {
+    type Output = Self;
+
+    fn sub(self, rhs: Self) -> Self::Output {
+        Self {
+            x: self.x - rhs.x,
+            y: self.y - rhs.y,
+        }
+    }
+}
+
+impl Mul for Point2D {
+    type Output = Self;
+
+    fn mul(self, rhs: Self) -> Self::Output {
+        Self {
+            x: self.x * rhs.x,
+            y: self.y * rhs.y,
+        }
+    }
+}
+
+impl Div for Point2D {
+    type Output = Self;
+
+    fn div(self, rhs: Self) -> Self::Output {
+        Self {
+            x: self.x / rhs.x,
+            y: self.y / rhs.y,
+        }
+    }
+}
+
+impl Add for Point3D {
+    type Output = Self;
+
+    fn add(self, rhs: Self) -> Self::Output {
+        Self {
+            x: self.x + rhs.x,
+            y: self.y + rhs.y,
+            z: self.z + rhs.z,
+        }
+    }
+}
+
+impl Sub for Point3D {
+    type Output = Self;
+
+    fn sub(self, rhs: Self) -> Self::Output {
+        Self {
+            x: self.x - rhs.x,
+            y: self.y - rhs.y,
+            z: self.z - rhs.z,
+        }
+    }
+}
+
+impl Mul for Point3D {
+    type Output = Self;
+
+    fn mul(self, rhs: Self) -> Self::Output {
+        Self {
+            x: self.x * rhs.x,
+            y: self.y * rhs.y,
+            z: self.z * rhs.z,
+        }
+    }
+}
+
+impl Div for Point3D {
+    type Output = Self;
+
+    fn div(self, rhs: Self) -> Self::Output {
+        Self {
+            x: self.x / rhs.x,
+            y: self.y / rhs.y,
+            z: self.z / rhs.z,
+        }
+    }
+}
+
+impl Point3D {
+    /// Add a Z component to a 2D point.
+    pub fn with_z(Point2D { x, y }: Point2D, z: f64) -> Self {
+        Self { x, y, z }
+    }
+
+    pub const ZERO: Self = Self {
+        x: 0.0,
+        y: 0.0,
+        z: 0.0,
+    };
+}
+
+impl Point2D {
+    pub const ZERO: Self = Self { x: 0.0, y: 0.0 };
+}
+
+impl From<Point3D> for Point2D {
+    fn from(Point3D { x, y, .. }: Point3D) -> Self {
+        Point2D { x, y }
+    }
+}

openapitor/src/lib.rs

@@ -294,7 +294,7 @@ pub fn generate(spec: &openapiv3::OpenAPI, opts: &Opts) -> Result<()> {
             .to_string();
 
         let persistent_modules = persistent_modules();
-        if persistent_modules.contains(&file_name) {
+        if persistent_modules.contains(&file_name.as_str()) {
             continue;
         }
 
@@ -536,8 +536,8 @@ impl Default for Opts {
 
 /// Return a list of the persistent modules.
 /// These are modules we do not nuke at generation time.
-fn persistent_modules() -> Vec<String> {
-    vec!["tests".to_string()]
+fn persistent_modules() -> Vec<&'static str> {
+    vec!["tests", "methods"]
 }
 
 fn generate_cargo_toml(opts: &Opts) -> String {