Method UploadFile

coderd/provisionerdserver/provisionerdserver.go:1488–1542

(stream proto.DRPCProvisionerDaemon_UploadFileStream)

}

func (s *server) UploadFile(stream proto.DRPCProvisionerDaemon_UploadFileStream) error {
	var file *sdkproto.DataBuilder
	// Always terminate the stream with an empty response.
	//nolint:errcheck // We can't do much about send errors here.
	defer stream.SendAndClose(&proto.Empty{})

	file, err := provisionersdk.HandleReceivingDataUpload(stream)
	if err != nil {
		return err
	}

	fileData, err := file.Complete()
	if err != nil {
		return xerrors.Errorf("complete file upload: %w", err)
	}

	// Just rehash the data to be sure it is correct.
	hashBytes := sha256.Sum256(fileData)
	hash := hex.EncodeToString(hashBytes[:])

	var insert database.InsertFileParams

	switch file.Type {
	case sdkproto.DataUploadType_UPLOAD_TYPE_MODULE_FILES:
		insert = database.InsertFileParams{
			ID:        uuid.New(),
			Hash:      hash,
			CreatedAt: dbtime.Now(),
			CreatedBy: uuid.Nil,
			Mimetype:  tarMimeType,
			Data:      fileData,
		}
	default:
		return xerrors.Errorf("unsupported file upload type: %s", file.Type)
	}

	//nolint:gocritic // Provisionerd actor
	_, err = s.Database.InsertFile(dbauthz.AsProvisionerd(s.lifecycleCtx), insert)
	if err != nil {
		// Duplicated files already exist in the database, so we can ignore this error.
		if !database.IsUniqueViolation(err, database.UniqueFilesHashCreatedByKey) {
			return xerrors.Errorf("insert file: %w", err)
		}
	}

	s.Logger.Info(s.lifecycleCtx, "file uploaded to database",
		slog.F("type", file.Type.String()),
		slog.F("hash", hash),
		slog.F("size", len(fileData)),
		// new_insert indicates whether the file was newly inserted or already existed.
		slog.F("new_insert", err == nil),
	)

	return nil
}

// DownloadFile pulls the requested file from the database and sends it over the protobuf stream in chunks.
func (s *server) DownloadFile(request *proto.FileRequest, stream proto.DRPCProvisionerDaemon_DownloadFileStream) error {