workers.tcl

#
# Worker Information
#
# Copyright: 2014, iAmShorty
# 
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# 
# http://www.apache.org/licenses/LICENSE-2.0
# 
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
######################################################################
##########           nothing to edit below this line        ##########
##########           use config.tcl for setting options     ##########
######################################################################

#
# get worker information
#
proc worker_info {nick host hand chan arg} {
	global help_blocktime help_blocked channels debug debugoutput output onlyallowregisteredusers ownersworkeronly output_workerinfo output_worker_online output_worker_offline command_protect

	if {$debug eq "1"} { putlog "running proc [dict get [info frame 0] proc]" }
	
	# only allow bot owners to get workers for 
	# specified users
	#
	if {$ownersworkeronly eq "1"} {
		if {[check_userrights $nick] eq "false"} {
			putlog "$nick tried to get worker for user $arg"
			putquick "PRIVMSG $chan :Access to Workers denied, only Botowners can check workers"
			return
		}
	} else {
		if {$onlyallowregisteredusers eq "1"} {
			if {[check_registereduser $chan $nick] eq "false"} {
				putquick "NOTICE $nick :you are not allowed to use this command"
				putquick "NOTICE $nick :please use !request command to get access to the bot"
				return
			}
		}
	}

	if {$arg eq "" || [llength $arg] < 2} {
		if {$debug eq "1"} { putlog "wrong arguments, must be !worker poolname username" }
		return
	}
	
	set action "/index.php?page=api&action=getuserworkers&id=[lindex $arg 1]&api_key="
	
	set mask [string trimleft $host ~]
	regsub -all {@([^\.]*)\.} $mask {@*.} mask	 	
	set mask *!$mask

	if {[info exists help_blocked($mask)]} {
		putquick "NOTICE $nick : You have been blocked for $help_blocktime Seconds, please be patient..."
		return
	}

	set pool_info [regexp -all -inline {\S+} [pool_vars [string toupper [lindex $arg 0]]]]

	if {$pool_info ne "0"} {
		if {$debug eq "1"} { putlog "COIN: [lindex $pool_info 0]" }
		if {$debug eq "1"} { putlog "URL: [lindex $pool_info 1]" }
		if {$debug eq "1"} { putlog "KEY: [lindex $pool_info 2]" }
	} else {
		if {$debug eq "1"} { putlog "no pool data" }
		return
	}

	if {$command_protect eq "1"} {
		if {[channel_command_acl $chan "worker"] eq "False"} {
			putquick "PRIVMSG $chan :command !worker not allowed in $chan"
			return
		}
	}
    
	set newurl [lindex $pool_info 1]
	append newurl $action
	append newurl [lindex $pool_info 2]
	
	set data [check_httpdata $newurl]
	if { [regexp -nocase {error} $data] } {
		putlog $data
		return
	}

	if {$debugoutput eq "1"} { putlog "xml: $data" }

	if {$data eq "Access denied"} { 
		putquick "PRIVMSG $chan :Access to Workers denied"
		return 0
	}

	set results [::json::json2dict $data]

	foreach {key value} $results {
		foreach {sub_key sub_value} $value {
			if {$sub_key eq "data"} {
				#putlog "Sub: $sub_value"
				foreach {elem} $sub_value {
					#putlog "Ele: $elem"
					foreach {elem2 elem_val2} $elem {
						#putlog "Ele: $elem2 - Val: $elem_val2"
						if {$elem2 eq "username"} {
							set worker_name "$elem_val2"
						}
						if {$elem2 eq "hashrate"} {
							if {$elem_val2 eq "0"} {
								set offlineWorkers($worker_name) $elem_val2
							} else {
								set onlineWorkers($worker_name) $elem_val2
							}
						} 						
					}
				}
			}
		}
	}
	
	if {$debug eq "1"} { putlog "onlineWorkers has [array size onlineWorkers] records" }
	if {$debug eq "1"} { putlog "offlineWorkers has [array size offlineWorkers] records" }
	
	if {[lindex $arg 2] eq "active"} {
		if {[array exists onlineWorkers]} {
			foreach key [array names onlineWorkers] {
				if {$debug eq "1"} { putlog "${key}=$onlineWorkers($key)" }
				if {![info exists worker_name]} {
					set lineoutput "${key} - $onlineWorkers($key) KH/s | " 
				} else {
					append lineoutput "${key} - $onlineWorkers($key) KH/s | "
				}
			}
		}
	} elseif {[lindex $arg 2] eq "inactive"} {
		if {[array exists offlineWorkers]} {
			foreach key [array names offlineWorkers] {
				if {$debug eq "1"} { putlog "${key}=$offlineWorkers($key)" }
				if {![info exists worker_name]} {
					set lineoutput "${key} - $offlineWorkers($key) KH/s | " 
				} else {
					append lineoutput "${key} - $offlineWorkers($key) KH/s | "
				}
			}
		}
	} else {
	
		if {[info exists output_workerinfo_percoin([string tolower [lindex $arg 0]])]} {
			if {$debug eq "1"} { putlog "-> [string toupper [lindex $arg 0]] - $output_workerinfo_percoin([string tolower [lindex $arg 0]])" }
				set lineoutput $output_workerinfo_percoin([string tolower [lindex $arg 0]])
			} else {
				if {$debug eq "1"} { putlog "no special output!" }
				set lineoutput $output_workerinfo
		}
		
		set lineoutput [replacevar $lineoutput "%workers_username%" [lindex $arg 1]]
		set lineoutput [replacevar $lineoutput "%workers_coinname%" [string toupper [lindex $pool_info 0]]]
		set lineoutput [replacevar $lineoutput "%workers_online_count%" [array size onlineWorkers]]
		set lineoutput [replacevar $lineoutput "%workers_offline_count%" [array size offlineWorkers]]
	}
	
	# split message if buffer is to big
	#
	set len [expr {512-[string len ":$::botname PRIVMSG $chan :\r\n"]}] 
	foreach line [wordwrap $lineoutput $len] { 
		if {$output eq "CHAN"} {
			putquick "PRIVMSG $chan :$line"
		} elseif {$output eq "NOTICE"} {
			putquick "NOTICE $nick :$line"
		} else {
			putquick "PRIVMSG $chan :please set output in config file"
			return 0
		}
	}
}

putlog "===>> Mining-Pool-Workers - Version $scriptversion loaded"