#!/bin/bash

function iosheader {
cat <<EOF
//
// DJI GO 4 - Frida Tweaks
//

if (ObjC.available) {

EOF
}

function iosfooter {
cat <<EOF
}
EOF
}

function functionheader {
cat <<EOF

  //
  // ${1}
  //

EOF
}
function jsobject {

cat <<EOF
  var ${1} = ObjC.classes.${1};
  var ${3} = ${1}['${2} ${3}:'];
  var ${3}Impl = ${3}.implementation;
  ${3}.implementation = ObjC.implement(${3}, function (handle, selector, originalResult) {
    ${3}Impl(handle, selector, ${4});
    if ( originalResult != ${4} ) {
      if ( ${7} == 0 || ${CLASS}${METHOD}SeenReplace == 0 ) {
        console.log("${5}");
        ${CLASS}${METHOD}SeenReplace = 1;
      }
    } else {
      if ( ${7} == 0 || ${CLASS}${METHOD}SeenHit == 0 ) {
        console.log("${6}");
        ${CLASS}${METHOD}SeenHit = 1;
      }
    }
  });
EOF
}

function jsfunction {
cat <<EOF
  var ${1} = ObjC.classes.${1};
  var ${3} = ${1}['${2} ${3}'];
  var ${3}Impl = ${3}.implementation;
  ${3}.implementation = ObjC.implement(${3}, function (handle, selector) {
    var originalResult = ${3}Impl(handle, selector);
    if ( originalResult != ${4} ) {
      if ( ${7} == 0 || ${CLASS}${METHOD}SeenReplace == 0 ) {
        console.log("${5}");
        ${CLASS}${METHOD}SeenReplace = 1;
      }
    } else {
      if ( ${7} == 0 || ${CLASS}${METHOD}SeenHit == 0 ) {
        console.log("${6}");
        ${CLASS}${METHOD}SeenHit = 1;
      }
    }
    return ${4};
  });
EOF
}


function jsfunctioniffcc {
cat <<EOF
  var ${1} = ObjC.classes.${1};
  var ${3} = ${1}['${2} ${3}'];
  var ${3}Impl = ${3}.implementation;
  ${3}.implementation = ObjC.implement(${3}, function (handle, selector) {
    var originalResult = ${3}Impl(handle, selector);
    if ( originalResult != ${4} ) {
      if ( ${7} == 0 || ${CLASS}${METHOD}SeenReplace == 0 ) {
        console.log("${5}");
        ${CLASS}${METHOD}SeenReplace = 1;
      }
    }

    if( DJIAppSettingssdr_force_fccSeenReplace == 0 ) {
      return ${4}; //mavic
    } else {
      return originalResult;
    }
  });
EOF
}


function defines {
cat <<EOF
//
// Create variables that we can use for "one shot" rules
//
EOF

while read X
do
        #TEMPLATE=`echo $X | cut -d "," -f 1`
        CLASS=`echo $X | cut -d "," -f 2`
        METHOD=`echo $X | cut -d "," -f 3`
	PLUSMINUS=`echo $X | cut -d "," -f 4`
        #REPLACE=`echo $X | cut -d "," -f 5`
        #LOGHIT=`echo $X | cut -d "," -f 6`
        #LOGCHANGED=`echo $X | cut -d "," -f 7`
	#ONESHOT=`echo $X | cut -d "," -f 8`

	echo "${CLASS}${METHOD}SeenHit=0;"
	echo "${CLASS}${METHOD}SeenReplace=0;"

done < PrettyWoman.csv
}

function mainloop {
while read X
do
        TEMPLATE=`echo $X | cut -d "," -f 1`
        CLASS=`echo $X | cut -d "," -f 2`
        METHOD=`echo $X | cut -d "," -f 3`
	PLUSMINUS=`echo $X | cut -d "," -f 4`
        REPLACE=`echo $X | cut -d "," -f 5`
        LOGHIT=`echo $X | cut -d "," -f 6`
        LOGCHANGED=`echo $X | cut -d "," -f 7`
	ONESHOT=`echo $X | cut -d "," -f 8`

        functionheader "${CLASS} ${METHOD} ${TEMPLATE}"
        ${TEMPLATE} ${CLASS} ${PLUSMINUS} ${METHOD} ${REPLACE} "${LOGHIT}" "${LOGCHANGED}" "0${ONESHOT}"
done < PrettyWoman.csv
}

iosheader
defines
mainloop
iosfooter