httpsnippet-fsless/src/targets/go/native.js

HTTP Request snippet generator for *most* languages without using node-fs

/**
 * @description
 * HTTP code snippet generator for native Go.
 *
 * @author
 * @montanaflynn
 *
 * for any questions or issues regarding the generated code snippet, please open an issue mentioning the author.
 */

'use strict'

var util = require('util')
var CodeBuilder = require('../../helpers/code-builder')

module.exports = function (source, options) {
  // Let's Go!
  var code = new CodeBuilder('\t')

  // Define Options
  var opts = util._extend({
    checkErrors: false,
    printBody: true,
    timeout: -1
  }, options)

  var errorPlaceholder = opts.checkErrors ? 'err' : '_'

  var errorCheck = function () {
    if (opts.checkErrors) {
      code.push(1, 'if err != nil {')
          .push(2, 'panic(err)')
          .push(1, '}')
    }
  }

  // Create boilerplate
  code.push('package main')
      .blank()
      .push('import (')
      .push(1, '"fmt"')

  if (opts.timeout > 0) {
    code.push(1, '"time"')
  }

  if (source.postData.text) {
    code.push(1, '"strings"')
  }

  code.push(1, '"net/http"')

  if (opts.printBody) {
    code.push(1, '"io/ioutil"')
  }

  code.push(')')
      .blank()
      .push('func main() {')
      .blank()

  // Create client
  var client
  if (opts.timeout > 0) {
    client = 'client'
    code.push(1, 'client := http.Client{')
        .push(2, 'Timeout: time.Duration(%s * time.Second),', opts.timeout)
        .push(1, '}')
        .blank()
  } else {
    client = 'http.DefaultClient'
  }

  code.push(1, 'url := "%s"', source.fullUrl)
      .blank()

  // If we have body content or not create the var and reader or nil
  if (source.postData.text) {
    code.push(1, 'payload := strings.NewReader(%s)', JSON.stringify(source.postData.text))
        .blank()
        .push(1, 'req, %s := http.NewRequest("%s", url, payload)', errorPlaceholder, source.method)
        .blank()
  } else {
    code.push(1, 'req, %s := http.NewRequest("%s", url, nil)', errorPlaceholder, source.method)
        .blank()
  }

  errorCheck()

  // Add headers
  if (Object.keys(source.allHeaders).length) {
    Object.keys(source.allHeaders).map(function (key) {
      code.push(1, 'req.Header.Add("%s", "%s")', key, source.allHeaders[key])
    })
    code.blank()
  }

  // Make request
  code.push(1, 'res, %s := %s.Do(req)', errorPlaceholder, client)
  errorCheck()

  // Get Body
  if (opts.printBody) {
    code.blank()
        .push(1, 'defer res.Body.Close()')
        .push(1, 'body, %s := ioutil.ReadAll(res.Body)', errorPlaceholder)
    errorCheck()
  }

  // Print it
  code.blank()
      .push(1, 'fmt.Println(res)')

  if (opts.printBody) {
    code.push(1, 'fmt.Println(string(body))')
  }

  // End main block
  code.blank()
      .push('}')

  return code.join()
}

module.exports.info = {
  key: 'native',
  title: 'NewRequest',
  link: 'http://golang.org/pkg/net/http/#NewRequest',
  description: 'Golang HTTP client request'
}