Partially address review

2 files changed, 18 insertions, 14 deletions

diff --git a/src/gleam/bit_array.gleam b/src/gleam/bit_array.gleam
index cba8da9..9f03cc8 100644
--- a/src/gleam/bit_array.gleam
+++ b/src/gleam/bit_array.gleam
@@ -1,5 +1,6 @@
 //// BitArrays are a sequence of binary data of any length.
 
+@target(erlang)
 import gleam/int
 import gleam/order
 import gleam/string
@@ -206,17 +207,14 @@ fn do_inspect(input: BitArray, accumulator: String) -> String {
 ///
 /// Only supported on Erlang target for now.
 ///
-pub fn compare(a: BitArray, with b: BitArray) -> order.Order {
-  do_compare(a, b)
-}
-
 @external(javascript, "../gleam_stdlib.mjs", "bit_array_compare")
-fn do_compare(a: BitArray, with b: BitArray) -> order.Order {
+pub fn compare(a: BitArray, with b: BitArray) -> order.Order {
   case a, b {
     <<first_byte, first_rest:bits>>, <<second_byte, second_rest:bits>> ->
-      case int.compare(first_byte, second_byte) {
-        order.Eq -> compare(first_rest, second_rest)
-        result -> result
+      case first_byte, second_byte {
+        f, s if f > s -> order.Gt
+        f, s if f < s -> order.Lt
+        _, _ -> compare(first_rest, second_rest)
       }
 
     <<>>, <<>> -> order.Eq
@@ -227,9 +225,15 @@ fn do_compare(a: BitArray, with b: BitArray) -> order.Order {
     // This happens when there's unusually sized elements.
     // Handle these special cases via custom erlang function.
     first, second ->
-      int.compare(bit_array_to_int(first), bit_array_to_int(second))
+      case bit_array_to_int_and_size(first), bit_array_to_int_and_size(second) {
+        #(a, _), #(b, _) if a > b -> order.Gt
+        #(a, _), #(b, _) if a < b -> order.Lt
+        #(_, size_a), #(_, size_b) if size_a > size_b -> order.Gt
+        #(_, size_a), #(_, size_b) if size_a < size_b -> order.Lt
+        _, _ -> order.Eq
+      }
   }
 }
 
-@external(erlang, "gleam_stdlib", "bit_array_to_int")
-fn bit_array_to_int(a: BitArray) -> Int
+@external(erlang, "gleam_stdlib", "bit_array_to_int_and_size")
+fn bit_array_to_int_and_size(a: BitArray) -> #(Int, Int)
diff --git a/src/gleam_stdlib.erl b/src/gleam_stdlib.erl
index 96a35b7..1fc15b2 100644
--- a/src/gleam_stdlib.erl
+++ b/src/gleam_stdlib.erl
@@ -14,7 +14,7 @@
     decode_tuple5/1, decode_tuple6/1, tuple_get/2, classify_dynamic/1, print/1,
     println/1, print_error/1, println_error/1, inspect/1, float_to_string/1,
     int_from_base_string/2, utf_codepoint_list_to_string/1, contains_string/2,
-    crop_string/2, base16_decode/1, string_replace/3, regex_replace/3, slice/3, bit_array_to_int/1
+    crop_string/2, base16_decode/1, string_replace/3, regex_replace/3, slice/3, bit_array_to_int_and_size/1
 ]).
 
 %% Taken from OTP's uri_string module
@@ -492,10 +492,10 @@ inspect_bit_array(Rest, Acc) ->
     Segment = <<X1/binary, ":size(", Size1/binary, ")">>,
     inspect_bit_array(<<>>, append_segment(Acc, Segment)).
 
-bit_array_to_int(A) ->
+bit_array_to_int_and_size(A) ->
     Size = bit_size(A),
     <<A1:Size>> = A,
-    A1.
+    {A1, Size}.
 
 append_segment(<<"<<">>, Segment) ->
     <<"<<", Segment/binary>>;