返回 登录
1

一个活在命令行里的程序员给我们分享的代码

阅读378

这位程序员写的这些脚本被收集整理后发到了 Github 上面,和大家一起分享,并且已经获得了 2 万多颗星星。

smack-my-bitch-up.sh: 这个脚本用于在加班的时候自动给他的老婆发短信。具体是怎么做到的呢,每天晚上 9 点的时候会检测这台电脑的命令行是否还在工作。如果命令行还在工作,就会从预置的三条信息中随机选一条发出去。
来看看他是怎么做到的:

!/bin/sh -e

Exit early if no sessions with my username are found

if ! who | grep -wq $USER; then
exit
fi

Phone numbers

MY_NUMBER=’+xxx’
HER_NUMBER=’+xxx’

REASONS=(
‘Working hard’
‘Gotta ship this feature’
‘Someone fucked the system again’
)
rand=[RANDOM % ${#REASONS[@]} ]

RANDOM_REASON=REASONS[$rand]MESSAGE=Lateatwork.RANDOM_REASON

Send a text message

RESPONSE=curl -fSs -u "$TWILIO_ACCOUNT_SID:$TWILIO_AUTH_TOKEN" \
-d "From=$MY_NUMBER" -d "To=$HER_NUMBER" -d "Body=$MESSAGE" \
"https://api.twilio.com/2010-04-01/Accounts/$TWILIO_ACCOUNT_SID/Messages"

Log errors

if [ ?gt0];thenechoFailedtosendSMS:RESPONSE”
exit 1
fi
这个脚本中,先用 who 和 grep 命令检测当前正在工作的终端用户有没有自己,如果没有就直接退出,如果有则继续操作。 接下来预设一些变量,比如自己和老婆的电话号码,加班原因。 然后通过 curl 命令调用 api.twilio.com 的短信发送接口将信息发出去。

最后判断发送是否成功,如果发送失败,将日志打印出来。看起来挺复杂的操作,其实看看实际的脚本并不困难吧。

kumar-asshole.sh: 这个也很有意思,从邮箱的收件箱里面扫描一个叫 Kumar(公司一个客户的 DBA) 的人发来的邮件,从邮件的内容中确认是否包含 help,trouble,sorry 这类词语,如果包含,就自动通过 SSH 链接这个客户的数据库,并将数据库回滚到上一次备份。然后再回一封邮件说:“不要担心,下次注意就好。”
来看看这个脚本:

!/usr/bin/env ruby

require ‘dotenv’
require ‘gmail’

Dotenv.load

GMAIL_USERNAME = ENV[‘GMAIL_USERNAME’]
GMAIL_PASSWORD = ENV[‘GMAIL_PASSWORD’]

GMAIL = Gmail.connect(GMAIL_USERNAME, GMAIL_PASSWORD)
KUMARS_EMAIL = ‘kumar.a@example.com’

DB_NAME_REGEX = /\S+_staging/
KEYWORDS_REGEX = /sorry|help|wrong/i

def create_reply(subject)
GMAIL.compose do
to KUMARS_EMAIL
subject “RE: #{subject}”
body “No problem. I’ve fixed it. \n\n Please be careful next time.”
end
end

GMAIL.inbox.find(:unread, from: KUMARS_EMAIL).each do |email|
if email.body.raw_source[KEYWORDS_REGEX] && (db_name = email.body.raw_source[DB_NAME_REGEX])
backup_file = “/home/backups/databases/#{db_name}-” + (Date.today - 1).strftime(‘%Y%m%d’) + ‘.gz’
abort ‘ERROR: Backup file not found’ unless File.exist?(backup_file)

# Restore DB
`gunzip -c #{backup_file} | psql #{db_name}`

# Mark as read, add label and reply
email.read!
email.label('Database fixes')
reply = create_reply(email.subject)
GMAIL.deliver(reply)

end
end
这个脚本是用 ruby 写的。 首先,调用 GMAIL.inbox.find 从收件箱中查找符合条件的邮件 — 发件人为 kumar.a@example.com,并且包含关键字 KEYWORDS_REGEX = /sorry|help|wrong/i。

如果找到相关邮件,先判断数据备份文件 backup_file 是否存在。 然后调用 gunzip 和 psql 命令还原这个备份。 最后调用 create_reply 方法创建回复邮件,并发出去。

这个脚本虽然说很方便,但还是比较依赖具体场景的。如果使用不得当也有可能会造成误还原等。当然它还是能帮我们打开一些思路的。

hangover.sh:又是一个定时通知的程序,这个在每天早上 8:45 的时候检测服务器的命令行中有没有自己登录,如果没有,自动给老板发短信请假。
代码如下:

!/bin/sh -e

Exit early if any session with my username is found

if who | grep -wq $USER; then
exit
fi

Phone numbers

MY_NUMBER=’+xxx’
NUMBER_OF_BOSS=’+xxx’

EXCUSES=(
‘Locked out’
‘Pipes broke’
‘Food poisoning’
‘Not feeling well’
)
rand=[RANDOM % ${#EXCUSES[@]} ]

RANDOM_EXCUSE=EXCUSES[$rand]MESSAGE=Gonnaworkfromhome.RANDOM_EXCUSE

Send a text message

RESPONSE=curl -fSs -u "$TWILIO_ACCOUNT_SID:$TWILIO_AUTH_TOKEN" \
-d "From=$MY_NUMBER" -d "To=$NUMBER_OF_BOSS" -d "Body=$MESSAGE" \
"https://api.twilio.com/2010-04-01/Accounts/$TWILIO_ACCOUNT_SID/Messages"

Log errors

if [ ?gt0];thenechoFailedtosendSMS:RESPONSE”
exit 1
fi
这个脚本基本逻辑和我们前面的差不多,就不多赘述了。

fucking-coffee.sh:这个是这里面最有意思的脚本,脚本运行时会先等待 17 秒,然后通过 telnet 给咖啡机发送命令制作一个中杯拿铁咖啡。咖啡制作完成后,再等待 24 秒,然后把制作好的咖啡放到杯子里面。 这个整好是这个程序员从桌子走到咖啡机前的时间。
当然,前提是你得有一台可以联网并且接收命令的咖啡机。

脚本如下:

!/usr/bin/env ruby

Exit early if no sessions with my username are found

exit unless who -q.include? ENV[‘USER’]

require ‘net/telnet’

coffee_machine_ip = ‘10.10.42.42’
password = ‘1234’
password_prompt = ‘Password: ’
delay_before_brew = 17
delay = 24

sleep delay_before_brew
con = Net::Telnet.new(‘Host’ => coffee_machine_ip)
con.cmd(‘String’ => password, ‘Match’ => /#{password_prompt}/)
con.cmd(‘sys brew’)
sleep delay
con.cmd(‘sys pour’)
con.close
这个脚本也很好理解,首先判断当前用户是否登录,只有用户登录才会发送下面的命令给咖啡机。

脚本介绍完了,要让这些跑起来还需要做一些准备工作:

used in smack-my-bitch-up and hangover scripts

TWILIO_ACCOUNT_SID=ACxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
TWILIO_AUTH_TOKEN=yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyy

used in kumar_asshole script

GMAIL_USERNAME=admin@example.org
GMAIL_PASSWORD=password
这个是设置 TWILIO 平台的 Key, 你需要注册一个,地址是 https://www.twilio.com。 还需要按照上面的格式设置一下你的 GMAIL 用户名和密码。

然后还需要在 Cron 里面设置一下定时执行任务的规则:

Runs smack-my-bitch-up.sh monday to friday at 9:20 pm.

20 21 * * 1-5 /path/to/scripts/smack-my-bitch-up.sh >> /path/to/smack-my-bitch-up.log 2>&1

Runs hangover.sh monday to friday at 8:45 am.

45 8 * * 1-5 /path/to/scripts/hangover.sh >> /path/to/hangover.log 2>&1

Runs kumar-asshole.sh every 10 minutes.

/10 * * * /path/to/scripts/kumar-asshole.sh

Runs fucking-coffee.sh hourly from 9am to 6pm on weekdays.

0 9-18 * * 1-5 /path/to/scripts/fucking-coffee.sh
结束
看完这篇文章后,你是觉得这些脚本非常有创意,还是很无聊呢? 这个就见仁见智了。 不过这位有意思的程序员确实帮我们扩展了一下思路。命令行脚本能做的事情比我们平时想到的还要多。

完整内容可以参看它的 Github 主页:https://github.com/NARKOZ/hacker-scripts。 当然也欢迎你的留言交流。

如果你觉得这篇文章有帮助,还可以QQ群:656164577,会有更多我的原创内容分享给你~

评论