#!/bin/bash
#update_description.sh

DB_USER='root' #'your_db_username'
DB_PASS='root' #'your_password'
DB='my_genie_sys_database' #'database_name'

# if less than two arguments supplied, display  error message
	if [  $# -le 0 ]
	then
		start='\033[0;33m'
		start_0='\033[0;33m'
		start_2='\033[0;31m' 
		end='\033[0m'
		echo  "\nUsage:\n$0 ${start}[gene_info/transcript_info] [file_name]${end}\nEx: ${start_2}sh update_descriptions.sh transcript_info/gene_info potra_descriptions.tsv${end}\n\nWhat it does?\n${start_0}This script will create a two columns(ids, descriptions) temporary table and load the [file_name] into it.\nThen it will match ids column in temporary table with transcript_ids/gene_ids and update the gene/transcript descriptions.\nFinally delete the temporary table.\n${end}"
		exit 1
	fi

table_name=$(echo $1 | awk '{split($0,a,"_");print a[1]}');
tmp_field_name=$table_name"_id"
mysql --host=localhost --user=$DB_USER --password=$DB_PASS --local_infile=1 --database=$DB <<EOFMYSQL
DROP TEMPORARY TABLE  IF EXISTS tmp_tb ;
UPDATE $1 SET description = '';
CREATE TEMPORARY TABLE tmp_tb(gene_name VARCHAR(255),annotation VARCHAR(500), PRIMARY KEY(gene_name));
load data local infile '$2' replace INTO TABLE tmp_tb fields terminated by '\t' LINES TERMINATED BY '\n' ignore 0 lines;
#DROP INDEX if exists $tmp_field_name ON $tmp_field_name; 
#ALTER TABLE $1 ADD INDEX $tmp_field_name ($tmp_field_name);
UPDATE $1 INNER JOIN tmp_tb on tmp_tb.gene_name = $1.$tmp_field_name SET $1.description = tmp_tb.annotation;
#select * from tmp_tb limit 100;
DROP TEMPORARY TABLE tmp_tb;
EOFMYSQL
#Following line is very important when we load data into the corresponding table. If we did not make index it takes so long time to load.
#ALTER TABLE transcript_info ADD INDEX `transcript_id` (`transcript_id`)