美文网首页Swift CafeiOS Developer
一个活在命令行里的程序员给我们分享的代码

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

作者: SwiftCafe | 来源:发表于2017-07-29 15:23 被阅读197次

    这篇内容源自 Github 上面一个有意思的分享。说的是这个分享者的一个前同事,作者说他是活在命令行中的人。只有要任何工作会耗费他 90 秒以上的时间,他就会写一个脚本来自动处理。

    这位程序员写的这些脚本被收集整理后发到了 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="Late at work. "$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 [ $? -gt 0 ]; then
      echo "Failed to send SMS: $RESPONSE"
      exit 1
    fi
    

    这个脚本中,先用 whogrep 命令检测当前正在工作的终端用户有没有自己,如果没有就直接退出,如果有则继续操作。 接下来预设一些变量,比如自己和老婆的电话号码,加班原因。 然后通过 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="Gonna work from home. "$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 [ $? -gt 0 ]; then
      echo "Failed to send SMS: $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。 当然也欢迎你的留言交流。

    更多精彩内容可关注微信公众号:
    swift-cafe

    相关文章

      网友评论

      • XIAODAO:你好,请问Using Swift with Cocoa and Objective-C有swift3.0或者更高版本的中文版吗?
      • cf2632a8e56a:真几把6
      • ce4c10d6271a:小白想问问这种脚本是放在终端里运行还是用什么工具生成一个应用程序

      本文标题:一个活在命令行里的程序员给我们分享的代码

      本文链接:https://www.haomeiwen.com/subject/tocclxtx.html