diff --git a/wrenchboard/src/shared_tool/account.cc b/wrenchboard/src/shared_tool/account.cc index cfe15e48..d91544c2 100644 --- a/wrenchboard/src/shared_tool/account.cc +++ b/wrenchboard/src/shared_tool/account.cc @@ -532,13 +532,21 @@ long WrenchLoadReferLink(CVars in, CVars &out) { logfmt(logINFO, "WrenchMyPageIntro()"); long ret = PHP_API_BAD_PARAM; const PGresult *res; + char buf[1500]; CVars x; try { REQ_STRING(in, "refer_link", 20, 100, "(.*)"); + long action_path = REQ_LONG(in, "action_path", 1, -1); - if ( load_db_record(out, "SELECT firstname, lastname, email FROM members_refer_friend " - " WHERE refer_link='%s' AND status = 1 ", in["refer_link"].c_str()) > 0){ + if (action_path == 200){ + sprintf(buf, "SELECT firstname, lastname, email FROM promo_member WHERE uid::text='%s' AND status = 1", in["refer_link"].c_str()); + } + else{ + sprintf(buf, "SELECT firstname, lastname, email FROM members_refer_friend WHERE refer_link='%s' AND status = 1", in["refer_link"].c_str()); + } + + if ( load_db_record(out, "%s ", buf) > 0){ ret = PHP_API_OK; } else{ diff --git a/www-api/app/Controllers/WrenchAccount.php b/www-api/app/Controllers/WrenchAccount.php index 7bf626fe..b479fc2e 100644 --- a/www-api/app/Controllers/WrenchAccount.php +++ b/www-api/app/Controllers/WrenchAccount.php @@ -52,7 +52,12 @@ class WrenchAccount extends BaseController $raw_array = json_decode($raw_json, true); $in = $raw_array; $in["action"] = WRENCHBOARD_LOAD_REFERLINK; + $in["action_path"] = 100; $in["loc"] = $_SERVER["REMOTE_ADDR"]; + if (isset($in["path"]) && $in["path"] == 'promoref'){ + $in["refer_link"] = str_replace('', "REFPROMO-", $in["refer_link"]); + $in["action_path"] = 200; + } $ret = $this->wrenchboard->wrenchboard_api($in, $out); $out['internal_return'] = $ret; return $this->respond( $this->summaryReturnData($in,$out), 200);