commit c1c5d590f281b36791e8f024ff3113201a06b5f3 · kitten.sh/wonka

+50 -1
scripts/minify-bucklescript-plugin.js
···

       110
       110
        
         };

     

       111
       111
        
       }

     

       112
       112
        
       

     

       113
       113
       +
       function squashImplicitUnitReturn({ types: t }) {

     

       114
       114
       +
         return {

     

       115
       115
       +
           visitor: {

     

       116
       116
       +
             ReturnStatement(path) {

     

       117
       117
       +
               if (

     

       118
       118
       +
                 t.isCallExpression(path.node.argument) &&

     

       119
       119
       +
                 t.isIdentifier(path.node.argument.callee) &&

     

       120
       120
       +
                 (path.node.argument.callee.name === 'sink' || path.node.argument.callee.name === 'source')

     

       121
       121
       +
               ) {

     

       122
       122
       +
                 path.replaceWithMultiple([

     

       123
       123
       +
                   t.expressionStatement(path.node.argument),

     

       124
       124
       +
                   t.returnStatement(),

     

       125
       125
       +
                 ]);

     

       126
       126
       +
               }

     

       127
       127
       +
             },

     

       128
       128
       +
             Function: {

     

       129
       129
       +
               exit(functionPath) {

     

       130
       130
       +
                 if (t.isIdentifier(functionPath.id) && functionPath.id.name === 'valFromOption') return;

     

       131
       131
       +
       

     

       132
       132
       +
                 let hasEmptyReturn = false;

     

       133
       133
       +
                 let hasCallReturnOnly = true;

     

       134
       134
       +
                 functionPath.traverse({

     

       135
       135
       +
                   Function(innerPath) {

     

       136
       136
       +
                     innerPath.skip();

     

       137
       137
       +
                   },

     

       138
       138
       +
                   ReturnStatement: {

     

       139
       139
       +
                     enter(path) {

     

       140
       140
       +
                       if (path.node.argument === null) {

     

       141
       141
       +
                         hasEmptyReturn = true;

     

       142
       142
       +
                       } else if (!t.isCallExpression(path.node.argument)) {

     

       143
       143
       +
                         hasCallReturnOnly = false;

     

       144
       144
       +
                       }

     

       145
       145
       +
                     },

     

       146
       146
       +
                     exit(path) {

     

       147
       147
       +
                       if (hasEmptyReturn && hasCallReturnOnly && path.node.argument !== null) {

     

       148
       148
       +
                         path.replaceWithMultiple([

     

       149
       149
       +
                           t.expressionStatement(path.node.argument),

     

       150
       150
       +
                           t.returnStatement(),

     

       151
       151
       +
                         ]);

     

       152
       152
       +
                       }

     

       153
       153
       +
                     },

     

       154
       154
       +
                   },

     

       155
       155
       +
                 });

     

       156
       156
       +
               },

     

       157
       157
       +
             },

     

       158
       158
       +
           },

     

       159
       159
       +
         };

     

       160
       160
       +
       }

     

       161
       161
       +
       

     

       113
       162
        
       function cleanup(opts = {}) {

     

       114
       163
        
         const filter = createFilter(opts.include, opts.exclude, {

     

       115
       164
        
           resolve: false,

     
···

       124
       173
        
             }

     

       125
       174
        
       

     

       126
       175
        
             return transform(code, {

     

       127
       127
       -
               plugins: [unwrapStatePlugin, curryGuaranteePlugin],

     

       176
       176
       +
               plugins: [unwrapStatePlugin, curryGuaranteePlugin, squashImplicitUnitReturn],

     

       128
       177
        
               babelrc: false,

     

       129
       178
        
             });

     

       130
       179
        
           },