fix(core & build): $refs 不生效 (#153)

Author: meixg

packages/mars-build/src/compiler/template/render.js

@@ -195,7 +195,7 @@ function genData(el, state) {
 
     // TODO: better genProps, omit static attrs
     // attributes
-    if (attrs.length > 0) {
+    if (attrs.length > 0 || el.ref) {
         const props = genProps(attrs, state, el);
         data += el.isComp ? `[_pp(${props})],` : `[${props}],`;
     }
@@ -386,6 +386,10 @@ function genProps(props, state, el) {
     if (fData.length) {
         res += `_p:${state.processFilterData(fData, 'p', el)},`;
     }
+
+    if (el.ref) {
+        res += `'ref': ${el.ref},`;
+    }
     return res.slice(0, -1) + '}';
 }
 

packages/mars-core/src/base/createComponent.js

@@ -76,6 +76,7 @@ export function makeCreateComponent(handleProxy, handleModel, setData, callHook)
         let props = normalizeProps(options.props);
         props = Object.assign(props, {
             compId: String,
+            ref: String,
             rootComputed: Object,
             rootUID: Number
         });

packages/mars-core/src/base/mixins.js

@@ -4,7 +4,7 @@
  */
 
 import {getPageInstance} from '../helper/instance';
-import {setObjectData} from '../helper/util';
+import {setObjectData, remove} from '../helper/util';
 import config from '../config';
 
 export function makePageMixin($api) {
@@ -33,7 +33,8 @@ export function makeGetCompMixin($api) {
     return function getCompMixin(options) {
         return {
             props: {
-                compId: String
+                compId: String,
+                ref: String
             },
             beforeCreate() {
                 this.$api = $api;
@@ -54,6 +55,20 @@ export function makeGetCompMixin($api) {
 
                 // 此时还没有 .$mp
                 this.$options.mpInstance.__curSwan__ = curIndex;
+
+                registerRef(this, this.ref);
+            },
+            destroyed() {
+                registerRef(this, this.ref, true);
+            },
+            watch: {
+                ref(newVal, val) {
+                    if (val !== newVal) {
+                        registerRef(this, val, true);
+                        registerRef(this, newVal);
+                        return;
+                    }
+                }
             },
             updated() {
                 this.$emit('vm.updated');
@@ -142,3 +157,34 @@ export function handleModel(event) {
 
 
 }
+
+function registerRef(vm, key, isRemoval) {
+    if (!key) {
+        return;
+    }
+
+    const refs = vm.$parent.$refs;
+    const ref = vm;
+
+    if (isRemoval) {
+        if (Array.isArray(refs[key])) {
+            remove(refs[key], ref);
+        }
+        else if (refs[key] === ref) {
+            refs[key] = undefined;
+        }
+    }
+    else {
+        if (vm.compId && vm.compId.split(',').pop().indexOf('-') !== -1) {
+            if (!Array.isArray(refs[key])) {
+                refs[key] = [ref];
+            }
+            else if (refs[key].indexOf(ref) < 0) {
+                refs[key].push(ref);
+            }
+        }
+        else {
+            refs[key] = ref;
+        }
+    }
+}

packages/mars-core/src/base/vue/vue.runtime.esm.js

@@ -104,7 +104,7 @@ var isBuiltInTag = makeMap('slot,component', true);
 /**
  * Check if an attribute is a reserved attribute.
  */
-var isReservedAttribute = makeMap('key,ref,slot,slot-scope,is');
+var isReservedAttribute = makeMap('key,slot,slot-scope,is');
 
 /**
  * Remove an item from an array.

packages/mars-core/src/helper/util.js

@@ -49,3 +49,15 @@ export function setObjectData(target, model, value) {
         }
     }
 }
+
+/**
+ * Remove an item from an array.
+ */
+export function remove(arr, item) {
+    if (arr.length) {
+        const index = arr.indexOf(item);
+        if (index > -1) {
+            return arr.splice(index, 1);
+        }
+    }
+}