feat: 增加对规则对env的提前识别

Author: heiazu

__test__/fixure/style.bin

@@ -8,4 +8,4 @@ Generated by [AVA](https://avajs.dev).
 
 > Snapshot 1
 
-    '{"fonts":[],"keyframes":[],"medias":[],"styles":[{"declarations":[[66,100],[69,"100px"],[67,"100px"],[68,"100px"]],"media":0,"selector":["px"]}]}'
+    '{"keyframes":[],"medias":[],"styles":[{"declarations":[[66,100],[69,"100px"],[67,"100px"],[68,"100px"]],"has_env":false,"media":0,"selector":["px"]}]}'

__test__/index.spec.mjs.md

@@ -4,6 +4,7 @@ use swc_core::common::DUMMY_SP;
 use swc_core::ecma::ast::*;
 
 use crate::constants::{Pseudo, SUPPORT_PSEUDO_KEYS};
+use crate::parse_style_properties::DeclsAndVars;
 use crate::style_propetries::style_value_type::StyleValueType;
 
 use crate::style_parser::{FontFaceItem, KeyFrameItem};
@@ -13,7 +14,7 @@ use crate::visitor::parse_style_values;
 use crate::{generate_expr_lit_num, generate_expr_lit_str, utils};
 
 pub struct JsonWriter {
-  styles: IndexMap<(u32, String), Vec<StyleValueType>>,
+  styles: IndexMap<(u32, String), DeclsAndVars>,
   keyframes: IndexMap<(u32, String), Vec<KeyFrameItem>>,
   medias: Vec<StyleMedia>,
   fonts: Vec<FontFaceItem>,
@@ -22,7 +23,7 @@ pub struct JsonWriter {
 
 impl JsonWriter {
   pub fn new(
-    styles: IndexMap<(u32, String), Vec<StyleValueType>>,
+    styles: IndexMap<(u32, String), DeclsAndVars>,
     keyframes: IndexMap<(u32, String), Vec<KeyFrameItem>>,
     medias: Vec<StyleMedia>,
     fonts: Vec<FontFaceItem>,
@@ -41,7 +42,7 @@ impl JsonWriter {
     let elems: Vec<Expr> = self
       .styles
       .iter()
-      .filter_map(|((media_index, selector), prop_or_spreads)| {
+      .filter_map(|((media_index, selector), item)| {
         Some({
           // 识别伪类
           let mut new_selector = selector.clone();
@@ -113,9 +114,13 @@ impl JsonWriter {
               key: PropName::Ident(Ident::new("declarations".into(), DUMMY_SP)),
               value: Box::new(Expr::Array(ArrayLit {
                 span: DUMMY_SP,
-                elems: parse_style_values(prop_or_spreads.clone(), Platform::Harmony),
+                elems: parse_style_values(item.decls.clone(), Platform::Harmony),
               })),
             }))),
+            PropOrSpread::Prop(Box::new(Prop::KeyValue(KeyValueProp {
+              key: PropName::Ident(Ident::new("has_env".into(), DUMMY_SP)),
+              value: Box::new(Expr::Lit(Lit::Bool(Bool { span: DUMMY_SP, value: item.has_env }))),
+            })))
           ];
 
           if pseudo_key.len() > 0 {

__test__/index.spec.mjs.snap

@@ -75,8 +75,15 @@ use crate::{
   utils::lowercase_first,
 };
 
-pub fn parse_style_properties(properties: &Vec<(String, Property)>) -> Vec<StyleValueType> {
+#[derive(Debug, Clone)]
+pub struct DeclsAndVars {
+  pub decls:Vec<StyleValueType>,
+  pub has_env: bool
+}
+
+pub fn parse_style_properties(properties: &Vec<(String, Property)>) -> DeclsAndVars {
   let mut final_properties = vec![];
+  let mut has_env = false;
   for (id, value) in properties.iter() {
     let mut is_env: bool = false;
     match value {
@@ -98,6 +105,7 @@ pub fn parse_style_properties(properties: &Vec<(String, Property)>) -> Vec<Style
       _ => {}
     };
     if is_env {
+      has_env = true;
       continue;
     }
 
@@ -539,5 +547,8 @@ pub fn parse_style_properties(properties: &Vec<(String, Property)>) -> Vec<Style
     }
   }
 
-  final_properties
+  DeclsAndVars {
+    has_env: has_env,
+    decls: final_properties
+  }
 }

pnpm-lock.yaml

@@ -1,6 +1,7 @@
 use std::{cell::RefCell, convert::Infallible, rc::Rc};
 
 use super::parse_style_properties::parse_style_properties;
+use crate::parse_style_properties::DeclsAndVars;
 use crate::{generate_expr_enum, generate_expr_lit_str};
 use crate::style_propetries::font_weight::{self, FontWeight};
 use crate::style_propetries::style_property_enum::ArkUI_FontWeight;
@@ -30,7 +31,7 @@ use crate::style_propetries::style_media::StyleMedia;
 pub type StyleValue = Vec<StyleValueType>;
 #[derive(Debug)]
 pub struct StyleData {
-  pub all_style: Rc<RefCell<IndexMap<(u32, String), StyleValue>>>,
+  pub all_style: Rc<RefCell<IndexMap<(u32, String), DeclsAndVars>>>,
   pub all_keyframes: Rc<RefCell<IndexMap<(u32, String), Vec<KeyFrameItem>>>>,
   pub all_medias: Rc<RefCell<Vec<StyleMedia>>>,
   pub all_fonts: Rc<RefCell<Vec<FontFaceItem>>>,
@@ -213,7 +214,7 @@ impl<'i> Visitor<'i> for StyleVisitor<'i> {
                   KeyframeSelector::From => 0.0,
                   KeyframeSelector::To => 1.0,
                 },
-                declarations: parse_style_properties(&properties),
+                declarations: parse_style_properties(&properties).decls,
               };
 
               keyframe_data.keyframes.push(keyframe_item)