Use bytes insteaf of script in parse_if

Author: lucidLuckylee

src/generate.rs

@@ -51,7 +51,7 @@ fn generate_escape(builder: TokenStream, expression: TokenStream, span: Span) ->
                 #[allow(clippy::all)]
 
                 use ::bitcoin::blockdata::script::Builder;
-                fn push(builder: Builder, value: impl pushable::Pushable) -> Builder {
+                fn push<T: pushable::Pushable>(builder: Builder, value: T) -> Builder {
                     value.bitcoin_script_push(builder)
                 }
                 push(builder, value)

src/parse.rs

@@ -110,9 +110,9 @@ where
                 let inner_block = block.stream();
                 escape.extend(quote! {
                     {
-                        script_var.push(script! {
+                        script_var.extend_from_slice(script! {
                             #inner_block
-                        });
+                        }.as_bytes());
                     }
                 });
 
@@ -130,7 +130,7 @@ where
     escape = quote! {
         {
             #escape;
-            script_var
+            bitcoin::script::ScriptBuf::from(script_var)
         }
     }
     .into();
@@ -152,9 +152,9 @@ where
                 let inner_block = block.stream();
                 escape.extend(quote! {
                     {
-                    script_var.extend_from_slice(script !{
-                        #inner_block
-                    }.as_bytes());
+                        script_var.extend_from_slice(script !{
+                            #inner_block
+                        }.as_bytes());
                     }
                     bitcoin::script::ScriptBuf::from(script_var)
                 });

tests/test.rs

@@ -143,28 +143,48 @@ fn test_if() {
 }
 
 #[test]
-fn test_performance() {
+fn test_performance_loop() {
     define_pushable! {};
     let loop_script = script! {
         OP_ADD
+        OP_ADD
+        OP_ADD
     };
 
     let script = script! {
-        for _ in 0..20_000 {
+        for _ in 0..5_000_000 {
             { loop_script.clone() }
         }
     };
 
-    assert_eq!(script.as_bytes()[0], 147)
+    assert_eq!(script.as_bytes()[5_000_000 - 1], 147)
 }
 
 #[test]
 fn test_performance_no_macro() {
     let mut builder = bitcoin::script::Builder::new();
-    for _ in 0..20_000 {
+    for _ in 0..40_000_000 {
         builder = builder.push_opcode(OP_ADD);
     }
-    
+
     let script = builder.as_script();
-    assert_eq!(script.as_bytes()[0], 147);
+    assert_eq!(script.as_bytes()[40_000_000 - 1], 147);
+}
+
+#[test]
+fn test_performance_if() {
+    define_pushable! {};
+
+    let script = script! {
+        for _ in 0..5_000_000 {
+            if true {
+                OP_ADD
+                OP_ADD
+            } else {
+                OP_ADD
+            }
+        }
+    };
+
+    assert_eq!(script.as_bytes()[5_000_000 - 1], 147)
 }