Exemplo n.º 1
0
// AttachFileAction attaches a file to an alert at OpsGenie.
func AttachFileAction(c *gcli.Context) {
	cli, err := NewAlertClient(c)
	if err != nil {
		os.Exit(1)
	}

	req := alerts.AttachFileAlertRequest{}
	if val, success := getVal("id", c); success {
		req.ID = val
	}
	if val, success := getVal("alias", c); success {
		req.Alias = val
	}
	if val, success := getVal("attachment", c); success {
		f, err := os.Open(val)
		defer f.Close()
		if err != nil {
			fmt.Printf("%s\n", err.Error())
			os.Exit(1)
		}
		req.Attachment = f
	}

	if val, success := getVal("indexFile", c); success {
		req.IndexFile = val
	}

	req.User = grabUsername(c)
	if val, success := getVal("source", c); success {
		req.Source = val
	}
	if val, success := getVal("note", c); success {
		req.Note = val
	}

	printVerboseMessage("Attach request prepared from flags, sending request to OpsGenie..")

	_, err = cli.AttachFile(req)
	if err != nil {
		fmt.Printf("%s\n", err.Error())
		os.Exit(1)
	}

	printVerboseMessage("File attached to alert successfully.")
}
Exemplo n.º 2
0
// AttachFile method attaches a file to an alert at OpsGenie.
func (cli *OpsGenieAlertClient) AttachFile(req alerts.AttachFileAlertRequest) (*alerts.AttachFileAlertResponse, error) {
	req.APIKey = cli.apiKey
	var b bytes.Buffer
	w := multipart.NewWriter(&b)

	path := req.Attachment.Name()
	file, err := os.Open(path)
	defer file.Close()
	if err != nil {
		message := "Attachment can not be opened for reading. " + err.Error()
		logging.Logger().Warn(message)
		return nil, errors.New(message)
	}
	// add the attachment
	fw, err := w.CreateFormFile("attachment", filepath.Base(path))
	if err != nil {
		message := "Can not build the request with the field attachment. " + err.Error()
		logging.Logger().Warn(message)
		return nil, errors.New(message)
	}
	if _, err := io.Copy(fw, file); err != nil {
		message := "Can not copy the attachment into the request. " + err.Error()
		logging.Logger().Warn(message)
		return nil, errors.New(message)
	}

	// Add the other fields
	// empty fields should not be placed into the request
	// otherwise it yields an incomplete boundary exception
	if req.APIKey != "" {
		if err = writeField(*w, "apiKey", req.APIKey); err != nil {
			return nil, err
		}
	}
	if req.ID != "" {
		if err = writeField(*w, "id", req.ID); err != nil {
			return nil, err
		}
	}
	if req.Alias != "" {
		if err = writeField(*w, "alias", req.Alias); err != nil {
			return nil, err
		}
	}
	if req.User != "" {
		if err = writeField(*w, "user", req.User); err != nil {
			return nil, err
		}
	}
	if req.Source != "" {
		if err = writeField(*w, "source", req.Source); err != nil {
			return nil, err
		}
	}
	if req.IndexFile != "" {
		if err = writeField(*w, "indexFile", req.IndexFile); err != nil {
			return nil, err
		}
	}
	if req.Note != "" {
		if err = writeField(*w, "note", req.Note); err != nil {
			return nil, err
		}
	}

	w.Close()
	httpReq, err := http.NewRequest("POST", cli.opsGenieAPIURL+attachFileAlertURL, &b)
	if err != nil {
		message := "Can not create the multipart/form-data request. " + err.Error()
		logging.Logger().Warn(message)
		return nil, errors.New(message)
	}
	httpReq.Header.Set("Content-Type", w.FormDataContentType())
	transport := &http.Transport{
		TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
		Proxy:           http.ProxyFromEnvironment,
		Dial: func(netw, addr string) (net.Conn, error) {
			conn, err := net.DialTimeout(netw, addr, cli.httpTransportSettings.ConnectionTimeout)
			if err != nil {
				message := "Error occurred while connecting: " + err.Error()
				logging.Logger().Warn(message)
				return nil, errors.New(message)
			}
			conn.SetDeadline(time.Now().Add(cli.httpTransportSettings.RequestTimeout))
			return conn, nil
		},
	}
	client := &http.Client{Transport: transport}
	// proxy settings
	if cli.proxy != nil {
		proxyURL, proxyErr := url.Parse(cli.proxy.toString())
		if proxyErr != nil {
			message := "Can not set the proxy configuration " + proxyErr.Error()
			logging.Logger().Warn(message)
			return nil, errors.New(message)
		}
		transport.Proxy = http.ProxyURL(proxyURL)
	}
	url := httpReq.URL.String()
	logging.Logger().Info("Executing OpsGenie request to [" + url + "] with multipart data.")
	var res *http.Response
	for i := 0; i < cli.httpTransportSettings.MaxRetryAttempts; i++ {
		res, err = client.Do(httpReq)
		if err == nil {
			defer res.Body.Close()
			break
		}
		if res != nil {
			logging.Logger().Info(fmt.Sprintf("Retrying request [%s] ResponseCode:[%d]. RetryCount: %d", url, res.StatusCode, (i + 1)))
		} else {
			logging.Logger().Info(fmt.Sprintf("Retrying request [%s] Reason:[%s]. RetryCount: %d", url, err.Error(), (i + 1)))
		}
		time.Sleep(timeSleepBetweenRequests)
	}

	if err != nil {
		message := "Can not attach the file, unable to send the request. " + err.Error()
		logging.Logger().Warn(message)
		return nil, errors.New(message)
	}

	httpStatusCode := res.StatusCode
	if httpStatusCode >= 400 {
		body, err := ioutil.ReadAll(res.Body)
		if err == nil {
			return nil, errorMessage(httpStatusCode, string(body[:]))
		}
		message := fmt.Sprint("Couldn't read the response, %s", err.Error())
		logging.Logger().Warn(message)
		return nil, errors.New(message)
	}

	attachFileAlertResp := alerts.AttachFileAlertResponse{Status: res.Status, Code: res.StatusCode}
	return &attachFileAlertResp, nil
}
Exemplo n.º 3
0
func (cli *OpsGenieAlertClient) AttachFile(req alerts.AttachFileAlertRequest) (*alerts.AttachFileAlertResponse, error) {
	req.ApiKey = cli.apiKey
	// validate the mandatory parameters: apiKey, alertId/alias, attachment
	if req.ApiKey == "" {
		return nil, errors.New("ApiKey is a mandatory field and can not be empty.")
	}
	if req.Attachment == "" {
		return nil, errors.New("Attachment is a mandatory field and can not be empty.")
	}
	if req.AlertId == "" && req.Alias == "" {
		return nil, errors.New("At least either Alert Id or Alias should be set in the request.")
	}
	if req.AlertId != "" && req.Alias != "" {
		return nil, errors.New("Either Alert Id or Alias should be set in the request not both.")
	}

	var b bytes.Buffer
	w := multipart.NewWriter(&b)

	fileName := req.Attachment
	file, err := os.Open(fileName)
	if err != nil {
		return nil, errors.New("Attachment can not be opened for reading.")
	}
	// add the attachment
	fw, err := w.CreateFormFile("attachment", fileName)
	if err != nil {
		return nil, errors.New("Can not build the request with the field attachment.")
	}
	if _, err := io.Copy(fw, file); err != nil {
		return nil, errors.New("Can not copy the attachment into the request.")
	}

	defer file.Close()

	// Add the other fields
	// empty fields should not be placed into the request
	// otherwise it yields an incomplete boundary exception
	if req.ApiKey != "" {
		if fw, err = w.CreateFormField("apiKey"); err != nil {
			return nil, errors.New("Can not build the request with the field apiKey.")
		}
		if _, err = fw.Write([]byte(req.ApiKey)); err != nil {
			return nil, errors.New("Can not write the ApiKey value into the request.")
		}
	}
	if req.AlertId != "" {
		if fw, err = w.CreateFormField("alertId"); err != nil {
			return nil, errors.New("Can not build the request with the field alertId.")
		}
		if _, err = fw.Write([]byte(req.AlertId)); err != nil {
			return nil, errors.New("Can not write the AlertId value into the request.")
		}
	}
	if req.Alias != "" {
		if fw, err = w.CreateFormField("alias"); err != nil {
			return nil, errors.New("Can not build the request with the field alias.")
		}
		if _, err = fw.Write([]byte(req.Alias)); err != nil {
			return nil, errors.New("Can not write the Alias value into the request.")
		}
	}
	if req.User != "" {
		if fw, err = w.CreateFormField("user"); err != nil {
			return nil, errors.New("Can not build the request with the field user.")
		}
		if _, err = fw.Write([]byte(req.User)); err != nil {
			return nil, errors.New("Can not write the User value into the request.")
		}
	}
	if req.Source != "" {
		if fw, err = w.CreateFormField("source"); err != nil {
			return nil, errors.New("Can not build the request with the field source.")
		}
		if _, err = fw.Write([]byte(req.Source)); err != nil {
			return nil, errors.New("Can not write the Source value into the request.")
		}
	}
	if req.IndexFile != "" {
		if fw, err = w.CreateFormField("indexFile"); err != nil {
			return nil, errors.New("Can not build the request with the field indexFile.")
		}
		if _, err = fw.Write([]byte(req.IndexFile)); err != nil {
			return nil, errors.New("Can not write the IndexFile value into the request.")
		}
	}
	if req.Note != "" {
		if fw, err = w.CreateFormField("note"); err != nil {
			return nil, errors.New("Can not build the request with the field note.")
		}
		if _, err = fw.Write([]byte(req.Note)); err != nil {
			return nil, errors.New("Can not write the Note value into the request.")
		}
	}

	w.Close()

	httpReq, err := http.NewRequest("POST", ATTACH_FILE_ALERT_URL, &b)
	if err != nil {
		return nil, errors.New("Can not create the multipart/form-data request.")
	}
	httpReq.Header.Set("Content-Type", w.FormDataContentType())
	client := &http.Client{}
	// proxy settings
	if cli.proxy != "" {
		proxyUrl, proxyErr := url.Parse(cli.proxy)
		if proxyErr != nil {
			return nil, errors.New("Can not set the proxy configuration")
		}
		client.Transport = &http.Transport{Proxy: http.ProxyURL(proxyUrl)}
	}
	var res *http.Response
	for i := 0; i < cli.retries; i++ {
		res, err = client.Do(httpReq)
		if err == nil {
			break
		}
		time.Sleep(TIME_SLEEP_BETWEEN_REQUESTS)
	}

	if err != nil {
		return nil, errors.New("Can not attach the file, unable to send the request.")
	}

	// check the returning HTTP status code
	httpStatusCode := res.StatusCode
	if httpStatusCode >= 400 && httpStatusCode < 500 {
		return nil, errors.New(fmt.Sprintf("Client error %d returned", httpStatusCode))
	}
	if httpStatusCode >= 500 {
		return nil, errors.New(fmt.Sprintf("Server error %d returned", httpStatusCode))
	}

	attachFileAlertResp := alerts.AttachFileAlertResponse{Status: res.Status, Code: res.StatusCode}
	return &attachFileAlertResp, nil
}